English

RL Fine-Tuning Heals OOD Forgetting in SFT

Machine Learning 2026-05-12 v3 Artificial Intelligence

Abstract

Supervised Fine-Tuning (SFT) followed by Reinforcement Learning (RL) is a standard post-training recipe for improving Large Language Models (LLM) reasoning, but why it works remains unclear. We revisit the common claim that ``SFT memorizes, RL generalizes'' through checkpoint-wise analyses of in-distribution (ID) and out-of-distribution (OOD) reasoning. We find that OOD performance often peaks early during SFT and then declines despite continued improvement in ID reasoning. RL typically does not surpass this early SFT peak; rather, it restores OOD capability lost during later SFT, and only from a bounded range of SFT checkpoints. Further spectral analysis shows that this forgetting-and-recovery pattern correlates with rotations of singular vectors, while singular values remain largely stable. These findings suggest a more precise view of post-training dynamics: SFT can forget, RL can recover, and controlling singular-vector rotation may improve OOD robustness. Code is available at \href{https://github.com/jinhangzhan/RL\_Heals\_SFT.git}{https://github.com/jinhangzhan/RL\_Heals\_SFT}.

Keywords

Cite

@article{arxiv.2509.12235,
  title  = {RL Fine-Tuning Heals OOD Forgetting in SFT},
  author = {Hangzhan Jin and Sitao Luan and Tianwei Ni and Sicheng Lyu and Guillaume Rabusseau and Reihaneh Rabbany and Doina Precup and Mohammad Hamdaqa},
  journal= {arXiv preprint arXiv:2509.12235},
  year   = {2026}
}

Comments

31 pages, 22 figures

R2 v1 2026-07-01T05:37:29.587Z