← Search

Kehan Wen

5 accepted papers

2026

Mini Diffuser: Fast Multi-Task Diffusion Policy Training Using Two-Level Mini-Batches

RA-L 2026

We present a method that reduces, by an order of magnitude, the time and memory needed to train multi-task vision-language robotic diffusion policies. This improvement arises from a previously underexplored distinction between action diffusion and the image diffusion techniques that inspired it: In

Cited by 0SourcecodeScholar
2026

Mini Diffuser: Fast Multi-Task Diffusion Policy Training Using Two-Level Mini-Batches

ICRA 2026poster

We present a method that reduces, by an order of magnitude, the time and memory needed to train multi-task vision-language robotic diffusion policies. This improvement arises from a previously underexplored distinction between action diffusion and the image diffusion techniques that inspired it: In …

2025

Constrained Style Learning from Imperfect Demonstrations under Task Optimality

CoRL 2025poster

Learning from demonstration has proven effective in robotics for acquiring natural behaviors, such as stylistic motions and lifelike agility, particularly when explicitly defining style-oriented reward functions is challenging. Synthesizing stylistic motions for real-world tasks usually requires bal…

Cited by 0SourceScholar
2025

M^3PC: Test-time Model Predictive Control using Pretrained Masked Trajectory Model

ICLR 2025poster

Recent work in Offline Reinforcement Learning (RL) has shown that a unified transformer trained under a masked auto-encoding objective can effectively capture the relationships between different modalities (e.g., states, actions, rewards) within given trajectory datasets. However, this information…