Skill-RAG: Failure-State-Aware Retrieval Augmentation via Hidden-State Probing and Skill Routing
On the map: Retrieval-augmented generation → RAG robustness and adaptive retrieval
Abstract (arXiv)
Retrieval-Augmented Generation (RAG) has emerged as a foundational paradigm for grounding large language models in external knowledge. While adaptive retrieval mechanisms have improved retrieval efficiency, existing approaches treat post-retrieval failure as a signal to retry rather than to diagnose -- leaving the structural causes of query-evidence misalignment unaddressed. We observe that a significant portion of persistent retrieval failures stem not from the absence of relevant evidence but from an alignment gap between the query and the evidence space. We propose Skill-RAG, a failure-aware RAG framework that couples a lightweight hidden-state prober with a prompt-based skill router. The prober gates retrieval at two pipeline stages; upon detecting a failure state, the skill router diagnoses the underlying cause and selects among four retrieval skills -- query rewriting, question decomposition, evidence focusing, and an exit skill for truly irreducible cases -- to correct misalignment before the next generation attempt. Experiments across multiple open-domain QA and complex reasoning benchmarks show that Skill-RAG substantially improves accuracy on hard cases persisting after multi-turn retrieval, with particularly strong gains on out-of-distribution datasets. Representation-space analyses further reveal that the proposed skills occupy structured, separable regions of the failure state space, supporting the view that query-evidence misalignment is a typed rather than monolithic phenomenon.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in RAG robustness and adaptive retrieval, one of 15 directions in Retrieval-augmented generation. Builds on 38 of its 42 references that are in the map. First posted on arXiv 17 Apr 2026.
Builds on
38 of 42 references.
Closest ancestors · 2
Same direction, newest first
- 2024Adaptive-RAG: Learning to Adapt Retrieval-Augmented Large Language Models through Question Complexity21 Mar
- 2023Self-RAG: Learning to Retrieve, Generate, and Critique through Self-Reflection17 Oct
Other references · 36
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,824
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,955
- 2023Retrieval-Augmented Generation for Large Language Models: A Surveycited by 2,212
- 2020REALM: Retrieval-Augmented Language Model Pre-Trainingcited by 2,141
- 2022Language Models (Mostly) Know What They Knowcited by 1,856
- 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,282
- 2022When Not to Trust Language Models: Investigating Effectiveness of Parametric and Non-Parametric Memoriescited by 1,128
24 more
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,075
- 2022Interleaving Retrieval with Chain-of-Thought Reasoning for Knowledge-Intensive Multi-Step Questionscited by 923
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 916
- 2023Large Language Models Can Be Easily Distracted by Irrelevant Contextcited by 841
- 2022Discovering Latent Knowledge in Language Models Without Supervisioncited by 767
- 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 712
- 2023The Internal State of an LLM Knows When It's Lyingcited by 659
- 2022Precise Zero-Shot Dense Retrieval without Relevance Labelscited by 517
- 2023Active Retrieval Augmented Generationcited by 516
- 2023DoLa: Decoding by Contrasting Layers Improves Factuality in Large Language Modelscited by 376
- 2023Enhancing Retrieval-Augmented Large Language Models with Iterative Retrieval-Generation Synergycited by 346
- 2022Demonstrate-Search-Predict: Composing retrieval and language models for knowledge-intensive NLPcited by 318
- 2024MultiHop-RAG: Benchmarking Retrieval-Augmented Generation for Multi-Hop Queriescited by 257
- 2023Query Rewriting for Retrieval-Augmented Large Language Modelscited by 213
- 2024Corrective Retrieval Augmented Generationcited by 197
- 2024Don't Hallucinate, Abstain: Identifying LLM Knowledge Gaps via Multi-LLM Collaborationcited by 180
- 2024Lookback Lens: Detecting and Mitigating Contextual Hallucinations in Large Language Models Using Only Attention Mapscited by 142
- 2022Task-aware Retrieval with Instructionscited by 107
- 2024DRAGIN: Dynamic Retrieval Augmented Generation based on the Information Needs of Large Language Modelscited by 63
- 2024Disentangling Memory and Reasoning Ability in Large Language Modelscited by 44
- 2024Probing-RAG: Self-Probing to Guide Language Models in Selective Document Retrievalcited by 29
- 2025DeepSieve: Information Sieving via LLM-as-a-Knowledge-Routercited by 13
- 2025SAGE: An Agentic Explainer Framework for Interpreting SAE Features in Language Modelscited by 10
- 2026Farther the Shift, Sparser the Representation: Analyzing OOD Mechanisms in LLMscited by 4
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Skill Is Not Document: Query-Conditioned Compatibility for LLM Agent Skill Routingcited by 7
- 2026From Relevance to Execution Utility: Reward-Aware Dynamic Execution Gating for Skill-Based LLM Agentscited by 2
- 2026What Would Fix This RAG Failure? Auditing Counterfactual Response with Paired Evidence Interventionscited by 0
- 2026From Relevance to Execution Utility: Reward-Aware Dynamic Execution Gating for Skill-Based LLM Agents10 Aug 2026
- 2026What Would Fix This RAG Failure? Auditing Counterfactual Response with Paired Evidence Interventions9 Aug 2026
- 2026Skill Is Not Document: Query-Conditioned Compatibility for LLM Agent Skill Routing2 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026Mapping the RAG Landscape: A Four Axis Taxonomy of Efficiency, Defense, Interactivity, and Reasoning
- 2025Retrieval is Not Enough: Enhancing RAG Reasoning through Test-Time Critique and Optimization
- 2026PruneRAG: Confidence-Guided Query Decomposition Trees for Efficient Retrieval-Augmented Generation
- 2024Astute RAG: Overcoming Imperfect Retrieval Augmentation and Knowledge Conflicts for Large Language Models
- 2026When to Retrieve During Reasoning: Adaptive Retrieval for Large Reasoning Models
- 2026Diagnosing and Repairing Factual Errors in RAG under Budget Constraints
- 2026Stateful Evidence-Driven Retrieval-Augmented Generation with Iterative Reasoning
- 2026Reason and Verify: A Framework for Faithful Retrieval-Augmented Generation