arXiv:2601.08421cs.LG2026-01被引 5

解释在线策略强化学习为何更快收敛并提出新采样方法

Coverage Improvement and Fast Convergence of On-policy Preference Learning

  • 提出覆盖率提升原理,说明在线更新能逐步改善数据覆盖范围
  • 证明当批量足够大时,算法收敛速度呈指数级,远快于离线方法
  • 设计新型混合采样器,两轮即可保证收敛,适合追求快速训练的场景

针对语言模型对齐中的在线策略偏好学习算法(如在线直接策略优化DPO),我们从理论上解释其为何显著优于离线版本。通过分析采样策略覆盖率在训练过程中的演化,提出并严格证明了‘覆盖率提升原则’:当批量足够大时,每次更新都会进入目标附近覆盖更优的区域,使后续数据更具信息量,从而实现快速收敛。在上下文老虎机设置下,使用布拉德利-特里偏好和线性软最大值策略类,我们证明在线DPO在批量超过广义覆盖率阈值时,收敛速度为指数级。而受限于初始策略离线样本的任何学习器,其最小最大率较慢,导致总样本复杂度出现明显差距。基于此分析,我们进一步提出一种基于新颖‘偏好性’G-最优设计的简单混合采样器,消除对覆盖率的依赖,仅需两轮即保证收敛。最后,在一般函数类设定下,我们提出了有理论依据的在线奖励蒸馏方案,并在另一种基于偏差的覆盖率定义下展示了更快的无噪声收敛速率。实验表明,在线DPO及所提奖励蒸馏算法均优于离线版本,且性能随迭代稳定单调提升。

原文摘要 · Abstract (English)

Online on-policy preference learning algorithms for language model alignment such as online direct policy optimization (DPO) can significantly outperform their offline counterparts. We provide a theoretical explanation for this phenomenon by analyzing how the sampling policy's coverage evolves throughout on-policy training. We propose and rigorously justify the \emph{coverage improvement principle}: with sufficient batch size, each update moves into a region around the target where coverage is uniformly better, making subsequent data increasingly informative and enabling rapid convergence. In the contextual bandit setting with Bradley-Terry preferences and linear softmax policy class, we show that on-policy DPO converges exponentially in the number of iterations for batch size exceeding a generalized coverage threshold. In contrast, any learner restricted to offline samples from the initial policy suffers a slower minimax rate, leading to a sharp separation in total sample complexity. Motivated by this analysis, we further propose a simple hybrid sampler based on a novel \emph{preferential} G-optimal design, which removes dependence on coverage and guarantees convergence in just two rounds. Finally, we develop principled on-policy schemes for reward distillation in the general function class setting, and show faster noiseless rates under an alternative deviation-based notion of coverage. Experimentally, we confirm that on-policy DPO and our proposed reward distillation algorithms outperform their off-policy counterparts and enjoy stable, monotonic performance gains across iterations.

强化学习偏好学习收敛性分析在线训练

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