English

One Step Is Enough: Dispersive MeanFlow Policy Optimization

Robotics 2026-01-29 v1

Abstract

Real-time robotic control demands fast action generation. However, existing generative policies based on diffusion and flow matching require multi-step sampling, fundamentally limiting deployment in time-critical scenarios. We propose Dispersive MeanFlow Policy Optimization (DMPO), a unified framework that enables true one-step generation through three key components: MeanFlow for mathematically-derived single-step inference without knowledge distillation, dispersive regularization to prevent representation collapse, and reinforcement learning (RL) fine-tuning to surpass expert demonstrations. Experiments across RoboMimic manipulation and OpenAI Gym locomotion benchmarks demonstrate competitive or superior performance compared to multi-step baselines. With our lightweight model architecture and the three key algorithmic components working in synergy, DMPO exceeds real-time control requirements (>120Hz) with 5-20x inference speedup, reaching hundreds of Hertz on high-performance GPUs. Physical deployment on a Franka-Emika-Panda robot validates real-world applicability.

Keywords

Cite

@article{arxiv.2601.20701,
  title  = {One Step Is Enough: Dispersive MeanFlow Policy Optimization},
  author = {Guowei Zou and Haitao Wang and Hejun Wu and Yukun Qian and Yuhang Wang and Weibing Li},
  journal= {arXiv preprint arXiv:2601.20701},
  year   = {2026}
}

Comments

Code and project page: https://guowei-zou.github.io/dmpo-page/

R2 v1 2026-07-01T09:24:06.084Z