Approximating Full Conformal Prediction at Scale via Influence Functions
Keywords:ML: Calibration & Uncertainty Quantification, PEAI: Safety, Robustness & Trustworthiness
AbstractConformal prediction (CP) is a wrapper around traditional machine learning models, giving coverage guarantees under the sole assumption of exchangeability; in classification problems, a CP guarantees that the error rate is at most a chosen significance level, irrespective of whether the underlying model is misspecified. However, the prohibitive computational costs of full CP led researchers to design scalable alternatives, which alas do not attain the same guarantees or statistical power of full CP. In this paper, we use influence functions to efficiently approximate full CP. We prove that our method is a consistent approximation of full CP, and empirically show that the approximation error becomes smaller as the training set increases; e.g., for 1,000 training points the two methods output p-values that are <0.001 apart: a negligible error for any practical application. Our methods enable scaling full CP to large real-world datasets. We compare our full CP approximation (ACP) to mainstream CP alternatives, and observe that our method is computationally competitive whilst enjoying the statistical predictive power of full CP.
How to Cite
Abad Martinez, J., Bhatt, U., Weller, A., & Cherubin, G. (2023). Approximating Full Conformal Prediction at Scale via Influence Functions. Proceedings of the AAAI Conference on Artificial Intelligence, 37(6), 6631-6639. https://doi.org/10.1609/aaai.v37i6.25814
AAAI Technical Track on Machine Learning I