arXiv:2602.04380cs.LGcs.AI2026-02被引 1

用更灵活的散度替代KL散度,提升大模型推理性能

Beyond KL Divergence: Policy Optimization with Flexible Bregman Divergences for LLM Reasoning

  • 引入Bregman散度替代传统KL散度进行策略优化
  • 数学推理任务准确率提升5.5点,代码生成效率更高
  • 随机初始化神经镜像映射已足够实用,无需复杂调参

基于分组的策略优化方法(如GRPO)在数学推理和代码生成任务中表现优异。然而,现有方法均仅使用KL散度进行策略正则化,未探索其他散度函数的潜力。本文提出基于镜像的分组策略优化(GBMPO),将分组优化扩展至灵活的Bregman散度,包括在概率空间设计的L2散度及可学习的神经镜像映射。在GSM8K数学推理任务上,设计的ProbL2-GRPO达到86.7%准确率,较Dr. GRPO基线提升5.5个百分点;在MBPP代码生成任务上,神经镜像映射实现60.1-60.8% pass@1,且随机初始化即可获得大部分性能收益。进化策略元学习带来的准确率提升有限,但显著降低方差(±0.2 vs ±0.6)并缩短响应长度15%,表明随机初始化在多数实际场景下已足够有效。结果证明,散度选择是分组策略优化中此前被忽视的关键设计维度。

原文摘要 · Abstract (English)

Policy optimization methods like Group Relative Policy Optimization (GRPO) and its variants have achieved strong results on mathematical reasoning and code generation tasks. Despite extensive exploration of reward processing strategies and training dynamics, all existing group-based methods exclusively use KL divergence for policy regularization, leaving the choice of divergence function unexplored. We introduce Group-Based Mirror Policy Optimization (GBMPO), a framework that extends group-based policy optimization to flexible Bregman divergences, including hand-designed alternatives (L2 in probability space) and learned neural mirror maps. On GSM8K mathematical reasoning, hand-designed ProbL2-GRPO achieves 86.7% accuracy, improving +5.5 points over the Dr. GRPO baseline. On MBPP code generation, neural mirror maps reach 60.1-60.8% pass@1, with random initialization already capturing most of the benefit. While evolutionary strategies meta-learning provides marginal accuracy improvements, its primary value lies in variance reduction ($\pm$0.2 versus $\pm$0.6) and efficiency gains (15% shorter responses on MBPP), suggesting that random initialization of neural mirror maps is sufficient for most practical applications. These results establish divergence choice as a critical, previously unexplored design dimension in group-based policy optimization for LLM reasoning.

大模型推理策略优化散度函数代码生成

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