atlas

Research map / Large language models / Retrieval and knowledge

RAPTOR: Recursive Abstractive Processing for Tree-Organized Retrieval

ICLR 2024 · Large language models › Retrieval and knowledge · approach: Retrieval-augmented reasoning · 28 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes RAPTOR, which builds a tree by recursively clustering (GMM+UMAP) and summarizing text chunks and retrieves across tree layers via a collapsed-tree search, improving QA on NarrativeQA, QASPER, and QuALITY.

Problem

Retrieval-augmented LMs retrieve only short contiguous chunks, which limits holistic understanding of long documents needed for thematic, multi-hop questions.

Approach

Recursively embed, cluster (soft GMM clustering with UMAP and BIC-selected cluster count), and summarize chunks into a bottom-up tree, then retrieve from all layers at once (collapsed tree) so that answers can draw on both granular details and high-level summaries.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add RAPTOR recursive clustering and summarization to SBERT retrievalUnifiedQA-3B · QuALITYaccuracy: 54.9% -> 56.6%Helped
add RAPTOR recursive clustering and summarization to SBERT retrievalUnifiedQA-3B · QASPERAnswer F1: 36.23% -> 36.70%Helped
add RAPTOR recursive clustering and summarization to SBERT retrievalUnifiedQA-3B · NarrativeQAROUGE: 29.26% -> 30.87%Helped
add RAPTOR recursive clustering and summarization to SBERT retrievalUnifiedQA-3B · NarrativeQABLEU-1: 22.56% -> 23.50%Helped
use RAPTOR retrieval instead of DPRGPT-4 · QASPERF-1 Match: 53.0 -> 55.7Helped
use RAPTOR retrieval instead of DPRGPT-3 · QuALITY devaccuracy: 60.4 -> 62.4Helped
use RAPTOR retrieval instead of CoLISAGPT-4 · QuALITY test setaccuracy: 62.3 -> 82.6Helped
use RAPTOR + UnifiedQA instead of Retriever + ReaderNarrativeQA · NarrativeQAMETEOR: 11.1 -> 19.1Helped
replace semantic clustering with contiguous-chunk balanced-tree construction (ablation)RAPTOR + SBERT embeddings + UnifiedQA · QuALITYaccuracy: 56.6% -> 55.8%Hurt
query all three layers starting at Layer 2 instead of only Layer 0 (ablation)RAPTOR · QuALITY Story 1accuracy: 57.9 -> 73.68Helped
query two layers starting at Layer 1 instead of one layer at Layer 1 (ablation)RAPTOR · QuALITY Story 2accuracy: 47.1 -> 64.7Helped
query all three layers starting at Layer 2 instead of only Layer 2 (ablation)RAPTOR · QuALITY Story 3accuracy: 61.1 -> 83.3Helped

Surprising findings

Limitations

Experimental setup

Datasets: NarrativeQA QASPER QuALITY

Benchmarks: NarrativeQA QASPER QuALITY

Metrics: accuracy F1 Match ROUGE-L BLEU-1 BLEU-4 METEOR Answer F1

Baselines: SBERT BM25 DPR Title + Abstract LongT5 XL CoLT5 XL CoLISA Recursively Summarizing Books Retriever + Reader BiDAF Longformer-base DPR and DeBERTaV3-large

Builds on: retrieval-augmented language models UMAP SBERT Dense Passage Retrieval recursive summarization

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy