← Search

Linwei Tao

9 accepted papers

2026

WATS: Wavelet-Aware Temperature Scaling for Reliable Graph Neural Networks

ICLR 2026poster

Graph Neural Networks (GNNs) have demonstrated strong predictive performance on relational data; however, their confidence estimates often misalign with actual predictive correctness, posing significant limitations for deployment in safety-critical settings. While existing graph-aware calibration me…

Cited by 0SourcecodeScholar
2025

Beyond One-Hot Labels: Semantic Mixing for Model Calibration

ICML 2025poster

Model calibration seeks to ensure that models produce confidence scores that accurately reflect the true likelihood of their predictions being correct. However, existing calibration approaches are fundamentally tied to datasets of one-hot labels implicitly assuming full certainty in all the annotati…

2025

Diffusion Attribution Score: Evaluating Training Data Influence in Diffusion Models

ICLR 2025spotlight

As diffusion models become increasingly popular, the misuse of copyrighted and private images has emerged as a major concern. One promising solution to mitigate this issue is identifying the contribution of specific training samples in generative models, a process known as data attribution. Existing…

Cited by 0SourcePDFScholar
2023

Calibrating a Deep Neural Network with Its Predecessors

IJCAI 2023poster

Confidence calibration - the process to calibrate the output probability distribution of neural networks - is essential for safety-critical applications of such networks. Recent works verify the link between mis-calibration and overfitting. However, early stopping, as a well-known technique to mitig…