Distribution-informed Efficient Conformal Prediction for Full Ranking

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Liao, Wenbo, Huang, Huipeng, Jia, Chen, Xi, Huajun, Zeng, Hao, Wei, Hongxin
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912863004655616
author Liao, Wenbo
Huang, Huipeng
Jia, Chen
Xi, Huajun
Zeng, Hao
Wei, Hongxin
author_facet Liao, Wenbo
Huang, Huipeng
Jia, Chen
Xi, Huajun
Zeng, Hao
Wei, Hongxin
contents Quantifying uncertainty is critical for the safe deployment of ranking models in real-world applications. Recent work offers a rigorous solution using conformal prediction in a full ranking scenario, which aims to construct prediction sets for the absolute ranks of test items based on the relative ranks of calibration items. However, relying on upper bounds of non-conformity scores renders the method overly conservative, resulting in substantially large prediction sets. To address this, we propose Distribution-informed Conformal Ranking (DCR), which produces efficient prediction sets by deriving the exact distribution of non-conformity scores. In particular, we find that the absolute ranks of calibration items follow Negative Hypergeometric distributions, conditional on their relative ranks. DCR thus uses the rank distribution to derive non-conformity score distribution and determine conformal thresholds. We provide theoretical guarantees that DCR achieves improved efficiency over the baseline while ensuring valid coverage under mild assumptions. Extensive experiments demonstrate the superiority of DCR, reducing average prediction set size by up to 36%, while maintaining valid coverage.
format Preprint
id arxiv_https___arxiv_org_abs_2601_23128
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Distribution-informed Efficient Conformal Prediction for Full Ranking
Liao, Wenbo
Huang, Huipeng
Jia, Chen
Xi, Huajun
Zeng, Hao
Wei, Hongxin
Machine Learning
Quantifying uncertainty is critical for the safe deployment of ranking models in real-world applications. Recent work offers a rigorous solution using conformal prediction in a full ranking scenario, which aims to construct prediction sets for the absolute ranks of test items based on the relative ranks of calibration items. However, relying on upper bounds of non-conformity scores renders the method overly conservative, resulting in substantially large prediction sets. To address this, we propose Distribution-informed Conformal Ranking (DCR), which produces efficient prediction sets by deriving the exact distribution of non-conformity scores. In particular, we find that the absolute ranks of calibration items follow Negative Hypergeometric distributions, conditional on their relative ranks. DCR thus uses the rank distribution to derive non-conformity score distribution and determine conformal thresholds. We provide theoretical guarantees that DCR achieves improved efficiency over the baseline while ensuring valid coverage under mild assumptions. Extensive experiments demonstrate the superiority of DCR, reducing average prediction set size by up to 36%, while maintaining valid coverage.
title Distribution-informed Efficient Conformal Prediction for Full Ranking
topic Machine Learning
url https://arxiv.org/abs/2601.23128