用人类纠正动作训练机器人,让策略更鲁棒。
Set-Supervised Diffusion Policy: Learning Action-Chunking Diffusion through Corrections

- 利用正负动作块对比学习,从人类纠正中提取监督信号。
- 在多个任务中提升策略性能,尤其在噪声数据下表现更优。
- 适合需要人机协作调试的机器人学习场景。
扩散策略近期成为机器人操作的强大框架。然而,与其它行为克隆方法类似,其对分布偏移敏感,部署时常需人工干预纠正失败。这些交互自然产生配对监督:机器人的错误动作与人类教师的纠正动作。现有数据聚合流程和标准行为克隆损失大多忽略错误动作中的负信号,导致过度拟合教师动作,且更依赖昂贵的专家数据。为此,我们提出集合监督扩散策略(SDP),一种利用对比动作块训练扩散策略的新框架。通过正负动作块构建期望动作集,并设计训练流程使扩散策略与该集合对齐。在多个机器人操作任务上的大量实验表明,SDP持续提升策略性能,尤其在噪声数据下增益显著。此外,SDP能生成高质量聚合数据集,实现更高效可靠的从人机协作纠正中学习。代码已公开于 https://set-supervised-diffusion-policy.github.io/。
原文摘要 · Abstract (English)
Diffusion policies have recently emerged as a powerful framework for robotic manipulation. However, like other behavior cloning methods, they remain vulnerable to distributional shift, often requiring human-in-the-loop interventions to correct failures during deployment. These interactions naturally provide paired supervision in the form of the robot's undesired actions and the human teacher's corrective actions. Yet existing data aggregation pipelines and standard behavior cloning losses largely ignore this negative signal from undesired actions, leading to overfitting to teacher's actions and an increasing reliance on costly expert data. To address this limitation, we propose Set-Supervised Diffusion Policy (SDP), a novel learning framework that utilizes contrastive action-chunk data to train diffusion policies from human corrections. From paired positive and negative action-chunks, SDP constructs a set of desired action-chunks and designs a training pipeline that encourages the diffusion policy to align with the set. Through extensive experiments across multiple robotic manipulation tasks, we demonstrate that SDP consistently improves policy performance, with particularly strong gains in robustness to noisy data. Moreover, SDP induces high-quality aggregated datasets, enabling more efficient and reliable policy learning from human-in-the-loop corrections. Our code is available at https://set-supervised-diffusion-policy.github.io/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。