mlsanityData to 14 Oct 2026

StateRAG: Typed State Contracts for Complex Retrieval-Augmented Generation

arXiv 2605.25379 ↗First posted 25 May 2026cs.CLcited by 0

On the map: Retrieval-augmented generation → RAG robustness and adaptive retrieval

Abstract (arXiv)

Complex retrieval-augmented generation requires evidence retrieval and control over what to retrieve next, which paths to explore, whether evidence is sufficient, and which intermediate results to retain. Existing RAG paradigms encode these decisions through method-specific model contexts, traversal procedures, verification signals, and memory. We introduce StateRAG, which represents retrieval control as a typed state external to the final reader. The state records the query plan, typed traversal path, candidate evidence, verification verdict, and reusable artifacts, with defined field semantics and designated update sources. Ordered role operators propose field values, and the controller validates and commits accepted proposals. A one-time compact-evidence check may select Bypass. Otherwise, the controller combines the committed verdict with the remaining budget to select Release, Revise, or Fallback. The final reader is invoked only after a terminal action and at most once per query. Among the evaluated methods, StateRAG achieves the best score on 10 of 11 reported quality metrics across LongBench, QASPER, and DocVQA. Its LongBench macro-average EM and F1 exceed ReAct by 6.3 and 6.1 percentage points, and its largest task-level gain is 9.8 F1 points on MuSiQue. Under normalized model-size weighting, StateRAG records lower mean LLM-token use per query than ReAct across all five workloads. In a matched LongBench control, the typed, role-owned carrier exceeds a free-form shared carrier by 1.3, 2.2, and 3.0 percentage points in macro EM, F1, and accuracy, with 6.2% lower mean RET. Removing TAM, MARS, or SMP lowers every reported LongBench and QASPER ablation metric.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in RAG robustness and adaptive retrieval, one of 15 directions in Retrieval-augmented generation. Builds on 14 of its 20 references that are in the map. First posted on arXiv 25 May 2026.

Builds on

14 of 20 references.

References · 14

  1. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
  2. 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
  3. 2024From Local to Global: A Graph RAG Approach to Query-Focused Summarizationcited by 1,453
  4. 2023MemGPT: Towards LLMs as Operating Systemscited by 1,389
  5. 2020DocVQA: A Dataset for VQA on Document Imagescited by 1,317
  6. 2024RAPTOR: Recursive Abstractive Processing for Tree-Organized Retrievalcited by 532
  7. 2025MemOS: A Memory OS for AI Systemcited by 138
  8. 2024MemoRAG: Boosting Long Context Processing with Global Memory-Enhanced Retrieval Augmentationcited by 119
  9. 2025Retrieval-Augmented Generation with Hierarchical Knowledgecited by 47
  10. 2026A-RAG: Scaling Agentic Retrieval-Augmented Generation via Hierarchical Retrieval Interfacescited by 24
  11. 2025Supervising the search process produces reliable and generalizable information-seeking agentscited by 20
  12. 2025Stop-RAG: Value-Based Retrieval Control for Iterative RAGcited by 14
2 more
  1. 2025Stronger Baselines for Retrieval-Augmented Generation with Long-Context Language Modelscited by 9
  2. 2025Hierarchical Lexical Graph for Enhanced Multi-Hop Retrievalcited by 7

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026BELIEFRAG: Making Adaptive RAG State-Aware under Evolving Evidence
  2. 2026Skill-RAG: Failure-State-Aware Retrieval Augmentation via Hidden-State Probing and Skill Routing
  3. 2026Orchestrating Specialized Agents for Trustworthy Enterprise RAG
  4. 2025RAGVUE: A Diagnostic View for Explainable and Automated Evaluation of Retrieval-Augmented Generation
  5. 2026PruneRAG: Confidence-Guided Query Decomposition Trees for Efficient Retrieval-Augmented Generation
  6. 2026Stateful Evidence-Driven Retrieval-Augmented Generation with Iterative Reasoning
  7. 2026S2G-RAG: Structured Sufficiency and Gap Judging for Iterative Retrieval-Augmented QA
  8. 2026Fast and Faithful: Real-Time Verification for Long-Document Retrieval-Augmented Generation Systems