跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

Diffusion Policy Optimization without Drifting Apart

文章摘要

该论文指出扩散策略强化学习后训练存在'双重漂移'问题:优化变分代理目标会使ELBO偏离真实对数似然,导致代理策略梯度与真实期望回报梯度错位。作者提出DiPOD框架,通过自我蒸馏与策略改进梯度更新交错进行,并在每次梯度更新中加入同策略ELBO正则项,在扩散语言模型后训练和连续控制任务中显著稳定训练并提升奖励。

来源arXiv Machine Learning作者: Haozhe Jiang, Haiwen Feng, Pieter Abbeel, Jiantao Jiao, Angjoo Kanazawa, Nika Haghtalab
Diffusion Policy Optimization without Drifting Apart
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

扩散模型在图像、语言生成以及连续控制等任务中已经成为一类重要的策略表示形式。如何通过强化学习对这类扩散策略进行后训练,从而进一步提升其表现,是近期研究关注的焦点。然而,arXiv 上编号为 2606.13795 的最新论文《Diffusion Policy Optimization without Drifting Apart》指出,现有的扩散策略梯度方法往往训练不稳定,难以实现可靠的策略改进。该论文由 Haozhe Jiang 等六位作者完成,于 2026 年 6 月 11 日提交至 arXiv,归类为机器学习(cs.LG)。

作者将这种不稳定性的根源归纳为“双重漂移”(double-drift)现象。具体而言,扩散策略的优化通常依赖变分下界(ELBO)作为替代目标,但在训练过程中,ELBO 可能与真实的的对数似然逐渐分离。一旦两者出现偏离,由替代目标推导出的代理策略梯度就不再与期望回报的真实策略梯度保持一致,从而导致策略改进方向出现偏差,训练过程随之变得不稳定。

针对这一问题,论文提出了名为 DiPOD 的扩散策略优化框架。其核心思路是在整个训练过程中保持紧界(tight-bound)行为,具体做法是将自蒸馏(self-distillation)与提升策略的梯度更新交替进行。这一设计最终落地为一个简单而实用的算法:在每一次扩散策略梯度更新时,附加一个同策略(on-policy)的 ELBO 正则项,从而持续约束替代目标与真实似然之间的偏离。

为了验证方法的有效性,作者在两类场景中进行了实验:一是扩散语言模型的后训练,二是连续控制任务中的扩散策略。实验结果显示,DiPOD 能够显著稳定训练过程,并取得比以往方法更高的奖励。论文还提供了项目主页链接,方便读者进一步了解细节。

总体来看,该工作的贡献在于清晰地诊断了扩散策略强化学习后训练中的漂移问题,并给出了一种易于实现的正则化方案。这一思路有望为扩散模型在强化学习场景下的稳定优化提供新的实践参考。

展开要点与分析

文章情报

工程师进阶

要点

  • 识别出扩散策略梯度方法中的'双重漂移'现象:ELBO与真实对数似然分离,导致代理梯度与真实策略梯度错位。
  • 提出DiPOD框架,将自我蒸馏与策略改进梯度更新交错,保持紧边界行为。
  • 算法实现简单:在每次扩散策略梯度更新中加入同策略ELBO正则项。
  • 在扩散语言模型后训练和连续控制扩散策略上验证,训练更稳定且奖励更高。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。