Obtaining accurate estimates of machine learning model uncertainties on newly predicted data is essential for understanding the accuracy of the model and whether its predictions can be trusted. A common approach to such uncertainty quantification is to estimate the variance from an ensemble of models, which are often generated by the generally applicable bootstrap method. In this work, we demonstrate that the direct bootstrap ensemble standard deviation is not an accurate estimate of uncertainty but that it can be simply calibrated to dramatically improve its accuracy. We demonstrate the effectiveness of this calibration method for both synthetic data and numerous physical datasets from the field of Materials Science and Engineering. The approach is motivated by applications in physical and biological science but is quite general and should be applicable for uncertainty quantification in a wide range of machine learning regression models.
Calibration after bootstrap for accurate uncertainty quantification in regression models
Glenn Palmer,Siqi Du,A. Politowicz,Joshua Paul Emory,Xiyu Yang,Anupraas Gautam,Grishma Gupta,Zhelong Li,R. Jacobs,Dane Morgan
Published 2022 in npj Computational Materials
ABSTRACT
PUBLICATION RECORD
- Publication year
2022
- Venue
npj Computational Materials
- Publication date
2022-05-20
- Fields of study
Not labeled
- Identifiers
- External record
- Source metadata
Semantic Scholar
CITATION MAP
EXTRACTION MAP
CLAIMS
- No claims are published for this paper.
CONCEPTS
- No concepts are published for this paper.
REFERENCES
Showing 1-44 of 44 references · Page 1 of 1
CITED BY
Showing 1-62 of 62 citing papers · Page 1 of 1