用简单检索提升复杂推理任务表现,效果远超预期。
Frustratingly Simple Retrieval Improves Challenging, Reasoning-Intensive Benchmarks
- 构建轻量级高质量网络数据集CompactDS,兼顾速度与覆盖
- 在多个推理基准上实现10%-33%的准确率提升,模型越大效果越明显
- 无需复杂系统,自研数据集可媲美谷歌搜索和智能体方案
检索增强生成(RAG)以往主要应用于事实类问答等简单场景,面对复杂推理任务如MMLU、MMLU Pro、AGI Eval、GPQA和MATH等,现有最小化RAG方法表现有限。本文指出此前研究缺失的关键要素:一个与预训练数据广度对齐的可用网络规模数据集。为此,提出CompactDS——一个多样、高质量、可扩展的网络数据集,在单机上实现高精度检索与亚秒级延迟。核心洞察为:大部分网页内容可过滤而不损失覆盖范围;通过内存中近似最近邻检索与磁盘精确搜索结合,平衡速度与召回率。使用CompactDS,最小化RAG流水线在所有基准及模型规模(8B–70B)上均取得稳定提升,相对增益达:MMLU+10%,MMLU Pro+33%,GPQA+14%,MATH+19%。单一数据源无法满足需求,凸显来源多样性的重要性(网络爬取、数学专库、学术论文、教科书)。最终验证,自研数据集在性能上可匹敌或超越谷歌搜索及近期复杂代理式RAG系统,同时保持简洁、可复现与自包含。项目代码与数据集已开源,支持后续检索驱动型AI研究。
原文摘要 · Abstract (English)
Retrieval-augmented Generation (RAG) has primarily been studied in limited settings, such as factoid question answering; more challenging, reasoning-intensive benchmarks have seen limited success from minimal RAG. In this work, we challenge this prevailing view on established, reasoning-intensive benchmarks: MMLU, MMLU Pro, AGI Eval, GPQA, and MATH. We identify a key missing component in prior work: a usable, web-scale datastore aligned with the breadth of pretraining data. To this end, we introduce CompactDS: a diverse, high-quality, web-scale datastore that achieves high retrieval accuracy and subsecond latency on a single-node. The key insights are (1) most web content can be filtered out without sacrificing coverage, and a compact, high-quality subset is sufficient; and (2) combining in-memory approximate nearest neighbor (ANN) retrieval and on-disk exact search balances speed and recall. Using CompactDS, we show that a minimal RAG pipeline achieves consistent accuracy improvements across all benchmarks and model sizes (8B--70B), with relative gains of 10% on MMLU, 33% on MMLU Pro, 14% on GPQA, and 19% on MATH. No single data source suffices alone, highlighting the importance of diversity of sources (web crawls, curated math, academic papers, textbooks). Finally, we show that our carefully designed in-house datastore matches or outperforms web search engines such as Google Search, as well as recently proposed, complex agent-based RAG systems--all while maintaining simplicity, reproducibility, and self-containment. We release CompactDS and our retrieval pipeline, supporting future research exploring retrieval-based AI systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。