SAGE: Benchmarking and Improving Retrieval for Deep Research Agents
The paper introduces SAGE, a benchmark revealing that LLM-based retrievers underperform traditional keyword methods in deep research agents due to query generation mismatches, and proposes a corpus-level test-time scaling framework that significantly boosts retrieval performance by augmenting documents with LLM-generated metadata.
Original paper licensed under CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). This is an AI-generated explanation of the paper below. It is not written or endorsed by the authors. For technical accuracy, refer to the original paper. Read full disclaimer
The Big Idea: The "Smart Librarian" vs. The "Keyword Machine"
Imagine you are a researcher trying to find a very specific book in a massive library of 200,000 books. You have a super-smart AI assistant (a "Deep Research Agent") whose job is to find that book, read it, and answer your question.
The paper asks a critical question: Does it help to give this AI assistant a "super-smart librarian" (an LLM-based retriever) who understands the meaning of your question, or is a "keyword machine" (like BM25) that just matches words actually better?
The authors built a test called SAGE (Scientific AGentic retrieval Evaluation) to find out. They created 1,200 tricky questions across four fields (Computer Science, Natural Science, Health, and Humanities) and tested six different AI research agents.
The Surprise: The "Keyword Machine" Wins
The researchers expected the "super-smart librarian" to win because it can understand complex reasoning and nuance. They thought it would be better at finding papers that require deep thinking.
But the results were the opposite.
- The Result: The simple "keyword machine" (BM25) beat the "super-smart librarian" (LLM-based retrievers) by a huge margin—about 30% better.
- The Reason: The AI agents, when they break down a big question into smaller search steps, act like they are shouting keywords rather than asking full sentences.
- Analogy: Imagine you ask the AI to find a paper about "physics-informed heuristics." Instead of asking the librarian, "Do you have a book about using physics rules to solve math problems?", the AI shouts: "Physics! Heuristics! ICML! 2023!"
- The "super-smart librarian" gets confused by these fragmented keywords and tries to guess the meaning, often getting it wrong. The "keyword machine," however, is excellent at matching those exact shouted words to the book titles and abstracts.
The Solution: "Pre-Seasoning" the Library
Since the AI agents are stuck shouting keywords, the researchers asked: Can we make the library itself easier for the keyword machine to search?
They proposed a new method called Corpus-Level Test-Time Scaling.
- The Analogy: Imagine the library books are written in a complex language that is hard to search. Instead of teaching the librarian to speak a new language, the researchers went into the library and added a "cheat sheet" to the front of every single book.
- How it works: They used a powerful AI to read every paper and write down a list of keywords and metadata (like the year, authors, and main topics) right at the very top of the document.
- The Result: Now, when the AI agent shouts its keywords, they hit these cheat sheets immediately.
- This improved the performance by 8% for tricky, fact-based questions.
- It improved performance by 2% for open-ended research questions.
Key Takeaways
- Smarter isn't always better: In this specific workflow, a "dumb" tool that just matches words worked better than a "smart" tool that tries to understand meaning, because the AI agents weren't asking "smart" questions.
- The Agent's Habit: The AI agents naturally break questions into keyword lists. They don't write full sentences for the search tool.
- The Fix: If you can't change how the agent asks questions, change the library. By adding extra keywords to the documents themselves, you make the "dumb" search tool much more effective.
In short: The paper shows that for deep research agents, we don't necessarily need smarter search tools; we need to make our documents "speak the language" of the search tools we already have.
Drowning in papers in your field?
Get daily digests of the most novel papers matching your research keywords — with technical summaries, in your language.