arXiv:2512.00499cs.LGcs.AI2025-12被引 9

ESPO通过熵分组提升强化学习训练效率与稳定性

ESPO: Entropy Importance Sampling Policy Optimization

  • 按预测熵分组序列,实现细粒度梯度更新
  • 动态裁剪机制使训练收敛速度提升30%以上
  • 适合需要稳定长文本生成的LLM微调场景

强化学习已成为大语言模型后训练的核心组件,尤其在需长时间生成的复杂推理任务中。然而,大规模训练常面临训练稳定性和效率之间的根本矛盾:令牌级优化虽精细但梯度方差高,易导致训练不稳定;序列级优化依赖激进裁剪以保证稳定,却会丢弃大量有效样本,造成梯度利用不足。本文提出熵重要性采样策略优化(ESPO),通过基于预测熵对序列分组,实现(1)熵分组重要性采样以捕捉序列内异质性,(2)熵自适应裁剪以根据模型不确定性动态分配信任区域。在数学推理基准上的实验表明,ESPO不仅加速收敛,还在挑战性任务上达到最优性能,显著提升准确率。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has become a central component of post-training for large language models (LLMs), particularly for complex reasoning tasks that require stable optimization over long generation horizons. However, achieving performance at scale often introduces a fundamental trade-off between training stability and training efficiency. Token-level optimization applies fine-grained updates at the individual units, but is prone to high variance in gradient estimation, which can result in unstable training dynamics. In contrast, Sequence-level optimization often relies on aggressive clipping mechanisms to ensure stable updates. However, such design may discard a large fraction of valid training samples, leading to inefficient gradient utilization and reduced training efficiency. We refer to this phenomenon as gradient underutilization. In this work, we propose Entropy Importance Sampling Policy Optimization (ESPO), a novel framework that aims to combine fine-grained updates with stable training. ESPO decomposes sequences into groups based on predictive entropy, enabling (1) Entropy Grouping Importance Sampling to capture intra-sequence heterogeneity, and (2) Entropy Adaptive Clipping to dynamically allocate trust regions based on model uncertainty. Extensive experiments on mathematical reasoning benchmarks demonstrate that ESPO not only accelerates convergence but also achieves state-of-the-art performance, notably improving accuracy on the challenging mathematical benchmarks.

强化学习大模型微调序列优化熵采样

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