← Search

Yuesong Shen

6 accepted papers

2026

Graph Neural Networks Are Not Continuous Across Graph Resolutions

ICML 2026poster

We show that contrary to conventional wisdom in the community, graph neural networks (GNNs) are not continuous with respect to all natural modes of graph convergence. As a result, GNNs may generate substantially different latent representations for graphs that are very similar. In particular they as…

Cited by 0SourceScholar
2024

Variational Learning is Effective for Large Deep Networks

ICML 2024spotlight

We give extensive empirical evidence against the common belief that variational learning is ineffective for large neural networks. We show that an optimizer called Improved Variational Online Newton (IVON) consistently matches or outperforms Adam for training large networks such as GPT-2 and ResNets…

2023

Beyond In-Domain Scenarios: Robust Density-Aware Calibration

ICML 2023poster

Calibrating deep learning models to yield uncertainty-aware predictions is crucial as deep neural networks get increasingly deployed in safety-critical applications. While existing post-hoc calibration methods achieve impressive results on in-domain test datasets, they are limited by their inability…

2022

What Makes Graph Neural Networks Miscalibrated?

NeurIPS 2022accept

Given the importance of getting calibrated predictions and reliable uncertainty estimations, various post-hoc calibration methods have been developed for neural networks on standard multi-class classification tasks. However, these methods are not well suited for calibrating graph neural networks (GN…

2021

Explicit pairwise factorized graph neural network for semi-supervised node classification

UAI 2021poster

Node features and structural information of a graph are both crucial for semi-supervised node classification problems. A variety of graph neural network (GNN) based approaches have been proposed to tackle these problems, which typically determine output labels through feature aggregation. This can b…