English

OptiSet: Unified Optimizing Set Selection and Ranking for Retrieval-Augmented Generation

Artificial Intelligence 2026-01-09 v1

Abstract

Retrieval-Augmented Generation (RAG) improves generation quality by incorporating evidence retrieved from large external corpora. However, most existing methods rely on statically selecting top-k passages based on individual relevance, which fails to exploit combinatorial gains among passages and often introduces substantial redundancy. To address this limitation, we propose OptiSet, a set-centric framework that unifies set selection and set-level ranking for RAG. OptiSet adopts an "Expand-then-Refine" paradigm: it first expands a query into multiple perspectives to enable a diverse candidate pool and then refines the candidate pool via re-selection to form a compact evidence set. We then devise a self-synthesis strategy without strong LLM supervision to derive preference labels from the set conditional utility changes of the generator, thereby identifying complementary and redundant evidence. Finally, we introduce a set-list wise training strategy that jointly optimizes set selection and set-level ranking, enabling the model to favor compact, high-gain evidence sets. Extensive experiments demonstrate that OptiSet improves performance on complex combinatorial problems and makes generation more efficient. The source code is publicly available.

Keywords

Cite

@article{arxiv.2601.05027,
  title  = {OptiSet: Unified Optimizing Set Selection and Ranking for Retrieval-Augmented Generation},
  author = {Yi Jiang and Sendong Zhao and Jianbo Li and Bairui Hu and Yanrui Du and Haochun Wang and Bing Qin},
  journal= {arXiv preprint arXiv:2601.05027},
  year   = {2026}
}

Comments

Code is available at https://github.com/liunian-Jay/OptiSet.git

R2 v1 2026-07-01T08:56:17.073Z