ArXiv

Why Multi-Step Tool-Use Reinforcement Learning Collapses and How Supervisory Signals Fix It

Authors
Yupu Hao, Zhuoran Jin, Huanxuan Liao...
Categories
cs.CL, cs.LG
arXiv
https://arxiv.org/abs/2606.26027v1
PDF
https://arxiv.org/pdf/2606.26027v1

Brief

The paper investigates why reinforcement learning for multi-step tool use in LLMs can collapse and how supervisory signals prevent it. Failures arise from probability spikes on control tokens that break execution structure while preserving latent tool capability. The authors test off-policy, hint-based, and erroneous-example supervision under synchronous and interleaved schemes, finding interleaved SFT+RL stabilizes training but harms OOD format/content generalization; learning-rate and generalization analyses are provided. Full code is released.

Why it matters

Agentic RL applied to multi-step tool-use with LLMs can experience catastrophic collapse: model performance abruptly drops and tool-invocation structures fail due to unexpected probability spikes on specific control tokens, even though underlying tool-use capability remains intact but format-obscured.

Key details

  • Interleaving supervised fine-tuning (SFT) with RL using diverse supervisory signals (off-policy supervision, hint-based guidance, erroneous-example supervision, etc.) substantially improves training stability; however, interleaved SFT+RL degrades format- and content-out-of-distribution generalization. The paper also analyzes learning-rate effects and cross-setting generalization. Code: https://github.com/hypasd-art/Tool-RL-Box (arXiv 2026-06-24).
Source evidence

Abstract

Tool use enables large language models (LLMs) to perform complex tasks, and recent agentic reinforcement learning (RL) methods show promise for enhancing model capabilities. However, RL alone often leads to instability or limited gains in tool-use tasks. In our experiments, some models exhibit catastrophic collapse, where performance abruptly drops and tool-invocation structures fail. The analysis reveals that these failures stem from unexpected probability spikes in specific control tokens, disrupting structured execution, yet the underlying tool-use capability remains intact, merely obscured by specific formats. To address this, we systematically investigate a diverse set of supervisory signals, including off-policy supervision, hint-based guidance, erroneous example supervision, and others, applied under both synchronous and interleaved training schemes. We find that interleaving supervised fine-tuning (SFT) with RL substantially improves stability, but exhibits degraded performance under format and content out-of-distribution (OOD) evaluation. We also analyze the impact of learning rates and generalization across settings. These results highlight the importance of understanding RL failures and demonstrate how diverse supervisory signals can guide exploratory learning, enabling robust training of LLMs for complex, multi-step tool-use tasks. Our Code is available at https://github.com/hypasd-art/Tool-RL-Box.