Agent-Reflex trains an 8B agent to recover when its MCP tools fail, and it gets derailed half as often

1 month ago

Agent-Reflex trains Qwen3-8B to notice and recover when MCP tools fail or a user corrects it. On AgentNoiseBench it is thrown off course in 28.5% of runs against 56.1% for the base model, and its mean across five tool-use benchmarks rises from 40.2 to 58.6. It still trails GPT-5.2 and Claude 4.5 Sonnet. From Renmin University of China and the National University of Singapore (arXiv 2610.17953v1, submitted 13 Oct 2026). Paper: https://arxiv.org/abs/2610.17953 Code: https://github.com/RUC-NLPIR/Agent-Reflex Model: https://huggingface.co/dongguanting/Agent-Reflex-8B Data: https://huggingface.co/datasets/dongguanting/Agent-Reflex-SFT-54K

Ask about this presentation

Chapters

  1. 0:00Agent-Reflex
  2. 0:14Tool calls fail often
  3. 0:29Frontier models drop
  4. 0:45Failures become rules
  5. 1:03One recovery rule
  6. 1:18Train on its own recoveries
  7. 1:36Derailed half as often
  8. 1:50Results that look valid
  9. 2:05Five tool-use benchmarks
  10. 2:22Where it stops
  11. 2:39Model, data and code
Show transcript
  1. Agent-Reflex teaches a small open model to notice when an MCP tool fails or a user corrects it, and to recover. It comes from Renmin University of China and the National University of Singapore.

    Slide text
    AI Papers

    Agent-Reflex trains an 8B agent to recover when its MCP tools fail, and it gets derailed half as often

    Guanting Dong, Xinyu Yang, Jiajie Jin and colleagues, with Zhicheng Dou · Renmin University of China, National University of Singapore

    Dong et al. · Agent-Reflex · arXiv 2610.17953v1 · submitted 13 Oct 2026 · Appendix E.1
  2. Real tools fail all the time. Replaying agents on nine public benchmarks, about one tool result in seven was an error. A third were services that were down, followed by permission failures and timeouts.

    Slide text
    01 02 03 The problem

    About one tool result in seven came back as an error

    Share of tool errors by family: service or gateway unavailable 33.6%, auth or permission 17.9%, timeout 14.7%, not found 10.5%, validation 10.2%, execution exception 8.3%, rate limit 4.6%, network 0.3%

    24,510 errors in 177,700 tool returns · agents replayed on 9 public benchmarks

    Dong et al. · Agent-Reflex · arXiv 2610.17953v1 · submitted 13 Oct 2026 · §3.1, Appendix D.2 · Figure 1d, Table 4
  3. Frontier models stumble too. On the airline, retail and telecom tasks of τ²-Bench, GPT-5.2 drops from about 80 to about 60 percent success once tool results turn faulty. Claude 4.5 Sonnet loses about 16 points.

    Slide text
    01 02 03 The problem

    GPT-5.2 and Claude 4.5 Sonnet finish fewer tasks when tool results are faulty

    GPT-5.2 81 → 61
    Claude-4.5-Sonnet 73 → 57

    Success rate, %, clean → tool noise · AgentNoiseBench on τ²-Bench airline, retail, telecom

    Dong et al. · Agent-Reflex · arXiv 2610.17953v1 · submitted 13 Oct 2026 · Table 2
  4. The recipe builds 557 simulated services from popular MCP servers and injects real error types into one tool result in five. Nine models attempt each task, and the difference between runs that recovered and runs that failed becomes a written rule.

    Slide text
    01 02 03 The recipe

    557 simulated MCP services, and 1,170 recovery rules mined from their failures

    Stage 1 pipeline: MCP servers, environment synthesis, noise injection, noisy environments, verifiable tasks, rollouts from an LLM agent pool, reflection meta-skills

    Failures injected into 1 tool result in 5 · 9 models from 4B to 284B attempt each task

    Dong et al. · Agent-Reflex · arXiv 2610.17953v1 · submitted 13 Oct 2026 · §3.1, Appendix D.2 · Figure 2, Table 5
  5. Here is one rule, for a service modeled on Airtable. Failing agents acted on a search result that only looked right. The rule says to confirm every identifier, write only on an exact match, then read the record back.

    Slide text
    01 02 03 The recipe

    One rule from the Airtable environment

    Bad pattern Acts on a similar search hit without confirming the base and table
    Rule Resolve base, table and record IDs in order, update only on an exact match, then re-read the record
    Dong et al. · Agent-Reflex · arXiv 2610.17953v1 · submitted 13 Oct 2026 · Appendix E.3
  6. Qwen3-8B fine-tunes on about 54,000 runs where it fixed its own mistakes, with the rule text stripped out and some user corrections wrong on purpose. Reinforcement learning follows, with a quarter of runs hitting failures. The finished model carries no rule library.

    Slide text
    01 02 03 The recipe

    Qwen3-8B practices recovering, then ships without the rules

    54K recovery runs its own verified fixes, with the rule text removed; some user corrections are deliberately wrong
    Noisy RL a quarter of practice runs hit injected failures; a copy that sees the rule scores each step
    Dong et al. · Agent-Reflex · arXiv 2610.17953v1 · submitted 13 Oct 2026 · §3.2, §3.3, Appendix D.3, D.4
  7. AgentNoiseBench injects its faults differently from the training setup. There, noise throws the base Qwen model off course in more than half its runs. Agent-Reflex goes off course in just over a quarter.

    Slide text
    01 02 03 What it found

    Noise derails Agent-Reflex half as often as its base model

    Qwen3-8B
    56.1%
    EnvScaler-8B
    44.8%
    Agent-Reflex-8B
    28.5%

    Share of runs thrown off course · AgentNoiseBench, tool and user noise · injected differently from training

    Dong et al. · Agent-Reflex · arXiv 2610.17953v1 · submitted 13 Oct 2026 · Appendix E.1, E.2
  8. The biggest tool-side gain is on results that pass every format check but point the wrong way or leave data out. The agent only catches those by reading the content. It beats the base model on all thirteen kinds of noise.

    Slide text
    01 02 03 What it found

    The largest tool-side lead comes on results that look valid but mislead

    Misleading lure a well-formed result that points the wrong way
    Incomplete information a well-formed result with data missing

    Ahead of Qwen3-8B on all 13 noise types · 165 tasks per type

    Dong et al. · Agent-Reflex · arXiv 2610.17953v1 · submitted 13 Oct 2026 · Appendix E.1 · Figure 5
  9. The recipe also helps when tools behave. Averaged over five tool-use benchmarks, Qwen3-8B goes from about 40 to almost 60, ahead of the other 8B training recipes. On τ²-Bench its score more than doubles.

    Slide text
    01 02 03 What it found

    On five tool-use benchmarks, Qwen3-8B rises from 40 to 59

    Qwen3-8B
    40.2
    EnvScaler-8B
    50.9
    Agent-World-8B
    53.6
    Agent-Reflex-8B
    58.6

    Mean of VitaBench, τ²-Bench, BFCL V4, ACEBench, ToolSandbox · τ²-Bench alone: 26.2 → 62.1

    Dong et al. · Agent-Reflex · arXiv 2610.17953v1 · submitted 13 Oct 2026 · Table 1 · mean of 3 runs
  10. It is still an 8B model. GPT-5.2 and Claude 4.5 Sonnet score higher on every test here. The rules come from one fixed snapshot of services, so changed tools need new mining, and the recipe has only been tried at 8B.

    Slide text
    01 02 03 The limits

    Still an 8B model, and still behind GPT-5.2 and Claude

    Behind frontier models 58.6 vs 64.4 and 72.6 on Table 1 · 29.1 vs 55.8 and 51.0 on AgentNoiseBench
    Rules mined once a fixed snapshot; changed tool interfaces need re-mining
    Tested at 8B only other model sizes untested
    Rebuilt baselines 4 competing recipes reproduced locally · no spread shown
    Dong et al. · Agent-Reflex · arXiv 2610.17953v1 · submitted 13 Oct 2026 · Tables 1, 2 · Appendices C, D.5, F, G
  11. The checkpoint, both training sets, the rule library and the code are public under the MIT license. Retraining it takes sixteen 80-gigabyte GPUs for fine-tuning and eight for reinforcement learning.

    Slide text
    01 02 03 The limits

    The Agent-Reflex model, training data and code are public

    Model huggingface.co/dongguanting/Agent-Reflex-8B
    Data Agent-Reflex-SFT-54K · Agent-Reflex-RL-2K
    Code github.com/RUC-NLPIR/Agent-Reflex · MIT

    Training: 16 × 80 GB GPUs for fine-tuning, 8 for RL

    Dong et al. · Agent-Reflex · arXiv 2610.17953v1 · submitted 13 Oct 2026 · Appendix D.2 · repository README

What happened in AI today, in your inbox

One short email each evening. Read it and you’re caught up.

Built with ❤️ in New York