arXiv:2608.05084cs.LGcs.RO2026-08

让扩散策略自动决定何时停止,提速近3倍还更准

Learning When to Stop: Prefix-Optimal Dynamic Diffusion Policies for Continuous Control

  • 通过递归学习每步去噪的前缀价值函数,指导中间输出质量
  • 在MuJoCo上减少2.7倍去噪步数,任务性能几乎无损
  • 适合需要高效连续控制的强化学习应用

扩散策略在连续控制中表现强大,但其迭代去噪过程带来显著计算瓶颈。为降低开销,需根据动作难度动态调整去噪步数,同时保持任务性能。本文提出前缀最优生成策略(POGP),通过去噪链上的贝尔曼递归学习每个中间步骤的前缀价值函数。该函数兼具双重作用:作为辅助训练目标,促使中间输出趋向高质量动作;在测试时提供终止规则,当后续步骤难以提升性能时即停止去噪。在四个MuJoCo环境中的实验表明,与12个基线对比,POGP将所需去噪迭代次数减少约2.7倍,同时保持接近完整的任务性能。相比当前最先进的动态扩散基线,前缀训练还将最终任务性能提升约3.5%。结果表明,监督中间去噪步骤不仅有助于自适应早停,还能作为辅助目标提升整体策略性能。

原文摘要 · Abstract (English)

Diffusion policies are a powerful policy class for continuous control, but their iterative denoising process creates a substantial computational bottleneck. Reducing this cost requires adapting the number of denoising steps to the difficulty of each action while preserving task performance. We introduce Prefix-Optimal Generative Policies (POGP), a framework that learns a prefix value function at every intermediate denoising step through a Bellman-style recursion over the denoising chain. The prefix value function serves two purposes: it provides an auxiliary training objective that encourages intermediate outputs to become high-quality actions, and it enables a test-time stopping rule that terminates denoising when additional steps are unlikely to produce meaningful improvement. Across four MuJoCo environments and comparisons with 12 baselines, POGP reduces the required number of denoising iterations by approximately 2.7-fold while retaining near-full task performance. Compared with state-of-the-art dynamic diffusion baselines, prefix training also improves final task performance by approximately 3.5%. These results indicate that supervising intermediate denoising steps is useful not only for adaptive early stopping, but also as an auxiliary objective that improves the learned policy.

扩散模型连续控制动态停止强化学习

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