arXiv:2505.19223cs.LG2025-05ACL被引 255

降低扩散模型训练方差,让大语言模型更懂人类偏好

LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Models

论文配图:LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Models
图 1 · 摘自论文原文
  • 通过理论分析优化梯度方差,提升偏好对齐稳定性
  • 在GSM8K等任务上性能提升4.7%以上,显著优于基线
  • 适合追求高效、稳定对齐的生成模型研究者使用

尽管掩码扩散模型(MDM)如LLaDA在语言建模中展现潜力,但其通过强化学习对齐人类偏好的研究仍较少。主要挑战源于基于证据下界(ELBO)的似然估计方差过高。为此,本文提出方差减少的偏好优化(VRPO)框架,系统分析了ELBO估计器的方差,并推导出偏好优化梯度的偏差与方差上界。基于此理论,引入无偏方差缩减策略,包括最优蒙特卡洛预算分配和反向采样,显著提升MDM对齐效果。将该方法应用于LLaDA,得到的LLaDA 1.5在数学(GSM8K +4.7)、代码(HumanEval +3.0, MBPP +1.8)及对齐基准(IFEval +4.0, Arena-Hard +4.3)上均显著超越仅用监督微调的前代模型。此外,其数学表现媲美当前主流语言扩散模型与自回归模型。

原文摘要 · Abstract (English)

While Masked Diffusion Models (MDMs), such as LLaDA, present a promising paradigm for language modeling, there has been relatively little effort in aligning these models with human preferences via reinforcement learning. The challenge primarily arises from the high variance in Evidence Lower Bound (ELBO)-based likelihood estimates required for preference optimization. To address this issue, we propose Variance-Reduced Preference Optimization (VRPO), a framework that formally analyzes the variance of ELBO estimators and derives bounds on both the bias and variance of preference optimization gradients. Building on this theoretical foundation, we introduce unbiased variance reduction strategies, including optimal Monte Carlo budget allocation and antithetic sampling, that significantly improve the performance of MDM alignment. We demonstrate the effectiveness of VRPO by applying it to LLaDA, and the resulting model, LLaDA 1.5, outperforms its SFT-only predecessor consistently and significantly across mathematical (GSM8K +4.7), code (HumanEval +3.0, MBPP +1.8), and alignment benchmarks (IFEval +4.0, Arena-Hard +4.3). Furthermore, LLaDA 1.5 demonstrates a highly competitive mathematical performance compared to strong language MDMs and ARMs. Project page: https://ml-gsai.github.io/LLaDA-1.5-Demo/.

扩散模型偏好对齐优化算法语言模型

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