DM1: MeanFlow with Dispersive Regularization for 1-Step Robotic Manipulation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zou, Guowei, Wang, Haitao, Wu, Hejun, Qian, Yukun, Wang, Yuhang, Li, Weibing
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914082436677632
author Zou, Guowei
Wang, Haitao
Wu, Hejun
Qian, Yukun
Wang, Yuhang
Li, Weibing
author_facet Zou, Guowei
Wang, Haitao
Wu, Hejun
Qian, Yukun
Wang, Yuhang
Li, Weibing
contents The ability to learn multi-modal action distributions is indispensable for robotic manipulation policies to perform precise and robust control. Flow-based generative models have recently emerged as a promising solution to learning distributions of actions, offering one-step action generation and thus achieving much higher sampling efficiency compared to diffusion-based methods. However, existing flow-based policies suffer from representation collapse, the inability to distinguish similar visual representations, leading to failures in precise manipulation tasks. We propose DM1 (MeanFlow with Dispersive Regularization for One-Step Robotic Manipulation), a novel flow matching framework that integrates dispersive regularization into MeanFlow to prevent collapse while maintaining one-step efficiency. DM1 employs multiple dispersive regularization variants across different intermediate embedding layers, encouraging diverse representations across training batches without introducing additional network modules or specialized training procedures. Experiments on RoboMimic benchmarks show that DM1 achieves 20-40 times faster inference (0.07s vs. 2-3.5s) and improves success rates by 10-20 percentage points, with the Lift task reaching 99% success over 85% of the baseline. Real-robot deployment on a Franka Panda further validates that DM1 transfers effectively from simulation to the physical world. To the best of our knowledge, this is the first work to leverage representation regularization to enable flow-based policies to achieve strong performance in robotic manipulation, establishing a simple yet powerful approach for efficient and robust manipulation.
format Preprint
id arxiv_https___arxiv_org_abs_2510_07865
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DM1: MeanFlow with Dispersive Regularization for 1-Step Robotic Manipulation
Zou, Guowei
Wang, Haitao
Wu, Hejun
Qian, Yukun
Wang, Yuhang
Li, Weibing
Robotics
Artificial Intelligence
The ability to learn multi-modal action distributions is indispensable for robotic manipulation policies to perform precise and robust control. Flow-based generative models have recently emerged as a promising solution to learning distributions of actions, offering one-step action generation and thus achieving much higher sampling efficiency compared to diffusion-based methods. However, existing flow-based policies suffer from representation collapse, the inability to distinguish similar visual representations, leading to failures in precise manipulation tasks. We propose DM1 (MeanFlow with Dispersive Regularization for One-Step Robotic Manipulation), a novel flow matching framework that integrates dispersive regularization into MeanFlow to prevent collapse while maintaining one-step efficiency. DM1 employs multiple dispersive regularization variants across different intermediate embedding layers, encouraging diverse representations across training batches without introducing additional network modules or specialized training procedures. Experiments on RoboMimic benchmarks show that DM1 achieves 20-40 times faster inference (0.07s vs. 2-3.5s) and improves success rates by 10-20 percentage points, with the Lift task reaching 99% success over 85% of the baseline. Real-robot deployment on a Franka Panda further validates that DM1 transfers effectively from simulation to the physical world. To the best of our knowledge, this is the first work to leverage representation regularization to enable flow-based policies to achieve strong performance in robotic manipulation, establishing a simple yet powerful approach for efficient and robust manipulation.
title DM1: MeanFlow with Dispersive Regularization for 1-Step Robotic Manipulation
topic Robotics
Artificial Intelligence
url https://arxiv.org/abs/2510.07865