通过历史动作序列提升机器人视觉动作策略的鲁棒性
CDP: Towards Robust Autoregressive Visuomotor Policy Learning via Causal Diffusion
- 用历史动作序列增强扩散模型,实现更连贯的视觉动作决策
- 在低质量观测下仍保持高精度,长时任务成功率显著提升
- 适合需要实时响应和抗干扰的机器人控制场景
扩散策略(DP)通过动作扩散模仿专家示范,使机器人学习复杂行为。但在实际应用中,硬件限制常导致数据质量下降,实时约束又要求仅基于即时状态和场景观测进行推理。这些因素严重削弱了从专家示范中学习的效能,造成物体定位失败、抓取规划失误及长时任务执行中断。为此,我们提出因果扩散策略(CDP),一种基于Transformer的扩散模型,通过引入历史动作序列作为条件,实现更连贯、上下文感知的视觉动作策略学习。为降低自回归推理带来的计算开销,还设计了缓存机制,存储前一时刻的注意力键值对,大幅减少重复计算。在模拟与真实环境中的多种2D/3D操控任务实验表明,CDP能有效利用历史动作序列,在多个任务上显著优于现有方法。即使在输入观测质量下降的情况下,仍能通过时间连续性推理保持高精度,展现出在真实、不完美条件下机器人控制的卓越鲁棒性。
原文摘要 · Abstract (English)
Diffusion Policy (DP) enables robots to learn complex behaviors by imitating expert demonstrations through action diffusion. However, in practical applications, hardware limitations often degrade data quality, while real-time constraints restrict model inference to instantaneous state and scene observations. These limitations seriously reduce the efficacy of learning from expert demonstrations, resulting in failures in object localization, grasp planning, and long-horizon task execution. To address these challenges, we propose Causal Diffusion Policy (CDP), a novel transformer-based diffusion model that enhances action prediction by conditioning on historical action sequences, thereby enabling more coherent and context-aware visuomotor policy learning. To further mitigate the computational cost associated with autoregressive inference, a caching mechanism is also introduced to store attention key-value pairs from previous timesteps, substantially reducing redundant computations during execution. Extensive experiments in both simulated and real-world environments, spanning diverse 2D and 3D manipulation tasks, demonstrate that CDP uniquely leverages historical action sequences to achieve significantly higher accuracy than existing methods. Moreover, even when faced with degraded input observation quality, CDP maintains remarkable precision by reasoning through temporal continuity, which highlights its practical robustness for robotic control under realistic, imperfect conditions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。