cs.CLApr 17, 2026

Skill-RAG: Failure-State-Aware Retrieval Augmentation via Hidden-State Probing and Skill Routing

Authors: Kai WeiRaymond LiXi ZhuZhaoqian XueJiaojiao HanJingcheng NiuFan Yang

Organizations: University of Michigan · University of British Columbia · Rutgers University · University of Pennsylvania · New Jersey Institute of Technology · Independent Researcher · Wake Forest University

Abstract

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.

Explore similar work

CardsList
  1. Diagnosing and Repairing Factual Errors in RAG under Budget Constraints

    Jun 28, 2026Soroush Hashemifar, Havva Alizadeh Noughabi, Fattane Zarrinkalam +1Token BudgetsErrors