Uncertainty Quantification Using Neural Networks for Molecular Property Prediction

Uncertainty quantification (UQ) is an important component of molecular property prediction, particularly for drug discovery applications where model predictions direct experimental design and where unanticipated imprecision wastes valuable time and resources. The need for UQ is especially acute for...

Full description

Bibliographic Details
Main Authors: Hirschfeld, Lior, Swanson, Kyle, Yang, Kevin, Barzilay, Regina, Coley, Connor W
Other Authors: Massachusetts Institute of Technology. Department of Chemical Engineering
Format: Article
Language:English
Published: American Chemical Society (ACS) 2021
Online Access:https://hdl.handle.net/1721.1/135244
_version_ 1826204057127092224
author Hirschfeld, Lior
Swanson, Kyle
Yang, Kevin
Barzilay, Regina
Coley, Connor W
author2 Massachusetts Institute of Technology. Department of Chemical Engineering
author_facet Massachusetts Institute of Technology. Department of Chemical Engineering
Hirschfeld, Lior
Swanson, Kyle
Yang, Kevin
Barzilay, Regina
Coley, Connor W
author_sort Hirschfeld, Lior
collection MIT
description Uncertainty quantification (UQ) is an important component of molecular property prediction, particularly for drug discovery applications where model predictions direct experimental design and where unanticipated imprecision wastes valuable time and resources. The need for UQ is especially acute for neural models, which are becoming increasingly standard yet are challenging to interpret. While several approaches to UQ have been proposed in the literature, there is no clear consensus on the comparative performance of these models. In this paper, we study this question in the context of regression tasks. We systematically evaluate several methods on five regression data sets using multiple complementary performance metrics. Our experiments show that none of the methods we tested is unequivocally superior to all others, and none produces a particularly reliable ranking of errors across multiple data sets. While we believe that these results show that existing UQ methods are not sufficient for all common use cases and further research is needed, we conclude with a practical recommendation as to which existing techniques seem to perform well relative to others.
first_indexed 2024-09-23T12:48:14Z
format Article
id mit-1721.1/135244
institution Massachusetts Institute of Technology
language English
last_indexed 2024-09-23T12:48:14Z
publishDate 2021
publisher American Chemical Society (ACS)
record_format dspace
spelling mit-1721.1/1352442023-02-23T16:31:33Z Uncertainty Quantification Using Neural Networks for Molecular Property Prediction Hirschfeld, Lior Swanson, Kyle Yang, Kevin Barzilay, Regina Coley, Connor W Massachusetts Institute of Technology. Department of Chemical Engineering Uncertainty quantification (UQ) is an important component of molecular property prediction, particularly for drug discovery applications where model predictions direct experimental design and where unanticipated imprecision wastes valuable time and resources. The need for UQ is especially acute for neural models, which are becoming increasingly standard yet are challenging to interpret. While several approaches to UQ have been proposed in the literature, there is no clear consensus on the comparative performance of these models. In this paper, we study this question in the context of regression tasks. We systematically evaluate several methods on five regression data sets using multiple complementary performance metrics. Our experiments show that none of the methods we tested is unequivocally superior to all others, and none produces a particularly reliable ranking of errors across multiple data sets. While we believe that these results show that existing UQ methods are not sufficient for all common use cases and further research is needed, we conclude with a practical recommendation as to which existing techniques seem to perform well relative to others. 2021-10-27T20:22:37Z 2021-10-27T20:22:37Z 2020 2020-12-01T18:07:21Z Article http://purl.org/eprint/type/JournalArticle https://hdl.handle.net/1721.1/135244 en 10.1021/acs.jcim.0c00502 Journal of Chemical Information and Modeling Creative Commons Attribution-Noncommercial-Share Alike http://creativecommons.org/licenses/by-nc-sa/4.0/ application/pdf American Chemical Society (ACS) arXiv
spellingShingle Hirschfeld, Lior
Swanson, Kyle
Yang, Kevin
Barzilay, Regina
Coley, Connor W
Uncertainty Quantification Using Neural Networks for Molecular Property Prediction
title Uncertainty Quantification Using Neural Networks for Molecular Property Prediction
title_full Uncertainty Quantification Using Neural Networks for Molecular Property Prediction
title_fullStr Uncertainty Quantification Using Neural Networks for Molecular Property Prediction
title_full_unstemmed Uncertainty Quantification Using Neural Networks for Molecular Property Prediction
title_short Uncertainty Quantification Using Neural Networks for Molecular Property Prediction
title_sort uncertainty quantification using neural networks for molecular property prediction
url https://hdl.handle.net/1721.1/135244
work_keys_str_mv AT hirschfeldlior uncertaintyquantificationusingneuralnetworksformolecularpropertyprediction
AT swansonkyle uncertaintyquantificationusingneuralnetworksformolecularpropertyprediction
AT yangkevin uncertaintyquantificationusingneuralnetworksformolecularpropertyprediction
AT barzilayregina uncertaintyquantificationusingneuralnetworksformolecularpropertyprediction
AT coleyconnorw uncertaintyquantificationusingneuralnetworksformolecularpropertyprediction