arXiv:2510.07865cs.ROcs.AI2025-10被引 6

用分散正则化让流模型一步生成更精准的机械臂动作。

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

  • 在流模型中加入多层分散正则化,防止表示坍缩。
  • 推理速度提升20-40倍,成功率提高10-20个百分点。
  • 适合追求高效高精度机械臂控制的研究与应用。

学习多模态动作分布对于机器人精准、鲁棒操作至关重要。基于流的生成模型最近成为学习动作分布的有前景方案,可实现一步动作生成,相比扩散方法显著提升采样效率。然而,现有流模型存在表示坍缩问题,难以区分相似视觉表征,导致精确操作失败。本文提出DM1(MeanFlow结合分散正则化的一步机器人操作方法),将分散正则化引入MeanFlow,防止坍缩同时保持一步效率。DM1在不同中间嵌入层采用多种分散正则化变体,鼓励训练批次间多样化表征,无需额外网络模块或特殊训练流程。在RoboMimic基准测试中,DM1实现20-40倍更快推理(0.07秒对比2-3.5秒),成功率提升10-20个百分点,其中Lift任务达99%成功,优于基线85%。真实机器人部署于Franka Panda进一步验证其从仿真到现实世界的有效迁移。据我们所知,这是首个通过表征正则化使流模型在机器人操作中表现强劲的工作,建立了一种简单而高效的鲁棒操作新范式。

原文摘要 · Abstract (English)

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.

机器人控制流模型一步生成正则化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。