From Correction to Mastery: Reinforced Distillation of Large Language Model Agents

Yuanjie Lyu, Chengyu Wang, Jun Huang, Tong Xu
9/12/2025
cs.CLcs.AI

No official code found. Best implementation: haruhi-sudo/SCoRe (3 stars, last commit 2026-01-29).

Abstract

Large Language Model agents excel at solving complex tasks through iterative reasoning and tool use, but typically depend on ultra-large, costly backbones. Existing distillation approaches train smaller students to imitate full teacher trajectories, yet reasoning and knowledge gaps between the teacher and student can cause compounding errors. We propose SCoRe, a student-centered framework in which the student generates training trajectories and the teacher corrects only the earliest error, producing training data matched to the student's ability and exposing specific weaknesses. The student is first fine-tuned on corrected trajectories. Subsequently, short-horizon reinforcement learning starts from the verified prefix preceding the earliest error, with target rewards assigned at that step. This design encourages autonomous problem-solving beyond imitation and enhances training stability. On 12 challenging benchmarks, a 7B-parameter student distilled with SCoRe matches the agentic performance of a 72B-parameter teacher.

View on arXivView PDF

Code Implementations(1)

haruhi-sudo/SCoReCommunity59%

FROM CORRECTION TO MASTERY: REINFORCED DISTILLATION OF LARGE LANGUAGE MODEL AGENTS

30Shell, PythonSep 15, 20258 months ago

Similar Papers

Cite this paper

@article{lyu2025from,
  title  = {From Correction to Mastery: Reinforced Distillation of Large Language Model Agents},
  author = {Yuanjie Lyu and Chengyu Wang and Jun Huang and Tong Xu},
  year   = {2025},
  eprint = {2509.14257},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2509.14257v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2509.14257.svg)](https://codeofpaper.com/paper/2509.14257)

Discussion