LGMay 21

Do Deep Ensembles Actually Capture Uncertainty in Graph Neural Networks?

arXiv:2605.2259346.5
Predicted impact top 53% in LG · last 90 daysOriginality Incremental advance
AI Analysis

For practitioners using graph neural networks, this paper reveals that the default deep ensemble method for uncertainty quantification fails to transfer from computer vision, highlighting the need for alternative approaches.

Deep ensembles provide marginal improvement over single models for uncertainty quantification in graph neural networks, with gains primarily from stabilizing optimization noise rather than better uncertainty estimates, due to epistemic collapse where independently trained networks converge to overly similar predictions.

While deep ensembles are widely considered to be the default method for uncertainty quantification in deep learning, their effectiveness for graph-structured data is often simply assumed based on successes in domains like computer vision. We investigate standard deep ensembles specifically for message-passing graph neural networks. Benchmarking across seven datasets representing varied tasks and complexities, we reveal that ensembles provide surprisingly little improvement over a single model. Instead, the observed marginal gains stem primarily from stabilizing optimization noise in point predictions rather than yielding meaningfully better uncertainty estimates. Through an aleatoric-epistemic decomposition, we identify epistemic collapse: independently trained networks consistently converge to overly similar predictions. Because disagreement is the fundamental mechanism through which ensembles capture epistemic uncertainty, this lack of diversity neutralizes their key advantage. Analyzing this phenomenon further, we suggest this collapse is driven by functional rather than weight-space convexity, where distinct parameter solutions induce almost identical behavior. Our results suggest that deep ensemble success does not seamlessly transfer to graph machine learning.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes