Streamlining Conformal Information Retrieval via Score Refinement
Information Retrieval
2024-10-07 v1 Artificial Intelligence
Machine Learning
Abstract
Information retrieval (IR) methods, like retrieval augmented generation, are fundamental to modern applications but often lack statistical guarantees. Conformal prediction addresses this by retrieving sets guaranteed to include relevant information, yet existing approaches produce large-sized sets, incurring high computational costs and slow response times. In this work, we introduce a score refinement method that applies a simple monotone transformation to retrieval scores, leading to significantly smaller conformal sets while maintaining their statistical guarantees. Experiments on various BEIR benchmarks validate the effectiveness of our approach in producing compact sets containing relevant information.
Cite
@article{arxiv.2410.02914,
title = {Streamlining Conformal Information Retrieval via Score Refinement},
author = {Yotam Intrator and Ori Kelner and Regev Cohen and Roman Goldenberg and Ehud Rivlin and Daniel Freedman},
journal= {arXiv preprint arXiv:2410.02914},
year = {2024}
}
Comments
6 pages