Get our free extension to see links to code for papers anywhere online!

Chrome logo  Add to Chrome

Firefox logo Add to Firefox


Reliable Graph Neural Networks for Drug Discovery Under Distributional Shift

Add code

Nov 25, 2021
Kehang Han, Balaji Lakshminarayanan, Jeremiah Liu


Share this with someone who'll enjoy it:


The concern of overconfident mis-predictions under distributional shift demands extensive reliability research on Graph Neural Networks used in critical tasks in drug discovery. Here we first introduce CardioTox, a real-world benchmark on drug cardio-toxicity to facilitate such efforts. Our exploratory study shows overconfident mis-predictions are often distant from training data. That leads us to develop distance-aware GNNs: GNN-SNGP. Through evaluation on CardioTox and three established benchmarks, we demonstrate GNN-SNGP's effectiveness in increasing distance-awareness, reducing overconfident mis-predictions and making better calibrated predictions without sacrificing accuracy performance. Our ablation study further reveals the representation learned by GNN-SNGP improves distance-preservation over its base architecture and is one major factor for improvements.

* 5 page main body, 5 page appendix. Accepted by NeurIPS DistShift Workshop 2021 


   Access Paper Source



Share this with someone who'll enjoy it: