Quantile-Free Uncertainty Quantification in Graph Neural Networks
Soyoung Park, Hwanjun Song, Sungsu Lim
摘要
Uncertainty quantification (UQ) in graph neural networks (GNNs) is crucial in high-stakes domains but remains a significant challenge. In graph settings, message passing often relies on strong assumptions such as exchangeability, which are rarely satisfied in practice. Moreover, achieving reliable UQ typically requires costly resampling or post-hoc calibration. To address these issues, we introduce Quantile-free Prediction Interval GNN (QpiGNN), a framework that builds on quantile regression (QR) to enable GNN-based UQ by directly optimizing coverage and interval width without requiring quantile inputs or post-processing. QpiGNN employs a dual-head architecture that decouples prediction and uncertainty, and is trained with label-only supervision through a quantile-free joint loss. This design allows efficient training and yields robust prediction intervals, with theoretical guarantees of asymptotic coverage and near-optimal width under mild assumptions. Experiments on 19 synthetic and real-world benchmarks show QpiGNN achieves average 22% higher coverage and 50% narrower intervals than baselines, while ensuring efficiency and robustness to noise and structural shifts.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper9
- Deep Evidential RegressionAlexander Amini, Wilko Schwarting, Ava Soleimany, Daniela RusNeurIPS 2020 · 被引用 777 次
- Graph Structure Learning for Robust Graph Neural NetworksWei Jin, Yao Ma, Xiaorui Liu, Xianfeng Tang 等KDD 2020 · 被引用 604 次
- Is Homophily a Necessity for Graph Neural Networks?Yao Ma, Xiaorui Liu, Neil Shah, Jiliang TangICLR 2022 · 被引用 295 次
- Uncertainty Aware Semi-Supervised Learning on Graph DataXujiang Zhao, Feng Chen, Shu Hu, Jin-Hee ChoNeurIPS 2020 · 被引用 178 次
- Graph Posterior Network: Bayesian Predictive Uncertainty for Node ClassificationMaximilian Stadler, Bertrand Charpentier, Simon Geisler, Daniel Zügner 等NeurIPS 2021 · 被引用 133 次
相关 Paper
- Conformalized Link Prediction on Graph Neural NetworksTianyi Zhao, Jian Kang, Lu ChengKDD 2024 · 被引用 9 次
- Relaxed Quantile Regression: Prediction Intervals for Asymmetric NoiseThomas Pouplin, Alan Jeffares, Nabeel Seedat, Mihaela van der SchaarICML 2024 · 被引用 9 次
- Uncertainty Quantification over Graph with Conformalized Graph Neural NetworksKexin Huang, Ying Jin, Emmanuel J. Candès, Jure LeskovecNeurIPS 2023 · 被引用 124 次
- Enhancing Trustworthiness of Graph Neural Networks with Rank-Based Conformal TrainingTing Wang, Zhixin Zhou, Rui LuoAAAI 2025 · 被引用 12 次
- Accurate and Scalable Estimation of Epistemic Uncertainty for Graph Neural NetworksPuja Trivedi, Mark Heimann, Rushil Anirudh, Danai Koutra 等ICLR 2024 · 被引用 8 次
