arXiv AI

Agents, Think Only When it Helps

The paper investigates the role of thought supervision in distilling large language model agents. By independently perturbing teacher thoughts and actions during distillation, the authors find that removing or altering thoughts has little impact on student performance across various benchmarks and model scales. However, thought supervision becomes important when the action does not reveal the information contained in the thought, suggesting that including thoughts in the training data may be unnecessary for many cases.

arXiv AI
Oct 26

Beyond On-Policy vs. Off-Policy: At What Level Should LLM Agent Distillation Follow the Student?

The paper proposes a hierarchical framework for distilling large language model agents, distinguishing between the policy that generates interaction histories and the policy that produces token prefixes for actions. It explores various combinations of teacher-generated and student-generated data across multiple models and environments, finding that pairing student-induced histories with teacher-generated action prefixes yields the strongest training signal, while fully on-policy data is only beneficial when mixed with other configurations. The study also shows that the specific pairing of teacher and student proportions at each level significantly impacts performance, and that strategically front-loading teacher-generated contexts and delaying on-policy contexts improves results.

By Ya Gao, Zhengtao Zou, Pekka Marttinen
arXiv AI
Sep 30

Distilling Agentic Systems: A Roadmap across Models, Artifacts, and Harnesses

The paper introduces Agent Distillation, a framework for transferring task‑solving knowledge from a teacher agent to a student agent. It categorizes where this knowledge is retained—within the model, as artifacts, through the execution harness, or across substrates—distinguishing transfer evidence from outcomes. An evaluation framework is proposed to link retention to causal contribution and practical utility, aiming to support reliable, maintainable, and safe agent development.

By Ziluowen Luo, Senzhang Wang, Chaozhuo Li, Jun Yin, Hao Yan, Ming Cheng, Chenxu Wang, Songyang Liu, Litian Zhang, Qiwei Ye, Zheng Liu, Philip S. Yu
arXiv AI
Oct 5

Harness-Aware Distillation for Small Language Model Agents

The paper introduces Harness-Aware Distillation (HAD), a method for training smaller language model agents that preserves the surrounding harness—software managing context, tools, and feedback—while focusing distillation on the teacher’s contributions beyond the harness. HAD combines an action preference that contrasts teacher actions with and without harness information, and a validity check that filters out contradictory preference pairs. Experiments on long-horizon agent benchmarks show that HAD outperforms standard on‑policy distillation, reducing unproductive loops and improving error recovery without requiring task rewards or future information.

By Moonseok Choi, Taehong Moon, Giung Nam, Juho Lee
arXiv Machine Learning
Aug 4

Look Ahead Before You Distill: Future Trajectory Validation of Teacher Guidance for Agentic On-Policy Distillation

arXiv:2608. 01953v1 Announce Type: cross Abstract: On-policy distillation (OPD) provides teacher supervision on states visited by the student, reducing the distribution gap between training and inference.

By Chishui Chen, Yaoyou Fan, Te Sun, Yi Yang, Chenghao Sun, Delin Mao, Hongbo Qiao, Zuowei Zhang, Junxi Wang, Chenxing Sun, Yangen Hu, Lu Pan, Xuyang Liu, Linfeng Zhang
arXiv Computation and Language
2d ago

Beyond Self-Reflection: An Execute-Distill-Verify Framework for Agentic Experience Learning

The paper introduces EDV, an Execute‑Distill‑Verify framework that constructs offline experience for large language model agents without updating model parameters. In EDV, multiple heterogeneous agents execute tasks, a distiller extracts candidate lessons from their trajectories and feedback, and executors review these lessons, keeping only those unanimously approved. Experiments on τ²‑bench, Mind2Web, and MMTB demonstrate improved performance on revisited tasks and within‑domain transfer, while a Retail subset evaluation shows EDV cuts the Self‑Confirmation Trap Rate from 11.0% to 5.0%.

By Shiding Zhu, Yudi Qi, Yajie Wang, Jiaze Li, Chao Song, Yaorui Shi, Yibo Miao, Hanqi Gao, Kai Zhang
arXiv Machine Learning
Oct 8

Train4Merge: A Controlled Single-Teacher Study of RL vs. SFT Teachers for OPD-Based Model Merging

The study compares supervised fine-tuning (SFT) and reinforcement learning (RL) as teacher training methods for on-policy distillation (OPD) in a single-teacher setting. Using Qwen3.5-9B models across Agentic, Reasoning, and Perception domains, RL-trained teachers consistently produce stronger students and better recovery of performance gains than SFT-trained teachers. The authors attribute this advantage to RL teachers’ smaller parameter shifts from the shared initialization, which facilitates learning through OPD.

By Jingyuan Huang, Zuming Huang, Yucheng Shi, Zhongzhi Li, Xiaoming Zhai, Wei Chu, Ninghao Liu
arXiv Computation and Language
Oct 8

Learning to Act with Task Progress: Distilling Small Agents from Compact Teacher Supervision

The paper introduces Task-Progress Distillation (TPD), an offline method that pairs each action in teacher demonstrations with a short label indicating the current task stage. A small 1.7B student model trained with 404 demonstrations using TPD achieves 72.4% mean unseen task success on ALFWorld, matching action-only supervision and outperforming a reasoning-trained student. At lower demonstration budgets, explicit stage labels significantly boost performance, raising success from 48.0% to 67.7% with 200 demos, and both approaches converge to 76.9% with 808 demos.

By Wenxi Gan
Hugging Face Trending Papers
Sep 24

From Self-Distillation to Self-Practice: Privileged Information for Multi-Turn Agents

The paper examines on‑policy self‑distillation (OPSD) for multi‑turn agents, showing that using privileged information (PI) in the loss can make agents appear confident yet underperform plain RL, sometimes worse than the untrained base model. To address this, the authors propose Privileged Self‑Practice (PSP), which keeps PI in the prompt and uses it only during sampling, not in the loss. PSP consistently outperforms plain GRPO across AppWorld and SWE‑bench Verified, improving task‑goal completion by up to 65% and resolved rate by up to 61%.