How and Why LLMs Generalize: A Fine-Grained Analysis of LLM Reasoning from Cognitive Behaviors to Low-Level Patterns
Haoyue Bai, Yiyou Sun, Wenjie Hu, Shi Qiu, Maggie Ziyu Huan, Peiyang Song, Robert Nowak, Dawn Song
TL;DR
The paper tackles why supervised fine-tuning and reinforcement-learning tuning yield different generalization in large language models by decomposing reasoning into five atomic cognitive skills: calculation, enumeration, simulation, fact retrieval, and diagnostic checking. It introduces a three-stage benchmark and a behavior–domain grid to measure these skills across math, science, coding, and non-reasoning domains, linking behavioral profiles to low-level statistical patterns. Across experiments with Qwen-derived models, RL consistently yields balanced skill distributions and better cross-domain transfer, while SFT induces over-specialization and surface-pattern reliance; long CoT can partially mitigate cross-domain gaps. The findings advocate for training strategies that reinforce core skills and monitor representations, guiding future design of robust, interpretable, and transferable reasoning in LLMs.
Abstract
Large Language Models (LLMs) display strikingly different generalization behaviors: supervised fine-tuning (SFT) often narrows capability, whereas reinforcement-learning (RL) tuning tends to preserve it. The reasons behind this divergence remain unclear, as prior studies have largely relied on coarse accuracy metrics. We address this gap by introducing a novel benchmark that decomposes reasoning into atomic core skills such as calculation, fact retrieval, simulation, enumeration, and diagnostic, providing a concrete framework for addressing the fundamental question of what constitutes reasoning in LLMs. By isolating and measuring these core skills, the benchmark offers a more granular view of how specific cognitive abilities emerge, transfer, and sometimes collapse during post-training. Combined with analyses of low-level statistical patterns such as distributional divergence and parameter statistics, it enables a fine-grained study of how generalization evolves under SFT and RL across mathematical, scientific reasoning, and non-reasoning tasks. Our meta-probing framework tracks model behavior at different training stages and reveals that RL-tuned models maintain more stable behavioral profiles and resist collapse in reasoning skills, whereas SFT models exhibit sharper drift and overfit to surface patterns. This work provides new insights into the nature of reasoning in LLMs and points toward principles for designing training strategies that foster broad, robust generalization.
