Uncertainty Quantification Using Neural Networks for Molecular Property Prediction
Lior Hirschfeld,Kyle Swanson,Kevin Yang,Regina Barzilay,Connor W. Coley
DOI: https://doi.org/10.1021/acs.jcim.0c00502
IF: 6.162
2020-07-23
Journal of Chemical Information and Modeling
Abstract: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.The Supporting Information is available free of charge at <a class="ext-link" href="/doi/10.1021/acs.jcim.0c00502?goto=supporting-info">https://pubs.acs.org/doi/10.1021/acs.jcim.0c00502</a>.Additional results relating to quantitative model performance on random splits, the negative log likelihood evaluation, and the Wilcoxon signed-rank test (<a class="ext-link" href="/doi/suppl/10.1021/acs.jcim.0c00502/suppl_file/ci0c00502_si_001.pdf">PDF</a>)This article has not yet been cited by other publications.
chemistry, multidisciplinary, medicinal,computer science, interdisciplinary applications, information systems