Predictive Inference with Feature Conformal Prediction
Jiaye Teng,Chuan Wen,Dinghuai Zhang,Yoshua Bengio,Yang Gao,Yang Yuan
DOI: https://doi.org/10.48550/arXiv.2210.00173
2023-04-08
Abstract:Conformal prediction is a distribution-free technique for establishing valid prediction intervals. Although conventionally people conduct conformal prediction in the output space, this is not the only possibility. In this paper, we propose feature conformal prediction, which extends the scope of conformal prediction to semantic feature spaces by leveraging the inductive bias of deep representation learning. From a theoretical perspective, we demonstrate that feature conformal prediction provably outperforms regular conformal prediction under mild assumptions. Our approach could be combined with not only vanilla conformal prediction, but also other adaptive conformal prediction methods. Apart from experiments on existing predictive inference benchmarks, we also demonstrate the state-of-the-art performance of the proposed methods on large-scale tasks such as ImageNet classification and Cityscapes image <a class="link-external link-http" href="http://segmentation.The" rel="external noopener nofollow">this http URL</a> code is available at \url{<a class="link-external link-https" href="https://github.com/AlvinWen428/FeatureCP" rel="external noopener nofollow">this https URL</a>}.
Machine Learning,Artificial Intelligence,Methodology