Distribution Free Prediction Sets for Node Classification
International Conference on Machine Learning (ICML), 2022
- AI4CE
Main:7 Pages
2 Figures
Bibliography:3 Pages
3 Tables
Appendix:1 Pages
Abstract
Graph Neural Networks (GNNs) are able to achieve high classification accuracy on many large real world datasets, but provide no rigorous notion of predictive uncertainty. We leverage recent advances in conformal prediction to construct prediction sets for node classification in inductive learning scenarios, and verify the efficacy of our approach across standard benchmark datasets using popular GNN models. The code is available at \href{https://github.com/jase-clarkson/graph_cp}{this link}.
View on arXivComments on this paper
