Clarifying trust of materials property predictions using neural networks with distribution-specific uncertainty quantification

Gruich, Cameron J and Madhavan, Varun and Wang, Yixin and Goldsmith, Bryan R (2023) Clarifying trust of materials property predictions using neural networks with distribution-specific uncertainty quantification. Machine Learning: Science and Technology, 4 (2). 025019. ISSN 2632-2153

[thumbnail of Gruich_2023_Mach._Learn.__Sci._Technol._4_025019.pdf] Text
Gruich_2023_Mach._Learn.__Sci._Technol._4_025019.pdf - Published Version

Download (17MB)

Abstract

It is critical that machine learning (ML) model predictions be trustworthy for high-throughput catalyst discovery approaches. Uncertainty quantification (UQ) methods allow estimation of the trustworthiness of an ML model, but these methods have not been well explored in the field of heterogeneous catalysis. Herein, we investigate different UQ methods applied to a crystal graph convolutional neural network to predict adsorption energies of molecules on alloys from the Open Catalyst 2020 dataset, the largest existing heterogeneous catalyst dataset. We apply three UQ methods to the adsorption energy predictions, namely k-fold ensembling, Monte Carlo dropout, and evidential regression. The effectiveness of each UQ method is assessed based on accuracy, sharpness, dispersion, calibration, and tightness. Evidential regression is demonstrated to be a powerful approach for rapidly obtaining tunable, competitively trustworthy UQ estimates for heterogeneous catalysis applications when using neural networks. Recalibration of model uncertainties is shown to be essential in practical screening applications of catalysts using uncertainties.

Item Type: Article
Subjects: Apsci Archives > Multidisciplinary
Depositing User: Unnamed user with email support@apsciarchives.com
Date Deposited: 10 Oct 2023 05:43
Last Modified: 10 Oct 2023 05:43
URI: http://eprints.go2submission.com/id/eprint/1572

Actions (login required)

View Item
View Item