Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

AgentIR: A Workload-Adaptive Cascade Retrieval Substrate for Long-Term Conversational Memory

About

Long-term conversational memory is a retrieval workload classical IR was not built for: the index grows during the query stream, query types shift intra-session, and the latency budget per retrieval is sub-10 ms. Lucene-class engines treat the index as static and the query as stateless, leaving the workload's structure unexploited. AgentIR treats fusion as a per-query decision along two axes: which fusion to apply (BM25, Dense, RRF, or agent-aware RRF), and whether the ~52 ms dense channel is worth running at all. The second axis is a confidence-triggered cascade router that decides from the BM25 top-k margin alone and re-tunes across workloads without retraining. On LongMemEval (n=500), where the dense channel does add information, the cascade skips 63% of queries at parity LLM-judged accuracy (2.67x faster under two judges, paired bootstrap p>=0.88); per-qtype thresholds extend this to 5.76x under 5-fold cross-validation. On LoCoMo (n=1,982), where BM25 alone is already the strongest single system, the same trigger auto-tunes to a 100% skip rate (132x faster, +0.089 Hit@5). Capacity on a shared 8-core VM rises from ~154 to ~1,400 concurrent agents (9x). Underneath the cascade, a time-partitioned index does O(log 1/epsilon) work independent of corpus size: 1234x corpus growth costs only 3.6x latency, ending in 1769x over sequential at sub-100 us p50 on 5M records. At parity quality with Lucene on 9 BEIR datasets up to 8.8M docs, the substrate runs 10x geo-mean over Pyserini 8T and 11x over PISA-1T BlockMax-WAND; an A100 reaches 1.8-39x over Pyserini 8T; chunked index build sustains 56.8K docs/sec on MS MARCO. Three subtle BM25/GPU correctness pitfalls that silently regress nDCG@10 by 6-8x are documented and fixed; post-fix CPU and GPU agree within 0.0002 nDCG@10 on all eight datasets that fit a single A100.

Aojie Yuan, Haiyue Zhang, Shahin Nazarian• 2026

Related benchmarks

TaskDatasetResultRank
Information RetrievalNFCorpus (test)
NDCG@100.3267
67
Information RetrievalSciFact BEIR (test)
nDCG@1068.28
36
Information RetrievalQuora BEIR--
22
Information RetrievalBEIR scidocs (test)
nDCG@100.1566
10
Information RetrievalMS MARCO Passage
nDCG@1040.59
7
Long-term Conversational Memory RetrievalLongMemEval
R@592.4
4
Information RetrievalNFCorpus
nDCG@100.3267
3
Information RetrievalSciFact
nDCG@100.6828
3
Information RetrievalArguAna
nDCG@1036.45
3
Information RetrievalTREC-COVID
NDCG@1064.36
3
Showing 10 of 33 rows

Other info

Follow for update