arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2510.08554cs.LGstat.ML
收藏

通过群体扩散策略优化提升扩散语言模型的推理能力

Improving Reasoning for Diffusion Language Models via Group Diffusion Policy Optimization

Georgia Institute of Technology(佐治亚理工学院);ML Research, Morgan Stanley(摩根士丹利机器学习研究部)

Kevin Rojas, Jiahe Lin, Kashif Rasul, Anderson Schneider, Yuriy Nevmyvaka, Molei Tao, Wei Deng

AI总结:

本文提出GDPO算法,通过减少ELBO估计的方差提升扩散语言模型的推理能力,实验证明其在多个基准上优于现有方法。

摘要:

扩散语言模型(DLMs)能够通过迭代细化实现并行、无序生成,提供了一种灵活的替代方案,用于自回归大语言模型(LLMs)。然而,将强化学习(RL)微调适应到DLMs仍是一个开放性挑战,因为存在不可计算的似然。开创性工作如diffu-GRPO通过一步解屏蔽估计令牌级似然。虽然计算效率高,但这种方法存在严重的偏差。更基础的理论在于序列级似然,其中证据下界(ELBO)作为替代品。然而,尽管有这种清晰的数学联系,ELBO方法由于似然评估的成本过高而受到限制。在本文中,我们重新审视ELBO估计并分离其方差来源。这种分解促使通过在少数关键维度上快速确定性积分近似来减少方差。基于这一见解,我们引入了群体扩散策略优化(GDPO),一种专为DLMs设计的新型RL算法。GDPO利用简单而有效的半确定性蒙特卡洛方案,以减轻在常规双重蒙特卡洛采样下的ELBO估计器的方差爆炸,从而在紧致的评估预算下获得可证明的低方差估计器。实证上,GDPO在预训练检查点上实现了持续的提升,并在大多数数学、推理和编码基准上优于diffu-GRPO,这是最先进的基线之一。

英文摘要:

Diffusion language models (DLMs) enable parallel, order-agnostic generation with iterative refinement, offering a flexible alternative to autoregressive large language models (LLMs). However, adapting reinforcement learning (RL) fine-tuning to DLMs remains an open challenge because of the intractable likelihood. Pioneering work such as diffu-GRPO estimated token-level likelihoods via one-step unmasking. While computationally efficient, this approach is severely biased. A more principled foundation lies in sequence-level likelihoods, where the evidence lower bound (ELBO) serves as a surrogate. Yet, despite this clean mathematical connection, ELBO-based methods have seen limited adoption due to the prohibitive cost of likelihood evaluation. In this work, we revisit ELBO estimation and disentangle its sources of variance. This decomposition motivates reducing variance through fast, deterministic integral approximations along a few pivotal dimensions. Building on this insight, we introduce Group Diffusion Policy Optimization (GDPO), a new RL algorithm tailored for DLMs. GDPO leverages simple yet effective Semi-deterministic Monte Carlo schemes to mitigate the variance explosion of ELBO estimators under vanilla double Monte Carlo sampling, yielding a provably lower-variance estimator under tight evaluation budgets. Empirically, GDPO achieves consistent gains over pretrained checkpoints and outperforms diffu-GRPO, one of the state-of-the-art baselines, on the majority of math, reasoning, and coding benchmarks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑