LAnR：在 LLM 隐空间中统一编码、检索与生成的 RAG 框架

View PDF HTML (experimental)

Abstract:Retrieval-Augmented Generation (RAG) has become a standard approach for enhancing large language models (LLMs) with external knowledge, mitigating hallucinations, and improving factuality. However, existing systems rely on generating natural language queries at each hop and maintaining a strict architectural separation between retriever and generator, preventing them from leveraging the full representational capacity of the LLM. We propose \textbf{LAnR} (Latent Abstraction for RAG), a unified framework in which a single LLM jointly performs encoding, retrieval, and generation entirely within its own latent space. Rather than generating textual queries, LAnR produces dense retrieval vectors from the hidden states of a designated \texttt{[PRED]} token and uses them to match against encoded document representations from the same model. Furthermore, LAnR adaptively decides when sufficient evidence has been retrieved using a lightweight MLP control head over those same hidden states, eliminating both the separate retriever and explicit token-level stopping reasoning. This design is motivated by our empirical observation that answer token entropy reliably signals retrieval sufficiency. Extensive experiments on six QA benchmarks spanning single-hop and multi-hop settings demonstrate that LAnR outperforms existing RAG methods, while achieving improved inference efficiency through reduced number of retrieval calls and tighter model integration.

Submission history

From: Minh Nguyen [view email]
[v1] Mon, 20 Apr 2026 06:26:13 UTC (534 KB)
[v2] Thu, 7 May 2026 12:30:55 UTC (1,316 KB)
[v3] Mon, 19 Oct 2026 08:18:07 UTC (1,321 KB)
[v4] Tue, 27 Oct 2026 08:09:49 UTC (1,321 KB)