Search papers, labs, and topics across Lattice.
This paper introduces Doctor-RAG (DR-RAG), a framework for diagnosing and repairing failures in Agentic RAG systems by localizing errors within reasoning trajectories using a coverage-gated taxonomy. DR-RAG then performs tool-conditioned local repair at the diagnosed failure point, reusing validated reasoning prefixes and retrieved evidence to minimize computational cost. Experiments on multi-hop QA benchmarks show that DR-RAG improves answer accuracy and reduces token consumption compared to full-pipeline reruns.
Agentic RAG gets a check-up: Doctor-RAG surgically repairs failures in reasoning trajectories, boosting accuracy while slashing token consumption compared to brute-force reruns.
Agentic Retrieval-Augmented Generation (Agentic RAG) has become a widely adopted paradigm for multi-hop question answering and complex knowledge reasoning, where retrieval and reasoning are interleaved at inference time. As reasoning trajectories grow longer, failures become increasingly common. Existing approaches typically address such failures by either stopping at diagnostic analysis or rerunning the entire retrieval-reasoning pipeline, which leads to substantial computational overhead and redundant reasoning. In this paper, we propose Doctor-RAG (DR-RAG), a unified diagnose-and-repair framework that corrects failures in Agentic RAG through explicit error localization and prefix reuse, enabling minimal-cost intervention. DR-RAG decomposes failure handling into two consecutive stages: (i) trajectory-level failure diagnosis and localization, which attributes errors to a coverage-gated taxonomy and identifies the earliest failure point in the reasoning trajectory; and (ii) tool-conditioned local repair, which intervenes only at the diagnosed failure point while maximally reusing validated reasoning prefixes and retrieved evidence. By explicitly separating error attribution from correction, DR-RAG enables precise error localization, thereby avoiding expensive full-pipeline reruns and enabling targeted, efficient repair. We evaluate DR-RAG across three multi-hop question answering benchmarks, multiple agentic RAG baselines, and different backbone models. Experimental results demonstrate that DR-RAG substantially improves answer accuracy while significantly reducing reasoning token consumption compared to rerun-based repair strategies.
