AISTATS 2023poster29 citations
Probabilistic Conformal Prediction Using Conditional Random Samples
Zhendong Wang, Ruijiang Gao, Mingzhang Yin, Mingyuan Zhou, David Blei
Abstract
This paper proposes probabilistic conformal prediction (PCP), a predictive inference algorithm that estimates a target variable by a discontinuous predictive set. Given inputs, PCP constructs the predictive set based on random samples from an estimated generative model. It is efficient and compatible with conditional generative models with either explicit or implicit density functions. We show that PCP guarantees correct marginal coverage with finite samples and give empirical evidence of conditional coverage. We study PCP on a variety of simulated and real datasets. Compared to existing conformal prediction methods, PCP provides sharper predictive sets.
BibTeX
@InProceedings{pmlr-v206-wang23n,
title = {Probabilistic Conformal Prediction Using Conditional Random Samples},
author = {Wang, Zhendong and Gao, Ruijiang and Yin, Mingzhang and Zhou, Mingyuan and Blei, David},
booktitle = {Proceedings of The 26th International Conference on Artificial Intelligence and Statistics},
pages = {8814--8836},
year = {2023},
editor = {Ruiz, Francisco and Dy, Jennifer and van de Meent, Jan-Willem},
volume = {206},
series = {Proceedings of Machine Learning Research},
month = {25--27 Apr},
publisher = {PMLR},
pdf = {https://proceedings.mlr.press/v206/wang23n/wang23n.pdf},
url = {https://proceedings.mlr.press/v206/wang23n.html},
abstract = {This paper proposes probabilistic conformal prediction (PCP), a predictive inference algorithm that estimates a target variable by a discontinuous predictive set. Given inputs, PCP constructs the predictive set based on random samples from an estimated generative model. It is efficient and compatible with conditional generative models with either explicit or implicit density functions. We show that PCP guarantees correct marginal coverage with finite samples and give empirical evidence of conditional coverage. We study PCP on a variety of simulated and real datasets. Compared to existing conformal prediction methods, PCP provides sharper predictive sets.}
}