Minimum-Length Conformal Prediction Sets for Ordinal Classification
Zijian Zhang, Xinyu Chen, Yuanjie Shi, Liyuan Lillian Ma, Zifan Xu, Yan Yan
摘要
Ordinal classification has been widely applied in many high-stakes applications, e.g., medical imaging and diagnosis, where reliable uncertainty quantification (UQ) is essential for decision making. Conformal prediction (CP) is a general UQ framework that provides statistically valid guarantees, which is especially useful in practice. However, prior ordinal CP methods mainly focus on heuristic algorithms or restrictively require the underlying model to predict a unimodal distribution over ordinal labels. Consequently, they provide limited insight into coverage–efficiency trade-offs, or a model-agnostic and distribution-free nature favored by CP methods. To this end, we fill this gap by propose an ordinal-CP method that is model-agnostic and provides instance-level optimal prediction intervals. Specifically, we formulate conformal ordinal classification as a minimum-length covering problem at the instance level. To solve this problem, we develop a sliding-window algorithm that is optimal on each calibration data, with only a linear time complexity in K, the # of label candidates. The local optimality per instance further also improves predictive efficiency in expectation. Moreover, we propose a length-regularized variant that shrinks prediction set size while preserving coverage. Experiments on four benchmark datasets from diverse domains are conducted to demonstrate the significantly improved predictive efficiency of the proposed methods over baselines (by 15%↓ on average over four datasets).
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper21
- Class-Conditional Conformal Prediction with Many ClassesTiffany Ding, Anastasios Angelopoulos, Stephen Bates, Michael I. Jordan 等NeurIPS 2023 · 被引用 160 次
- Learning Optimal Conformal ClassifiersDavid Stutz, Krishnamurthy Dvijotham, Ali Taylan Cemgil, Arnaud DoucetICLR 2022 · 被引用 123 次
- Large language model validity via enhanced conformal prediction methodsJohn J. Cherian, Isaac Gibbs, Emmanuel J. CandèsNeurIPS 2024 · 被引用 120 次
- IEBins: Iterative Elastic Bins for Monocular Depth EstimationShuwei Shao, Zhongcai Pei, Xingming Wu, Zhong Liu 等NeurIPS 2023 · 被引用 114 次
- Moving Window Regression: A Novel Approach to Ordinal RegressionNyeong-Ho Shin, Seon-Ho Lee, Chang-Su KimCVPR 2022 · 被引用 79 次
相关 Paper
- Conformal Prediction Sets for Ordinal ClassificationPrasenjit Dey, Srujana Merugu, Sivaramakrishnan R. KaveriNeurIPS 2023 · 被引用 12 次
- Efficient and Differentiable Conformal Prediction with General Function ClassesYu Bai, Song Mei, Huan Wang, Yingbo Zhou 等ICLR 2022 · 被引用 29 次
- Multi-model Ensemble Conformal Prediction in Dynamic EnvironmentsErfan Hajihashemi, Yanning ShenNeurIPS 2024 · 被引用 13 次
- Model Uncertainty Quantification by Conformal Prediction in Continual LearningRui Gao, Weiwei LiuICML 2025
- Conformal Prediction for Class-wise Coverage via Augmented Label Rank CalibrationYuanjie Shi, Subhankar Ghosh, Taha Belkhouja, Jana Doppa 等NeurIPS 2024 · 被引用 29 次
