MatFold: systematic insights into materials discovery models' performance through standardized cross-validation protocols

07 August 2024, Version 1
This content is a preprint and has not undergone peer review at the time of posting.

Abstract

Machine learning (ML) models in the materials sciences that are validated by overly simplistic cross-validation (CV) protocols can yield biased performance estimates for downstream modeling or materials screening tasks. This can be particularly counterproductive for applications where the time and cost of failed validation efforts (experimental synthesis, characterization, and testing) are consequential. We propose a set of standardized and increasingly difficult splitting protocols for chemically and structurally motivated CV that can be followed to validate any ML model for materials discovery. Among several benefits, this enables systematic insights into model generalizability, improvability, and uncertainty, provides benchmarks for fair comparison between competing models with access to differing quantities of data, and systematically reduces possible data leakage through increasingly strict splitting protocols. A general-purpose, model-agnostic toolkit, MatFold, is provided to automate the construction of these CV splits and encourage further community use.

Keywords

Machine Learning
Cross Validation
Defects
Work Function
Software

Supplementary materials

Title
Description
Actions
Title
Supplementary Information
Description
Supplementary Information to the main text, including additional CV analysis showing inference performance for additional hold-out strategies, as well as MAE heatmaps and parity plots for leave-one-element-out splits.
Actions
Title
Supplementary Files
Description
The aggregated vacancy formation energy data set used in this work.
Actions

Comments

Comments are not moderated before they are posted, but they can be removed by the site moderators if they are found to be in contravention of our Commenting Policy [opens in a new tab] - please read this policy before you post. Comments should be used for scholarly discussion of the content in question. You can find more information about how to use the commenting feature here [opens in a new tab] .
This site is protected by reCAPTCHA and the Google Privacy Policy [opens in a new tab] and Terms of Service [opens in a new tab] apply.