Uncertainty quantification of molecular property prediction using Bayesian neural network models

Abstract

In chemistry, deep neural network models have been increasingly utilized in a variety of applications such as molecular property predictions, novel molecule designs, and planning chemical reactions. Despite the rapid increase in the use of state-of-the-art models and algorithms, deep neural network models often produce poor predictions in real applications because model performance is highly dependent on the quality of training data. In the field of molecular analysis, data are mostly obtained from either complicated chemical experiments or approximate mathematical equations, and then quality of data may be questioned.In this paper, we quantify uncertainties of prediction using Bayesian neural networks in molecular property predictions. We estimate both model-driven and data-driven uncertainties, demonstrating the usefulness of uncertainty quantification as both a quality checker and a confidence indicator with the three experiments. Our results manifest that uncertainty quantification is necessary for more reliable molecular applications and Bayesian neural network models can be a practical approach.

Workshop on "Machine Learning for Molecules and Materials", NIPS 2018. arXiv admin note: substantial text overlap with arXiv:1903.08375

References in corpus (15)

• Semi-Supervised Classification with Graph Convolutional Networks
• What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?
• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• Learning to Plan Chemical Syntheses
• Quantum-Chemical Insights from Deep Tensor Neural Networks
• Weight Uncertainty in Neural Networks
• Junction Tree Variational Autoencoder for Molecular Graph Generation
• Machine learning prediction errors better than DFT accuracy
• Objective-Reinforced Generative Adversarial Networks (ORGAN) for Sequence Generation Models
• MolGAN: An implicit generative model for small molecular graphs
• Learning Deep Generative Models of Graphs
• Graph Convolutional Policy Network for Goal-Directed Molecular Graph Generation
• Multiplicative Normalizing Flows for Variational Bayesian Neural Networks
• Atomic Convolutional Networks for Predicting Protein-Ligand Binding Affinity
• Deeply learning molecular structure-property relationships using attention- and gate-augmented graph convolutional network