Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Graph Posterior Network: Bayesian Predictive Uncertainty for Node Classification

About

The interdependence between nodes in graphs is key to improve class predictions on nodes and utilized in approaches like Label Propagation (LP) or in Graph Neural Networks (GNN). Nonetheless, uncertainty estimation for non-independent node-level predictions is under-explored. In this work, we explore uncertainty quantification for node classification in three ways: (1) We derive three axioms explicitly characterizing the expected predictive uncertainty behavior in homophilic attributed graphs. (2) We propose a new model Graph Posterior Network (GPN) which explicitly performs Bayesian posterior updates for predictions on interdependent nodes. GPN provably obeys the proposed axioms. (3) We extensively evaluate GPN and a strong set of baselines on semi-supervised node classification including detection of anomalous features, and detection of left-out classes. GPN outperforms existing approaches for uncertainty estimation in the experiments.

Maximilian Stadler, Bertrand Charpentier, Simon Geisler, Daniel Z\"ugner, Stephan G\"unnemann• 2021

Related benchmarks

TaskDatasetResultRank
OOD DetectionCora
AUROC85.65
43
OOD DetectionPubmed
AUROC86.75
24
Out-of-Distribution DetectionPubmed--
24
OOD DetectionCiteseer
AUROC76.89
19
OOD DetectionOgbn-arxiv
AUROC83.46
18
Out-of-Distribution DetectionCiteseer
In-Distribution Accuracy76.76
14
OOD DetectionwikiCS
FPR@9555.17
13
OOD DetectionBooks-History
FPR@9551.02
13
O.O.D. detectionSquirrel (LoC)
AUC-PR (Aleatoric)59.8
12
O.O.D. detectionPubMed (LoC)
AUC-PR (Aleatoric)55.7
12
Showing 10 of 20 rows

Other info

Follow for update