394

Distribution Free Prediction Sets for Node Classification

International Conference on Machine Learning (ICML), 2022
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 arXiv
Comments on this paper