保护预训练模型的局部偏好结构,避免微调时能力退化。
LP-SFT: Local-Preserving Supervised Fine-Tuning via Multimodal Entropy Structure

- 通过构建局部候选词集并使用KL散度保持原有偏好分布。
- 在多领域和单领域任务中均提升准确率与多样性平衡。
- 适合需要保留原始模型能力的下游应用。
监督微调(SFT)是将预训练语言模型适配到下游任务的标准方法,但常以牺牲已有能力为代价。标准交叉熵微调仅强化目标标签,未约束其他可能选项的概率分布,可能破坏预训练阶段学习到的丰富局部偏好结构。我们通过香农与瑞尼熵分析发现,预训练模型具有规律的多峰熵结构,其峰值对应不同数量的合理候选词,表明模型内含超越单一标签的分布知识。为此提出LP-SFT:一种局部保持型监督微调方法,在每一步从冻结的基础分布中提取局部top-K候选词,移除目标标签以避免冲突,并使用局部归一化的KL散度维持非标签候选间的相对偏好关系。在混合领域与单领域微调实验中,LP-SFT优于基线SFT与近期增强方法,在pass@1准确率与pass@k性能间取得最佳平衡,表明局部保持可缓解能力退化且不压缩采样可访问的多样性。
原文摘要 · Abstract (English)
Supervised fine-tuning (SFT) is the standard approach for adapting pretrained language models to downstream domains, yet it often improves target-domain behavior at the cost of degrading pre-existing capabilities. Standard cross-entropy fine-tuning promotes only the observed label token and leaves unconstrained how probability mass is redistributed over other plausible alternatives, potentially distorting the rich local preference structure learned during pretraining. We first analyze next-token predictions using Shannon and Renyi entropies, revealing that pretrained models exhibit a regular multimodal entropy structure. These entropy peaks correspond to varying numbers of plausible alternatives, indicating that the base model intrinsically encodes rich distributional knowledge beyond the single supervised token. Motivated by this observation, we propose LP-SFT, a Local-Preserving Supervised Fine-Tuning objective designed to explicitly protect this inherent entropy structure. At each step, LP-SFT constructs a local top-K support of alternative tokens from the frozen base distribution. Crucially, it removes the supervised target token from this set to avoid conflicting with the cross-entropy objective, and applies a locally normalized KL divergence to maintain the base model's relative preference structure among the remaining non-label alternatives. Across mixed-domain and single-domain fine-tuning experiments, LP-SFT improves overall performance over vanilla SFT and recent SFT-enhancement baselines, achieving the best balance between pass@1 accuracy and pass@k performance. These results suggest that local preservation helps mitigate capability degradation without collapsing sampling-accessible diversity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。