LP-SFT: Local-Preserving Supervised Fine-Tuning via Multimodal Entropy Structure

Abstract: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.

20 pages, 3 figures. Code is available at this https URL

Computation and Language (cs.CL); Machine Learning (cs.LG)

arXiv:2607.04733 [cs.CL]

(or arXiv:2607.04733v2 [cs.CL] for this version)

https://doi.org/10.48550/arXiv.2607.04733 Focus to learn more arXiv-issued DOI via DataCite

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.