提出单步扩散生成运动基元,实现高速实时机器人动作控制。
FODMP: Fast One-Step Diffusion of Movement Primitives Generation for Time-Dependent Robot Actions
- 将扩散模型压缩到运动基元参数空间,用单步解码生成动作。
- 在标准基准上比MPD快10倍,比分段策略快7倍,成功率相当或更高。
- 适合需要快速响应的场景,如抓取高速飞行物体。
扩散模型在机器人学习中应用日益广泛,但现有方法存在明显权衡:如ManiCM等动作分段扩散策略运行速度快,但仅能预测短时动作片段,难以捕捉具有加速度与减速度动态特性的时变运动基元。近期的运动基元扩散(MPD)通过概率动态运动基元(ProDMPs)参数化完整轨迹,实现了时间结构化动作生成,但其将运动解码器嵌入多步扩散过程,导致推理延迟过高,不适用于实时控制。本文提出FODMP(Fast One-step Diffusion of Movement Primitives),通过单步一致性蒸馏将扩散模型提炼至ProDMP轨迹参数空间,实现单步生成。该方法保留了运动基元的时间结构,同时消除推理瓶颈,在MetaWorld和ManiSkill等标准操控基准测试中,推理速度比MPD快10倍,比分段扩散策略快7倍,成功率相当或更优。此外,得益于快速的加速-减速动作生成能力,FODMP使机器人可实时拦截并安全接住高速飞行球体,而传统分段策略与MPD响应过慢,无法完成此类任务。
原文摘要 · Abstract (English)
Diffusion models are increasingly used for robot learning, but current designs face a clear trade-off. Action-chunking diffusion policies like ManiCM are fast to run, yet they only predict short segments of motion. This makes them reactive, but unable to capture time-dependent motion primitives, such as following a spring-damper-like behavior with built-in dynamic profiles of acceleration and deceleration. Recently, Movement Primitive Diffusion (MPD) partially addresses this limitation by parameterizing full trajectories using Probabilistic Dynamic Movement Primitives (ProDMPs), thereby enabling the generation of temporally structured motions. Nevertheless, MPD integrates the motion decoder directly into a multi-step diffusion process, resulting in prohibitively high inference latency that limits its applicability in real-time control settings. We propose FODMP (Fast One-step Diffusion of Movement Primitives), a new framework that distills diffusion models into the ProDMPs trajectory parameter space and generates motion using a single-step decoder. FODMP retains the temporal structure of movement primitives while eliminating the inference bottleneck through single-step consistency distillation. This enables robots to execute time-dependent primitives at high inference speed, suitable for closed-loop vision-based control. On standard manipulation benchmarks (MetaWorld, ManiSkill), FODMP runs up to 10 times faster than MPD and 7 times faster than action-chunking diffusion policies, while matching or exceeding their success rates. Beyond speed, by generating fast acceleration-deceleration motion primitives, FODMP allows the robot to intercept and securely catch a fast-flying ball, whereas action-chunking diffusion policy and MPD respond too slowly for real-time interception.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。