Online Bayesian Node Classification on Inductive Graphs under Distribution Shift
Not on the map. Why?
Abstract (arXiv)
On evolving graphs, node classifiers must satisfy two key requirements: inductive generalization to newly arriving nodes under distribution shift and calibrated uncertainty for safety-sensitive applications. Standard graph neural networks (GNNs) are typically trained once and address neither requirement. We adapt the Bayesian last-layer (BLL) model by placing random last-layer parameters on top of a deterministic GNN encoder for uncertainty quantification. The categorical softmax likelihood required for classification breaks Gaussian conjugacy, so neither the training posterior nor the test-time streaming update has a closed-form solution. To address both challenges, we introduce a variational Bayesian last-layer (VBLL) objective that jointly trains the encoder and an approximate last-layer posterior by maximizing an evidence lower bound with a Monte Carlo expected log-likelihood. At test time, we freeze the encoder and apply an online Laplace update to the last-layer posterior. This update corresponds to a power-prior Bayesian model with exponential forgetting and a Kullback-Leibler anchor to the training posterior. Across five node-classification benchmarks under distribution shift, online GVBLL is the only method to achieve the best accuracy and negative log-likelihood on every dataset. It improves accuracy by up to 17 percentage points on Cora and 14 percentage points on ogbn-arxiv over the strongest non-GVBLL baseline, while remaining competitive in calibration with MC Dropout, Deep Ensembles, Temperature Scaling, and Gaussian-process classifiers.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 20 of its 23 references that are in the map. First posted on arXiv 12 Sep 2026.
Builds on
20 of 23 references.
References · 20
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2017Graph Attention Networkscited by 8,923
- 2017Inductive Representation Learning on Large Graphscited by 6,320
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,787
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
- 2018How Powerful are Graph Neural Networks?cited by 4,393
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2015Deep Kernel Learningcited by 445
- 2017Variational Continual Learningcited by 339
- 2022Handling Distribution Shifts on Graphs: An Invariance Perspectivecited by 164
- 2021Be Confident! Towards Trustworthy Graph Neural Networks via Confidence Calibrationcited by 92
8 more
- 2018Bayesian graph convolutional neural networks for semi-supervised classificationcited by 89
- 2020Bayesian Graph Neural Networks with Adaptive Connection Samplingcited by 81
- 2020Graph Neural Networks: Architectures, Stability and Transferabilitycited by 80
- 2024Variational Bayesian Last Layerscited by 69
- 2020Streaming Graph Neural Networks via Continual Learningcited by 64
- 2021Knowledge-Adaptation Priorscited by 20
- 2024Online scalable Gaussian processes with conformal prediction for guaranteed coveragecited by 3
- 2025Conformalized Gaussian processes for online uncertainty quantification over graphscited by 1
Built on by
Not cited yet.