让扩散策略在测试时自我修正并积累经验,提升任务成功率。
ReGuide: From Test-Time Guidance to Self-Improving Diffusion Policies

- 用分阶段引导生成纠正轨迹,只在可恢复阶段干预
- 成功纠正的轨迹回传给模型,使策略持续改进
- 适合需要高鲁棒性的机器人控制场景
行为克隆的扩散策略表达能力强,但对协变量偏移敏感:演示状态的微小偏差可能累积导致任务失败。现有方法要么通过专家修正或合成增强扩展训练分布(成本高或依赖假设),要么仅在测试时用学习模型引导冻结策略(纠正轨迹用完即弃)。我们提出 ReGuide,一个自提升框架,将引导回溯视为可重用的在线恢复数据。ReGuide 首先使用相位条件引导(PCG)生成纠正轨迹:构建分阶段潜在目标,仅在漂移但可恢复阶段引导,并通过估计的干净动作匹配动态模型的训练分布。成功引导的轨迹随后通过 ReGuide-FT(微调当前检查点)或 ReGuide-FS(从头重新训练增广数据集)回传至策略;两者可组合迭代。在 Robomimic Can、Square、Transport 和 Tool Hang 上,ReGuide 将基础策略成功率提升 1.3 到 7.7 倍,在仅测试时引导设置中优于 LPB,匹配数据消融实验表明收益来自引导恢复数据,而非额外回溯本身。
原文摘要 · Abstract (English)
Behavior-cloned diffusion policies are expressive but remain vulnerable to covariate shift: small deviations from demonstrated states can compound into task failure. Existing methods address this either by expanding the training distribution through expert corrections or synthetic augmentation, or by steering a frozen policy at test time with guidance from a learned model. The former can be expensive or assumption-dependent, while the latter discards the corrected trajectories after execution. We introduce ReGuide, a self-improving framework that treats guided rollouts as reusable on-policy recovery data. ReGuide first uses Phase-Conditioned Guidance (PCG) to generate corrective rollouts: it constructs phase-specific latent targets, applies guidance only in the drifted-but-recoverable regime, and guides through the estimated clean action to match the dynamics model's training distribution. Successful guided rollouts are then absorbed back into the policy through ReGuide-FT, which fine-tunes the current checkpoint, or ReGuide-FS, which retrains from scratch on the augmented dataset; the two can also be composed and iterated. On Robomimic Can, Square, Transport, and Tool Hang, ReGuide improves base-policy success by $1.3$--$7.7\times$, outperforms LPB in the test-time-only setting, and matched-data ablations show that the gains come from guided recovery data rather than additional rollouts alone.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。