Why Multi-Step Tool-Use Reinforcement Learning Collapses and How Supervisory Signals Fix It
Summary
This paper investigates why multi-step tool-use reinforcement learning leads to catastrophic collapse in LLMs and how interleaving supervised fine-tuning with diverse supervisory signals can stabilize training.
View Cached Full Text
Cached at: 06/26/26, 06:05 AM
Paper page - Why Multi-Step Tool-Use Reinforcement Learning Collapses and How Supervisory Signals Fix It
Source: https://huggingface.co/papers/2606.26027
Abstract
Research investigates how different supervisory signals and training strategies improve the stability and performance of large language models in tool-use tasks, addressing issues like catastrophic collapse and format sensitivity through interleaved supervised fine-tuning and reinforcement learning.
Tool use enables large language models (LLMs) to perform complex tasks, and recentagentic reinforcement learning(RL) methods show promise for enhancing model capabilities. However, RL alone often leads to instability or limited gains intool-use tasks. In our experiments, some models exhibitcatastrophic collapse, where performance abruptly drops and tool-invocation structures fail. The analysis reveals that these failures stem from unexpected probability spikes in specificcontrol 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, includingoff-policy supervision,hint-based guidance,erroneous example supervision, and others, applied under both synchronous andinterleaved trainingschemes. We find that interleavingsupervised 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 oflearning ratesand generalization across settings. These results highlight the importance of understanding RL failures and demonstrate how diverse supervisory signals can guideexploratory learning, enabling robust training of LLMs for complex, multi-steptool-use tasks. Our Code is available at https://github.com/hypasd-art/Tool-RL-Box.
View arXiv pageView PDFGitHub0Add to collection
Get this paper in your agent:
hf papers read 2606\.26027
Don’t have the latest CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash
Models citing this paper0
No model linking this paper
Cite arxiv.org/abs/2606.26027 in a model README.md to link it from this page.
Datasets citing this paper0
No dataset linking this paper
Cite arxiv.org/abs/2606.26027 in a dataset README.md to link it from this page.
Spaces citing this paper0
No Space linking this paper
Cite arxiv.org/abs/2606.26027 in a Space README.md to link it from this page.
Collections including this paper0
No Collection including this paper
Add this paper to acollectionto link it from this page.
Similar Articles
Why Multi-Step Tool-Use Reinforcement Learning Collapses and How Supervisory Signals Fix It
This paper investigates why multi-step tool-use reinforcement learning (RL) often collapses or yields limited gains, identifying probability spikes in control tokens as a key cause. It shows that interleaving supervised fine-tuning with RL improves stability and explores various supervisory signals to guide robust training.
A Few Neurons Reveal When LLMs Misuse Tools: Sparse Detection and Selective Steering for Reliable Tool Use
This paper introduces PRISMS, a framework that uses a small set of failure-specific MLP neurons to detect and steer LLM tool-use errors (over-calling, missing calls, invalid arguments) with sparse readouts, improving reliability across multiple model families.
Mechanistic origins of catastrophic forgetting: why RL preserves circuits better than SFT?
This paper investigates the mechanistic origins of catastrophic forgetting in LLMs, finding that reinforcement learning preserves internal computational circuits better than supervised fine-tuning, resulting in less forgetting of prior capabilities.
Diagnosing Training Inference Mismatch in LLM Reinforcement Learning
This paper diagnoses Training-Inference Mismatch (TIM) in LLM reinforcement learning, showing that small numerical disagreements between training and inference token probabilities can cause training collapse, and proposes remedies.
Are Tools Always Beneficial? Learning to Invoke Tools Adaptively for Dual-Mode Multimodal LLM Reasoning
Introduces AutoTool, a model that adaptively decides whether to invoke tools for multimodal LLM reasoning, achieving significant accuracy and efficiency gains through reinforcement learning and dual-mode reasoning.