Association Is Not Similarity: Learning Corpus-Specific Associations for Multi-Hop Retrieval
On the map: LLM reasoning → Knowledge- and tool-augmented reasoning
Abstract (arXiv)
Dense retrieval systems rank passages by embedding similarity to a query, but multi-hop questions require passages that are associatively related through shared reasoning chains. We introduce Association-Augmented Retrieval (AAR), a lightweight transductive reranking method that trains a small MLP (4.2M parameters) to learn associative relationships between passages in embedding space using contrastive learning on co-occurrence annotations. At inference time, AAR reranks an initial dense retrieval candidate set using bi-directional association scoring. On HotpotQA, AAR improves passage Recall@5 from 0.831 to 0.916 (+8.6 points) without evaluation-set tuning, with gains concentrated on hard questions where the dense baseline fails (+28.5 points). On MuSiQue, AAR achieves +10.1 points in the transductive setting. An inductive model trained on training-split associations and evaluated on unseen validation associations shows no significant improvement, suggesting that the method captures corpus-specific co-occurrences rather than transferable patterns. Ablation studies support this interpretation: training on semantically similar but non-associated passage pairs degrades retrieval below the baseline, while shuffling association pairs causes severe degradation. A downstream QA evaluation shows retrieval gains translate to +6.4 exact match improvement. The method adds 3.7ms per query, trains in under two minutes on a single GPU, and requires no LLM-based indexing.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Knowledge- and tool-augmented reasoning, one of 17 directions in LLM reasoning. Builds on 10 of its 18 references that are in the map. First posted on arXiv 13 Feb 2026.
Builds on
10 of 18 references.
References · 10
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,686
- 2024From Local to Global: A Graph RAG Approach to Query-Focused Summarizationcited by 1,411
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,259
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 893
- 2024RAPTOR: Recursive Abstractive Processing for Tree-Organized Retrievalcited by 519
- 2025From RAG to Memory: Non-Parametric Continual Learning for Large Language Modelscited by 223
- 2024CRAG -- Comprehensive RAG Benchmarkcited by 94
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Beyond One-Shot Expansion: Contrastive Evidence Exploration for Multi-Hop Retrieval
- 2026From Association to Causation: Improving Retrieval Precision of Retrieval-Augmented Generation via Causal Relations and an Attention Mechanism
- 2025Embedding-Based Context-Aware Reranker
- 2025From Ranking to Selection: A Simple but Efficient Dynamic Passage Selector for Retrieval Augmented Generation
- 2026Efficient Retrieval-Augmented Generation via Token Co-occurrence Graphs
- 2026CAGE: Coherence-Aware Graph Encoding for Retrieval-Augmented Generation
- 2025A Comprehensive Evaluation of Transformer-Based Question Answering Models and RAG-Enhanced Design
- 2025Beyond Independent Passages: Adaptive Passage Combination Retrieval for Retrieval Augmented Open-Domain Question Answering