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.