HoTS: Homophily-Aware Temperature Scaling for Graph Neural Network Calibration
Not on the map. Why?
Abstract (arXiv)
For graph node classification, calibrated class probabilities are needed when confidence scores, usually the maximum predicted class probability, are used to rank predictions, defer uncertain nodes to human review, or control risk. Existing post-hoc calibrators either apply one global temperature or use graph-aware modules without a principled structural form. We study how local graph structure should enter node-level calibration. Our first results show that a logit-only temperature rule is insufficient when nodes with identical logits but different local homophily require different optimal temperatures. We then analyze a population-concentration contextual stochastic block model with Gaussian features and a one-layer linear GCN. Under equidistant class means, the Bayes posterior over class-template scores is a temperature-scaled softmax whose inverse-temperature is governed by a homophily-dependent signal strength. In the positive-signal homophilic regime, the resulting temperature decreases approximately inversely with normalized local homophily. This law motivates Homophily-aware Temperature Scaling (HoTS), a simple post-hoc calibrator that assigns each node a positive scalar temperature from entropy-based logit concentration and estimated local homophily. HoTS has three temperature parameters, preserves the predicted class, and learns the strength of the structural correction from calibration data. Across 18 node-classification benchmarks, two GNN backbones, and eight calibration baselines, HoTS achieves the best mean Expected Calibration Error (ECE) of 4.79%, the best average rank, and the most reliable confidence ranking in selective classification. Code is available at https://github.com/inu0104/HoTS.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 25 of its 39 references that are in the map. First posted on arXiv 26 Sep 2026.
Builds on
25 of 39 references.
References · 25
- 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
- 2020Open Graph Benchmark: Datasets for Machine Learning on Graphscited by 2,163
- 2018Pitfalls of Graph Neural Network Evaluationcited by 1,007
- 2016Revisiting Semi-Supervised Learning with Graph Embeddingscited by 976
- 2020Geom-GCN: Geometric Graph Convolutional Networkscited by 826
- 2019Beyond temperature scaling: Obtaining well-calibrated multiclass probabilities with Dirichlet calibrationcited by 364
- 2023A critical look at the evaluation of GNNs under heterophily: Are we really making progress?cited by 343
- 2020Mix-n-Match: Ensemble and Compositional Methods for Uncertainty Calibration in Deep Learningcited by 190
- 2021Is Homophily a Necessity for Graph Neural Networks?cited by 182
13 more
- 2021Be Confident! Towards Trustworthy Graph Neural Networks via Confidence Calibrationcited by 92
- 2020Intra Order-preserving Functions for Calibration of Multi-Class Neural Networkscited by 65
- 2022What Makes Graph Neural Networks Miscalibrated?cited by 40
- 2022Adaptive Temperature Scaling for Robust Calibration of Deep Neural Networkscited by 24
- 2020Contextual Stochastic Block Model: Sharp Thresholds and Contiguitycited by 22
- 2019Are Graph Neural Networks Miscalibrated?cited by 14
- 2023On Performance Discrepancies Across Local Homophily Levels in Graph Neural Networkscited by 13
- 2023Optimal Inference in Contextual Stochastic Block Modelscited by 12
- 2023SimCalib: Graph Neural Network Calibration based on Similarity between Nodescited by 10
- 2022On Calibration of Graph Neural Networks for Node Classificationcited by 9
- 2024GETS: Ensemble Temperature Scaling for Calibration in Graph Neural Networkscited by 4
- 2025WATS: Calibrating Graph Neural Networks with Wavelet-Aware Temperature Scalingcited by 2
- 2026Post-Calibration Reliability Reranking of Relevance Decisions via Label-wise Monotone Projectioncited by 2
Built on by
Not cited yet.