English
Related papers

Related papers: GNN's Uncertainty Quantification using Self-Distil…

200 papers

Uncertainty quantification is central to safe and efficient deployments of deep learning models, yet many computationally practical methods lack lacking rigorous theoretical motivation. Random network distillation (RND) is a lightweight…

Machine Learning · Computer Science 2026-02-27 Moritz A. Zanger , Yijun Wu , Pascal R. Van der Vaart , Wendelin Böhmer , Matthijs T. J. Spaan

Graph Neural Networks (GNN) provide a powerful framework that elegantly integrates Graph theory with Machine learning for modeling and analysis of networked data. We consider the problem of quantifying the uncertainty in predictions of GNN…

Machine Learning · Computer Science 2022-05-23 Sai Munikoti , Deepesh Agarwal , Laya Das , Balasubramaniam Natarajan

Graph Neural Networks (GNNs) have been extensively used in various real-world applications. However, the predictive uncertainty of GNNs stemming from diverse sources such as inherent randomness in data and model training errors can lead to…

Machine Learning · Computer Science 2025-03-11 Fangxin Wang , Yuqing Liu , Kay Liu , Yibo Wang , Sourav Medya , Philip S. Yu

Uncertainty quantification is a critical aspect of reinforcement learning and deep learning, with numerous applications ranging from efficient exploration and stable offline reinforcement learning to outlier detection in medical…

Machine Learning · Computer Science 2025-03-27 Moritz A. Zanger , Pascal R. Van der Vaart , Wendelin Böhmer , Matthijs T. J. Spaan

Uncertainty estimation is critical for reliable medical image segmentation, particularly in retinal vessel analysis, where accurate predictions are essential for diagnostic applications. Deep Ensembles, where multiple networks are trained…

Computer Vision and Pattern Recognition · Computer Science 2025-09-16 Jeremiah Fadugba , Petru Manescu , Bolanle Oladejo , Delmiro Fernandez-Reyes , Philipp Berens

Recently, the teacher-student knowledge distillation framework has demonstrated its potential in training Graph Neural Networks (GNNs). However, due to the difficulty of training over-parameterized GNN models, one may not easily obtain a…

Machine Learning · Computer Science 2021-05-03 Yuzhao Chen , Yatao Bian , Xi Xiao , Yu Rong , Tingyang Xu , Junzhou Huang

In this work we aim to obtain computationally-efficient uncertainty estimates with deep networks. For this, we propose a modified knowledge distillation procedure that achieves state-of-the-art uncertainty estimates both for in and…

Machine Learning · Computer Science 2019-06-14 Erik Englesson , Hossein Azizpour

While several methods for predicting uncertainty on deep networks have been recently proposed, they do not readily translate to large and complex datasets. In this paper we utilize a simplified form of the Mixture Density Networks (MDNs) to…

Machine Learning · Computer Science 2019-12-05 Nicholas Wilkins , Michael Johnson , Ifeoma Nwogu

Consistency training is a popular method to improve deep learning models in computer vision and natural language processing. Graph neural networks (GNNs) have achieved remarkable performance in a variety of network science learning tasks,…

Machine Learning · Computer Science 2021-10-14 Cole Hawkins , Vassilis N. Ioannidis , Soji Adeshina , George Karypis

Deep neural networks lack interpretability and tend to be overconfident, which poses a serious problem in safety-critical applications like autonomous driving, medical imaging, or machine vision tasks with high demands on reliability.…

Computer Vision and Pattern Recognition · Computer Science 2024-03-27 Steven Landgraf , Kira Wursthorn , Markus Hillemann , Markus Ulrich

To bridge the gaps between topology-aware Graph Neural Networks (GNNs) and inference-efficient Multi-Layer Perceptron (MLPs), GLNN proposes to distill knowledge from a well-trained teacher GNN into a student MLP. Despite their great…

Machine Learning · Computer Science 2023-06-12 Lirong Wu , Haitao Lin , Yufei Huang , Stan Z. Li

Graph, such as citation networks, social networks, and transportation networks, are prevalent in the real world. Graph Neural Networks (GNNs) have gained widespread attention for their robust expressiveness and exceptional performance in…

Machine Learning · Computer Science 2023-03-01 Jing Liu , Tongya Zheng , Guanzheng Zhang , Qinfen Hao

Recent advances in graph neural networks (GNNs) have enabled more comprehensive modeling of molecules and molecular systems, thereby enhancing the precision of molecular property prediction and molecular simulations. Nonetheless, as the…

Machine Learning · Computer Science 2023-10-31 Filip Ekström Kelvinius , Dimitar Georgiev , Artur Petrov Toshev , Johannes Gasteiger

Deep ensembles deliver state-of-the-art, reliable uncertainty quantification, but their heavy computational and memory requirements hinder their practical deployments to real applications such as on-device AI. Knowledge distillation…

Machine Learning · Computer Science 2025-10-27 Sehyun Park , Jongjin Lee , Yunseop Shin , Ilsang Ohn , Yongdai Kim

Thanks to graph neural networks (GNNs), semi-supervised node classification has shown the state-of-the-art performance in graph data. However, GNNs have not considered different types of uncertainties associated with class probabilities to…

Machine Learning · Computer Science 2020-11-26 Xujiang Zhao , Feng Chen , Shu Hu , Jin-Hee Cho

Considering uncertainty estimation of modern neural networks (NNs) is one of the most important steps towards deploying machine learning systems to meaningful real-world applications such as in medicine, finance or autonomous systems. At…

Machine Learning · Computer Science 2022-05-20 Martin Ferianc , Miguel Rodrigues

Graph Neural Networks (GNNs) have emerged as a prominent class of data-driven methods for molecular property prediction. However, a key limitation of typical GNN models is their inability to quantify uncertainties in the predictions. This…

Machine Learning · Computer Science 2024-07-02 Shengli Jiang , Shiyi Qin , Reid C. Van Lehn , Prasanna Balaprakash , Victor M. Zavala

Graph neural networks (GNNs) have been shown to be astonishingly capable models for molecular property prediction, particularly as surrogates for expensive density functional theory calculations of relaxed energy for novel material…

Machine Learning · Computer Science 2024-08-27 Joseph Musielewicz , Janice Lan , Matt Uyttendaele , John R. Kitchin

Uncertainty quantification is essential for deploying reliable Graph Neural Networks (GNNs), where existing approaches primarily rely on Bayesian inference or ensembles. In this paper, we introduce the first credal graph neural networks…

Machine Learning · Computer Science 2025-12-03 Matteo Tolloso , Davide Bacciu

Recent research on graph neural networks (GNNs) has explored mechanisms for capturing local uncertainty and exploiting graph hierarchies to mitigate data sparsity and leverage structural properties. However, the synergistic integration of…

Machine Learning · Computer Science 2025-05-06 Yoonhyuk Choi , Jiho Choi , Taewook Ko , Chong-Kwon Kim
‹ Prev 1 2 3 10 Next ›