mlsanityData to 14 Oct 2026

Post-Training is About States, Not Tokens: A State Distribution View of SFT, RL, and On-Policy Distillation

arXiv 2605.22731 ↗First posted 21 May 2026cs.LG cs.AIcited by 3

On the map: Knowledge distillation → On-policy and self-distillation for LLMs

Abstract (arXiv)

Large language model post-training methods such as supervised fine-tuning (SFT), reinforcement learning (RL), and distillation are often analyzed through their loss functions: maximum likelihood, policy gradients, forward KL, reverse KL, or related objective-level variants. We study a complementary factor: the state distribution on which supervision is applied. For an autoregressive policy, a state is a prompt plus generated prefix. SFT trains on fixed dataset states, while RL and on-policy distillation (OPD) train on states induced by the current learner. We formalize post-training as state-distribution shaping and run a controlled smallscale study using Qwen3-0.6B-Base on GSM8K, with TruthfulQA and MMLU as retention evaluations. Our results show three phenomena. First, a mild SFT run improves GSM8K with little forgetting, while a stress SFT run causes substantial retention loss. Second, OPD from a degraded SFT teacher surpasses that teacher on GSM8K, TruthfulQA, and MMLU, despite using the teacher as its only supervision source. Third, a lightweight on-policy RL run improves GSM8K while preserving retention. These results support a state-centric view of post-training: the source and locality of training states can be as important as the form of the supervision signal.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 19 of its 27 references that are in the map. First posted on arXiv 21 May 2026.

Builds on

19 of 27 references.

References · 19

  1. 2022Training language models to follow instructions with human feedbackcited by 15,971
  2. 2017Proximal Policy Optimization Algorithmscited by 15,215
  3. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  4. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  5. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  6. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  7. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  8. 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
  9. 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
  10. 2017Deep reinforcement learning from human preferencescited by 3,912
  11. 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,975
  12. 2020Learning to summarize from human feedbackcited by 2,588
7 more
  1. 2019Fine-Tuning Language Models from Human Preferencescited by 2,150
  2. 2017Gradient Episodic Memory for Continual Learningcited by 1,898
  3. 2020Knowledge Distillation: A Surveycited by 1,603
  4. 2023A General Theoretical Paradigm to Understand Learning from Human Preferencescited by 912
  5. 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 803
  6. 2016Sequence-Level Knowledge Distillationcited by 801
  7. 2015Sequence Level Training with Recurrent Neural Networkscited by 576

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Demystifying Reinforcement Learning Post-Training of Language Models
  2. 2026Supervised Fine-Tuning versus Reinforcement Learning: A Study of Post-Training Methods for Large Language Models
  3. 2025Towards a Unified View of Large Language Model Post-Training
  4. 2026Large Language Model Post-Training: A Unified View of Off-Policy and On-Policy Learning
  5. 2026Distilled Reinforcement Learning for LLM Post-training
  6. 2025Retaining by Doing: The Role of On-Policy Data in Mitigating Forgetting
  7. 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes
  8. 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes