Uncertainty quantification for predictions of atomistic neural networks

The value of uncertainty quantification on predictions for trained neural networks (NNs) on quantum chemical reference data is quantitatively explored. For this, the architecture of the PhysNet NN was suitably modified and the resulting model (PhysNet-DER) was evaluated with different metrics to quantify its calibration, the quality of its predictions, and whether prediction error and the predicted uncertainty can be correlated. Training on the QM9 database and evaluating data in the test set within and outside the distribution indicate that error and uncertainty are not linearly related. However, the observed variance provides insight into the quality of the data used for training. Additionally, the influence of the chemical space covered by the training data set was studied by using a biased database. The results clarify that noise and redundancy complicate property prediction for molecules even in cases for which changes – such as double bond migration in two otherwise identical molecules – are small. The model was also applied to a real database of tautomerization reactions. Analysis of the distance between members in feature space in combination with other parameters shows that redundant information in the training dataset can lead to large variances and small errors whereas the presence of similar but unspecific information returns large errors but small variances. This was, e.g., observed for nitro-containing aliphatic chains for which predictions were difficult although the training set contained several examples for nitro groups bound to aromatic molecules. The finding underlines the importance of the composition of the training data and provides chemical insight into how this affects the prediction capabilities of a ML model. Finally, the presented method can be used for information-based improvement of chemical databases for target applications through active learning optimization.

Paper

References (12)

02QTautobase: A quantum tautomerization database2021
03Machine learning re  ned: Foundations, algorithms, and applications2020
04Physical Chemistry Chemical Physics (2022)2019 · 1M. Meuwly, Chem. Rev
05Wiley Inter-disciplinary Reviews: Data Mining and Knowledge Discovery 9 , e1289 (2019). 29 O. A. Von Lilienfeld2018 · Angew. Chem. Int. Ed.
07Compendium of Chemical Terminology (the “ Gold Book ” )1997 · IUPAC
08International Union of Pure and Applied Chemistry: Z¨urich, Switzerland1997
10FIG. S2. Distribution of standard deviation( σ = √ V ar ) for different values of hyperparameter λ All the values were considered
11. S3. Evolution of the Expected Normalized Calibration Error (ENCE), sharpness, and the Coefficient of Variation ( C v ) depending on λ for the 95% of the variance distribution
12FIG. S1. Learning curves showing the improvement of model with respect to training set size 37

Similar papers

© 2026 NYSGPT2525 LLC