English

Principled and Scalable Diversity-Aware Retrieval via Cardinality-Constrained Binary Quadratic Programming

Computation and Language 2026-04-06 v1 Information Retrieval

Abstract

Diversity-aware retrieval is essential for Retrieval-Augmented Generation (RAG), yet existing methods lack theoretical guarantees and face scalability issues as the number of retrieved passages kk increases. We propose a principled formulation of diversity retrieval as a cardinality-constrained binary quadratic programming (CCBQP), which explicitly balances relevance and semantic diversity through an interpretable trade-off parameter. Inspired by recent advances in combinatorial optimization, we develop a non-convex tight continuous relaxation and a Frank--Wolfe based algorithm with landscape analysis and convergence guarantees. Extensive experiments demonstrate that our method consistently dominates baselines on the relevance-diversity Pareto frontier, while achieving significant speedup.

Keywords

Cite

@article{arxiv.2604.02554,
  title  = {Principled and Scalable Diversity-Aware Retrieval via Cardinality-Constrained Binary Quadratic Programming},
  author = {Qiheng Lu and Nicholas D. Sidiropoulos},
  journal= {arXiv preprint arXiv:2604.02554},
  year   = {2026}
}
R2 v1 2026-07-01T11:52:02.427Z