DiscoDemo: Discovering Efficient and Diverse Robot Demonstrations for Imitation Learning
Abstract
DiscoDemo uses reinforcement learning to generate diverse successful robot manipulation trajectories for training more robust visuomotor imitation policies.
Behavior cloning for robotic manipulation is highly sensitive to the coverage of its training data, yet collecting diverse successful demonstrations remains costly. We study reinforcement learning (RL) not as a deployable controller, but as a data generator for downstream visuomotor imitation learning. Our key premise is that multiple successful solution modes cover a broader range of successful trajectories than repeatedly collecting a single narrow solution. We introduce DiscoDemo, which trains a privileged latent-conditioned RL generator with a parallel reverse curriculum and a success-gated diversity objective to discover diverse solutions while preserving task success. The trained generator produces multiple successful closed-loop behaviors, which are safety-filtered and exported as a static dataset for downstream imitation learning. Across four manipulation tasks, DiscoDemo produces broader successful-state coverage and improves downstream visuomotor policies in both simulation and the real world. Averaged across tasks, downstream success increases to 62% from 40% for the strongest baseline in simulation, and to 78% from 39% on the real robot, respectively. These results suggest that successful solution diversity improves imitation learning for a pretrained flow-matching VLA and remains complementary to corrective data collection. Project page: https://davian-robotics.github.io/DiscoDemo
Community
DiscoDemo: Diverse Robot Demonstrations from Just 20 Human Demos
What DiscoDemo offers:
- ๐ฏ 20 โ 3,000 demonstrations: Turn a small set of human demonstrations into 3,000 successful robot demonstrations through simulation-based reinforcement learning.
- ๐ Diversity by design: Discover multiple successful strategies for the same task, rather than generating near-identical trajectories.
- ๐ง Skill-conditioned generation: Combine a reverse curriculum with a success-gated diversity reward to generate diverse manipulation behaviors.
- ๐ค Real-robot results: Improve ฯโ.โ 's task success rate from 39% to 78% by fine-tuning on generated demonstrations.
- ๐ฆ Open resources: Explore the released DiscoDemo datasets on Hugging Face and use them to experiment with diverse demonstration generation and robot policy learning.
๐ค Hugging Face datasets: https://huggingface.co/collections/DAVIAN-Robotics/discodemo-6ac696e70d028fddf0c3dfeb
๐ Project page: https://davian-robotics.github.io/DiscoDemo/
๐ป Code: https://github.com/DAVIAN-Robotics/DiscoDemo
๐ Paper: https://arxiv.org/abs/2610.32093
Feedback and questions are very welcome!
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- SynthDemo-RL: Breaking the Zero-Reward Barrier in VLA Adaptation with LLM-Guided Synthetic Demonstrations (2026)
- Task-Space Imitation Guidance for Efficient Reinforcement Learning (2026)
- DynaForge: Planning-Guided Residual Learning for Dynamic Manipulation Demonstration Generation (2026)
- Res-HIL: Human-Guided Residual Reinforcement Learning for Sample-Efficient Dexterous Manipulation (2026)
- Towards High-DoF Dexterous Manipulation through VLA Post-Training (2026)
- OneDemo: Unified Policy Learning and Data Generation from a Single Demonstration (2026)
- Find Something You Can't Do: Agentic Real-World Reinforcement Learning for Self-Improving VLA Models (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Get this paper in your agent:
hf papers read 2610.32093 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 16
DAVIAN-Robotics/DiscoDemo-Stage1_RL-PnP_Banana
Datasets citing this paper 12
DAVIAN-Robotics/DiscoDemo-Stage2_GenData-StackCube-alpha0
DAVIAN-Robotics/DiscoDemo-Stage2_GenData-FMB_SqCircle
Spaces citing this paper 0
No Space linking this paper