English

Supervised Reinforcement Learning: From Expert Trajectories to Step-wise Reasoning

Computation and Language 2026-03-02 v3 Artificial Intelligence Machine Learning

Abstract

Large Language Models (LLMs) often struggle with problems that require multi-step reasoning. For small-scale open-source models, Reinforcement Learning with Verifiable Rewards (RLVR) fails when correct solutions are rarely sampled even after many attempts, while Supervised Fine-Tuning (SFT) tends to overfit long demonstrations through rigid token-by-token imitation. To address this gap, we propose Supervised Reinforcement Learning (SRL), a framework that reformulates problem solving as generating a sequence of logical "actions". SRL trains the model to generate an internal reasoning monologue before committing to each action. It provides smoother rewards based on the similarity between the model's actions and expert actions extracted from the SFT dataset in a step-wise manner. This supervision offers richer learning signals even when all rollouts are incorrect, while encouraging flexible reasoning guided by expert demonstrations. As a result, SRL enables small models to learn challenging problems previously unlearnable by SFT or RLVR. Moreover, initializing training with SRL before refining with RLVR yields the strongest overall performance. Beyond reasoning benchmarks, SRL generalizes effectively to agentic software engineering tasks, establishing it as a robust and versatile training framework for reasoning-oriented LLMs.

Keywords

Cite

@article{arxiv.2510.25992,
  title  = {Supervised Reinforcement Learning: From Expert Trajectories to Step-wise Reasoning},
  author = {Yihe Deng and I-Hung Hsu and Jun Yan and Zifeng Wang and Rujun Han and Gufeng Zhang and Yanfei Chen and Wei Wang and Tomas Pfister and Chen-Yu Lee},
  journal= {arXiv preprint arXiv:2510.25992},
  year   = {2026}
}

Comments

Paper accepted by ICLR 2026. The first two authors contribute equally

R2 v1 2026-07-01T07:12:55.442Z