arXiv:2507.10543cs.RO2025-07AAAI被引 35

用单次网络计算生成机器人抓取轨迹,速度更快精度更高

MP1: MeanFlow Tames Policy Learning in 1-step for Robotic Manipulation

  • 通过均值流直接学习速度区间平均值,避免额外约束
  • 实测推理速度达6.8毫秒,比同类方法快19倍,成功率高10.2%
  • 适合少样本场景,轻量级损失提升泛化能力

在机器人操作中,生成式模型面临扩散模型采样慢与基于流的方法受结构约束的矛盾。本文提出MP1,将3D点云输入与均值流结合,实现一次网络评估(1-NFE)生成动作轨迹。通过直接学习区间平均速度的“均值流恒等式”,策略无需额外一致性约束,消除推理时的数值微分误差,提升轨迹精度。同时引入条件控制生成(CFG)增强可控性,仍保持1-NFE推理。针对少样本学习中细微场景变化的关键影响,设计轻量级分散损失,在训练中排斥状态嵌入,提升泛化性而不增加推理开销。在Adroit和Meta-World基准及真实场景中验证,MP1平均任务成功率优于DP3 10.2%、FlowPolicy 7.3%,平均推理时间仅6.8毫秒,比DP3快19倍,接近FlowPolicy的两倍。项目页:https://mp1-2254.github.io/,代码地址:https://github.com/LogSSim/MP1。

原文摘要 · Abstract (English)

In robot manipulation, robot learning has become a prevailing approach. However, generative models within this field face a fundamental trade-off between the slow, iterative sampling of diffusion models and the architectural constraints of faster Flow-based methods, which often rely on explicit consistency losses. To address these limitations, we introduce MP1, which pairs 3D point-cloud inputs with the MeanFlow paradigm to generate action trajectories in one network function evaluation (1-NFE). By directly learning the interval-averaged velocity via the "MeanFlow Identity", our policy avoids any additional consistency constraints. This formulation eliminates numerical ODE-solver errors during inference, yielding more precise trajectories. MP1 further incorporates CFG for improved trajectory controllability while retaining 1-NFE inference without reintroducing structural constraints. Because subtle scene-context variations are critical for robot learning, especially in few-shot learning, we introduce a lightweight Dispersive Loss that repels state embeddings during training, boosting generalization without slowing inference. We validate our method on the Adroit and Meta-World benchmarks, as well as in real-world scenarios. Experimental results show MP1 achieves superior average task success rates, outperforming DP3 by 10.2% and FlowPolicy by 7.3%. Its average inference time is only 6.8 ms-19x faster than DP3 and nearly 2x faster than FlowPolicy. Our project page is available at https://mp1-2254.github.io/, and the code can be accessed at https://github.com/LogSSim/MP1.

机器人操作生成模型轨迹生成高效推理

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