Retrieval-as-memory hybrids
37 systems in the retrieval-as-memory hybrids category of the AI Agent Infrastructure Landscape, grouped by maturity tier.
Tier 1 — battle-tested (1)
- GraphRAG (Microsoft) Community summary + Leiden detection
Modular graph-based RAG. Leiden community detection over knowledge graph; community summaries at multiple levels. DRIFT Search hybrid (Oct 2024).
Tier 3 — emerging (20)
- Adaptive-RAG Query-complexity routing
Smaller classifier LM predicts query complexity, then routes to no-retrieval / single-step / iterative retrieval as appropriate. NAACL 2024.
- Atlas Few-shot retrieval-augmented LM
Meta AI. Jointly fine-tunes retriever + LM for few-shot tasks.
- CAG (Cache-Augmented Generation) KV-cache preloading replaces retrieval
"Don't Do RAG." Preloads all relevant documents into context, caches the resulting KV state; at inference the model answers from cache with no retrieval step — eliminates retrieval latency + retrieval-error failure modes.
- FLARE Confidence-triggered forward retrieval
Forward-Looking Active Retrieval. Uses model's low-confidence token predictions as signal to anticipate future information needs and retrieve proactively. Retrieves only when token probability falls below threshold; generator stays frozen.
- Generative Semantic Workspace (GSW) Episodic memory for RAG
Replaces chunk-retrieval with neuro-inspired generative memory. Operator maps observations to intermediate semantic structures; Reconciler integrates them into a persistent workspace with temporal/spatial/logical coherence. Targets 100k–…
- HippoRAG / HippoRAG2 Personalised PageRank over KG
Hippocampus-inspired RAG. LLMs + knowledge graphs + Personalized PageRank to mimic neocortex/hippocampus roles. NeurIPS 2024.
- HippoRAG 2 Hippocampus-inspired RAG v2
OSU follow-on to HippoRAG — adds online updates, recall-precision tradeoff calibration, and multi-hop reasoning improvements.
- HiRAG Hierarchical KG indexing + retrieval
Multi-layer knowledge graph: higher layers hold coarse summary entities improving connectivity between similar lower-layer entities (HiIndex). HiRetrieval traverses across granularity levels.
- HyDE Hypothetical document embeddings
LLM generates a hypothetical answer document; embeds that instead of the query. Closes the query-document semantic gap for retrieval. arxiv 2212.10496.
- ITER-RETGEN Output-conditioned iterative retrieval
Uses model's generated output as a rich context signal for the next retrieval round rather than just the original query. Processes all retrieved knowledge as a whole, preserving generation flexibility.
- KAG KG-grounded hybrid reasoning
Knowledge-Augmented Generation. LLM-friendly KG representation + mutual indexing between KGs and raw chunks + logical-form-guided hybrid reasoning. Handles numerical / temporal / multi-hop chains that vector similarity misses.
- LazyGraphRAG Deferred-indexing GraphRAG variant
Performs minimal up-front indexing; defers LLM use until query time. Being integrated into the main GraphRAG library.
- LightRAG Graph-augmented dual-level RAG
Hong Kong U graph-RAG variant — dual-level retrieval (local + global) over an entity-relation graph built incrementally from corpus updates.
- LightRAG Dual-level graph retrieval
Entity-relationship graph + broader-theme retrieval. Incremental update support. Integrates with RAG-Anything for multimodal docs.
- RAFT Retrieval-augmented fine-tuning
UC Berkeley. Trains the model to ignore distractor documents and quote the relevant text in chain-of-thought.
- RAPTOR Recursive abstractive tree RAG
Stanford recursive-summarisation tree over the corpus — clusters and summarises documents into a hierarchical index; retrieves at multiple levels. ICLR 2024.
- RAPTOR Recursive abstractive summary tree
Recursively embeds, clusters, and summarises chunks into a tree of abstractions. Retrieval traverses the tree at query time. ICLR 2024.
- REALM Retrieval-augmented LM pre-training
Foundational paper. Augments LM pre-training with a latent knowledge retriever — model retrieves and attends over Wikipedia-scale corpus during pre-training, fine-tuning, and inference. Backprop through retrieval over millions of documen…
- RETRO Trillion-token retrieval transformer
DeepMind. Retrieval-Enhanced Transformer with chunked cross-attention over a 2T-token retrieval database. Frozen BERT retriever + differentiable encoder.
- TC-RAG Turing-complete state-variable RAG
Frames RAG as Turing-complete with explicit monitored state variables, enabling adaptive retrieval halting and convergence control. Medical-LLM case study.
Tier 4 — early / experimental (16)
- AutoRAG AutoML-style greedy RAG pipeline
Applies AutoML to RAG: greedy algorithm evaluates and selects best-performing module at each pipeline stage (query expansion, retrieval, reranking, generation). Validated on ARAGOG (423 papers, 107 QAs).
- Beyond RAG for Agent Memory Decoupled-aggregate memory retrieval
Argues standard RAG breaks on bounded agent conversation streams (highly correlated, redundant spans). Decomposes the stream into semantic units, organises them into themes, then inverts that structure for retrieval.
- BGE-M3 Unified dense-sparse-multi-vector embedding
Single embedding model supporting dense, sparse, and multi-vector (ColBERT-style) retrieval simultaneously. 100+ languages, inputs up to 8192 tokens. Self-knowledge distillation across the three modes.
- ChunkRAG LLM-scored chunk-level filtering
Semantic chunking divides documents into coherent sections; LLM-based relevance scorer filters each chunk against the query before generation, pruning irrelevant chunks and reducing hallucination.
- CRAG Retrieval self-correction with web fallback
Lightweight retrieval evaluator scores docs and triggers different actions: use, supplement with web search, or discard. Decompose-then-recompose algorithm filters irrelevant content from retrieved passages.
- GraphRAFT RAG fine-tuning over KGs
Retrieval-augmented fine-tuning for knowledge graphs on graph databases.
- LongRAG Long retrieval-unit grouping
Groups related Wikipedia documents into 4K-token units. Reduces NQ corpus 22M→600K units, HotpotQA 5M→500K. Hands longer passages to a long-context reader (GPT-4o).
- MemoRAG Memory-augmented RAG
RAG variant with a long-context global memory model that drafts answer clues used to guide passage retrieval — improves recall in noisy multi-hop QA.
- Multi-Head RAG (MHRAG) Attention-head multi-aspect embeddings
Uses transformer multi-head attention activations (not decoder-layer embeddings) as retrieval keys, yielding multi-aspect embeddings that independently capture different semantic facets for heterogeneous queries.
- PathRAG Path-pruned graph RAG
Retrieves over an indexed graph by pruning candidate relational paths — reduces noise in graph-RAG vs flat path expansion.
- PathRAG Path-pruned graph retrieval
Retrieves key relational paths from knowledge graph; flow-based pruning + reliability scoring; path-based prompting. Outperforms baselines on six datasets. arxiv Feb 2025.
- RouteRAG RL-routed text + graph + hybrid
End-to-end RL-trained RAG. Token-level policy chooses when to retrieve and which mode (text / KG / hybrid via RRF). Two-stage Group-Relative Policy Optimisation (GRPO). No public repo at time of writing.
- Self-RAG Reflection-token self-critique
Trains an LLM end-to-end to emit reflection tokens that decide whether to retrieve, assess passage relevance, and critique output quality — all within a single model.
- Speculative RAG Small-model draft, large-model verify
Smaller distilled specialist generates multiple RAG drafts in parallel from distinct document subsets. Larger generalist verifies in a single pass. Subsets formed by content-similarity clustering to minimise redundancy.
- StructRAG Task-adaptive structurisation
Identifies optimal structure (table, catalog, algorithm, graph) per task; converts source documents into that structure at inference time and reasons over the structured form. Motivated by cognitive theories of human knowledge conversion.
- WeKnow-RAG Sparse-dense web + KG self-assessment
Domain KGs + multi-stage web-page retrieval (sparse + dense) + LLM self-assessment of trustworthiness before finalising answer.