arXiv:2605.29303cs.AI2026-05

通过熵与KL散度筛选,让大模型在少样本时更专注任务能力而非死记硬背。

Entropy-KL Divergence-based Token Masking: A Novel Approach for Selective Fine-tuning of Large Language Models

论文配图:Entropy-KL Divergence-based Token Masking: A Novel Approach for Selective Fine-tuning of Large Language Models
图 1 · 摘自论文原文
  • 用熵和KL散度识别高不确定性的词元并屏蔽它们
  • 在数学推理任务上比标准微调提升显著,后续强化学习表现更好
  • 适合数据稀缺场景,尤其对需要保持原始分布的模型优化有帮助

监督微调(SFT)后接强化学习(RL)已成为大语言模型后训练的标准范式。该范式为RL探索提供冷启动,避免纯强化学习中策略采样导致正样本不足的问题。然而,实践中常用于SFT的数据量远小于RL阶段,易使模型拟合有限样本,偏离预训练分布,从而阻碍后续RL的有效探索。为此,我们提出在低数据情况下,SFT应优先激活任务相关能力,而非记忆具体内容。据此,我们提出EKSFT(熵-KL选择性微调),通过屏蔽高熵或与参考模型高KL散度的词元,排除高不确定性、分布偏移的样本进行模仿学习。这使得模型注入任务知识的同时保持预训练分布完整性。在数学推理基准上的实证评估表明,EKSFT持续优于标准SFT;从EKSFT模型出发的进一步强化学习也表现出更优的最终性能,说明其提升了RL阶段的探索能力。代码与数据集见https://github.com/MINE-USTC/EKSFT。

原文摘要 · Abstract (English)

Supervised fine-tuning (SFT) followed by reinforcement learning (RL) has become a standard post-training paradigm for large language models. This paradigm provides a cold-start for RL exploration, avoiding the inefficiency of pure RL where on-policy sampling yields insufficient positive samples. However, in practice, existing approaches often use a small amount of data for SFT initialization compared to the RL phase, which can cause the model to fit the limited samples and shift away from its pre-trained distribution. This distribution shift impedes the model's ability to effectively explore during subsequent RL training. To address this challenge, we propose that in low-data regimes, SFT should prioritize activating task-relevant capabilities rather than memorizing specific content. Along this line, we propose EKSFT (Entropy-KL Selective Fine-Tuning), which selectively masks tokens that exhibit either high entropy or high KL divergence from a reference model. By excluding these high-uncertainty, distribution-shifting tokens from imitation, EKSFT injects task-specific knowledge while preserving the integrity of the model's pre-trained distribution. Empirical evaluations on mathematical reasoning benchmarks demonstrate that EKSFT consistently outperforms standard SFT. Further RL fine-tuning from the EKSFT model yields consistently better post-RL performance, indicating improved exploration for the RL stage. Our codes and datasets are available at https://github.com/MINE-USTC/EKSFT.

微调强化学习分布保持少样本

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