Research map / Large language models / Agents, code and math
Reducing Tool Hallucination via Reliability Alignment
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Defines two types of tool hallucination (selection and usage), builds RelyToolBench with missing-parameter and unmatched-tool subsets, and proposes Relign, which adds indecisive actions (ChangeTools, TalkToUser) and trains with SFT then DPO to reduce tool hallucinations.
Problem
LLM agents often hallucinate tool use: they select inappropriate tools or misuse them (fabricated or missing parameters), causing erroneous task execution, extra cost, and unreliable outputs. Existing tool-learning work mainly optimizes decisive tool calls and lacks a way to defer, clarify, or switc
Approach
Expand the tool-use action space with indecisive actions (ChangeTools, TalkToUser, give up) so the model can defer or clarify when tool preconditions are unmet, and align it with SFT on synthesized boundary cases followed by DPO on preference pairs ranking correct > indecisive > hallucinated actions.
Key results
- Relign on Llama3.1-8B raises overall RePR to 77.2 from 65.3 baseline and cuts tool hallucination rate from 50.8% to 14.6%.
- Relign on ToolLLaMA lowers overall tool hallucination from 56.5% to 18.3% and raises RePR from 62.5 to 69.1.
- On APIBench OOD with BM25 retrieval, Relign-trained Llama3.1 reduces HuggingFace hallucination from 9.7 to 6.4 and raises AST from 14.3 to 15.7.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| add Relign reliability alignment | Toolllama 7B · RelyToolBench Overall | RePR: 62.5 -> 69.1 | Helped |
| add Relign reliability alignment | Toolllama 7B · RelyToolBench Overall | Tool Hallu: 56.5 -> 1\textbf{8.3} | Helped |
| add Relign reliability alignment | Llama3.1 8B · RelyToolBench Overall | RePR: 65.3 -> 77.2 | Helped |
| add Relign reliability alignment | Llama3.1 8B · RelyToolBench Overall | Tool Hallu: 50.8 -> 14.6 | Helped |
| add Relign reliability alignment | Qwen2.5 7B · RelyToolBench Overall | RePR: 65.1 -> 73.0 | Helped |
| add Relign reliability alignment | Qwen2.5 7B · RelyToolBench Overall | Tool Hallu: 49.2 -> 17.6 | Helped |
| use SFT without DPO instead of Relign (ablation) | Toolllama 7B · RelyToolBench Overall | Tool Hallu: 29.0 vs 1\textbf{8.3} | Hurt |
| use DPO without SFT instead of Relign (ablation) | Toolllama 7B · RelyToolBench Overall | Tool Hallu: 41.1 vs 1\textbf{8.3} | Hurt |
| use SFT and DPO joint training instead of Relign (ablation) | Llama3.1 8B · RelyToolBench Overall | RePR: 73.7 vs 77.2 | Hurt |
| use DPO without SFT instead of Relign (ablation) | Llama3.1 8B · RelyToolBench Overall | Tool Hallu: 46.1 vs 14.6 | Hurt |
| use SFT and DPO joint training instead of Relign (ablation) | Qwen2.5 7B · RelyToolBench Overall | RePR: 66.1 vs 73.0 | Hurt |
| increase training data from 1k to 5k (ablation) | Llama3.1 · RelyToolBench | RePR | Helped |
Surprising findings
- StepTool improves pass rate but does not reduce tool hallucination rate or tool call count.
- DPO alone increases hallucination relative to SFT in several settings, while SFT alone gives the largest hallucination reduction.
Limitations
- Hallucination evaluation relies on GPT-4o as judge; human check reports 92.7% accuracy on 150 cases.
- RelyToolBench is derived from StableToolBench and its unmatched-tool and missing-parameter subsets are synthetically constructed.
- Timing hallucinations show smaller improvement than other types.
Experimental setup
Datasets: ToolBench StableToolBench
Benchmarks: RelyToolBench APIBench
Metrics: RePR (reliable pass rate) tool hallucination rate average tool call number benefit-cost utility AST accuracy
Baselines: gpt-3.5-turbo GPT-4o-mini GPT-4o ToolLLaMA + RLHF StepTool SFT DPO SFT&DPO
Builds on: ToolBench StableToolBench StepTool Direct Preference Optimization
Closest papers by meaning
- CodeHalu: Investigating Code Hallucinations in LLMs via Execution-based Verification (AAAI 2025 · similarity 0.87)
- Exposing and Evaluating Hallucinations for GUI Grounding (CVPR 2026 · similarity 0.87)
- MCP-Atlas: A Large-Scale Benchmark for Tool-Use Competency with Real MCP Servers (NeurIPS 2026 · similarity 0.86)
- MLLMs Fail to Refuse when Using Tools Agentically (NeurIPS 2026 · similarity 0.86)
- Learning Evolving Tools for Large Language Models (ICLR 2025 · similarity 0.86)
- Beyond Accuracy: A Cognitive Load Framework for Mapping the Capability Boundaries of Tool-use Agents (AAAI 2026 · similarity 0.85)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
