cs.LGSep 12, 2026

Online Bayesian Node Classification on Inductive Graphs under Distribution Shift

Authors: Jinwen Xu, Gonzalo Mateos Buckstein, Qin Lu

Organizations: School of Electrical and Computer Engineering, University of Georgia, Athens, GA, USA · Department of Electrical and Computer Engineering, University of Rochester, Rochester, NY, USA

Abstract

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.

Figures & tables

Explore similar work

CardsList
  1. A Unified Uncertainty Representation for Graph Neural Networks via Doubly-Spectral Stochastic Expansion

    Sep 28, 2026Fred Xu, Thomas Markovich, Florence Regol +1Graph Neural NetworksUncertainty Quantification

  2. Hidden Axis of Uncertainty: Latent-Posterior Alignment in Graph Neural Networks with Bayesian Output Layers

    Aug 21, 2026Suk Hoon Choi, Damdae Park, Junhyuk Choi +4Bayesian Neural NetworksGraph Neural Networks

  3. Variational Filtering for Bayesian Last Layers with an Independent Binary Probit Surrogate

    Oct 12, 2026Dimitrije Markovic, Conor Heins, Alexander Tschantz +1Bayesian Neural NetworksBayesian Inference