English

Robust Tool Use via Fission-GRPO: Learning to Recover from Execution Errors

Machine Learning 2026-04-21 v2 Artificial Intelligence

Abstract

Large language models (LLMs) can call tools effectively, yet they remain brittle in multi-turn execution: after a tool-call error, smaller models often fall into repetitive invalid re-invocations instead of interpreting the feedback and recovering. This failure mode persists because current training paradigms do not explicitly teach models how to recover from execution errors. In particular, standard reinforcement learning (RL) collapses rich failure experience into sparse negative rewards, while pre-collected error-correction datasets become mismatched to the policy's evolving failure modes. To bridge this gap, we propose Fission-GRPO, a framework that converts execution errors into on-policy corrective supervision within the RL training loop. Our core mechanism fissions each failed trajectory into a new training instance by augmenting it with diagnostic feedback from a fine-tuned Error Simulator, then resampling multiple recovery rollouts on-policy. This enables the model to learn from the precise errors it makes during exploration, rather than from static, pre-collected error cases. On BFCL v4 Multi-Turn, Fission-GRPO improves the error recovery rate of Qwen3-8B by 5.7% absolute and overall accuracy by 4.0% (from 42.75% to 46.75%), outperforming both RL baselines and specialized tool-use agents. The method further generalizes to TAU-Bench and TAU2-Bench, achieving leading results across most settings with gains up to +17.4%.

Keywords

Cite

@article{arxiv.2601.15625,
  title  = {Robust Tool Use via Fission-GRPO: Learning to Recover from Execution Errors},
  author = {Zhiwei Zhang and Fei Zhao and Rui Wang and Zezhong Wang and Bin Liang and Jiakang Wang and Yao Hu and Shaosheng Cao and Kam-Fai Wong},
  journal= {arXiv preprint arXiv:2601.15625},
  year   = {2026}
}

Comments

9 pages, 4 figures, 4 tables. Accepted to ACL 2026 Main Conference

R2 v1 2026-07-01T09:15:12.339Z