English

Shifting from Ranking to Set Selection for Retrieval Augmented Generation

Computation and Language 2025-07-11 v2 Information Retrieval

Abstract

Retrieval in Retrieval-Augmented Generation(RAG) must ensure that retrieved passages are not only individually relevant but also collectively form a comprehensive set. Existing approaches primarily rerank top-k passages based on their individual relevance, often failing to meet the information needs of complex queries in multi-hop question answering. In this work, we propose a set-wise passage selection approach and introduce SETR, which explicitly identifies the information requirements of a query through Chain-of-Thought reasoning and selects an optimal set of passages that collectively satisfy those requirements. Experiments on multi-hop RAG benchmarks show that SETR outperforms both proprietary LLM-based rerankers and open-source baselines in terms of answer correctness and retrieval quality, providing an effective and efficient alternative to traditional rerankers in RAG systems. The code is available at https://github.com/LGAI-Research/SetR

Keywords

Cite

@article{arxiv.2507.06838,
  title  = {Shifting from Ranking to Set Selection for Retrieval Augmented Generation},
  author = {Dahyun Lee and Yongrae Jo and Haeju Park and Moontae Lee},
  journal= {arXiv preprint arXiv:2507.06838},
  year   = {2025}
}

Comments

Accepted to ACL 2025 main (Oral Presentation)

R2 v1 2026-07-01T03:53:10.637Z