flâneur

Why Does Reinforcement Learning Generalize? A Feature-Level Mechanistic Study of Post-Training in Large Language Models

arxiv.org · 8,059 words · saved by 1 readers

Reinforcement learning (RL)-based post-training often improves the reasoning performance of large language models (LLMs) beyond the training domain, while supervised fine-tuning (SFT) frequently leads to general capabilities forgetting. However, the mechanisms underlying this contrast remain unclear. To bridge this gap, we present a feature-level mechanistic analysis methodology to probe RL generalization using a controlled experimental setup, where RL- and SFT-tuned models are trained from the same base model on identical data. Leveraging our interpretability framework, we align internal activations across models within a shared feature space and analyze how features evolve during post-training. We find that SFT rapidly introduces many highly specialized features that stabilize early in training, whereas RL induces more restrained and continually evolving feature changes that largely preserve base models’ representations. Focusing on samples where RL succeeds but the base model fails,

Dan Shi Affiliation: TJUNLP Lab, School of Computer Science and Technology, Tianjin University, China Email: shidan@tju.edu.cn Zhuowen Han Affiliation: TJUNLP Lab, School of Computer Science and Technology, Tianjin University, China Email: dyxiong@tju.edu.cn Simon Ostermann Affiliation: German Research Center for Artificial Intelligence (DFKI), Saarbrücken, Germany Affiliation: Saarland University, Saarbrücken, Germany Renren Jin Josef van Genabith Affiliation: German Research Center for Artificial Intelligence (DFKI), Saarbrücken, Germany Affiliation: Saarland University,…

saved by

related reading