Prediction rigidities for data-driven chemistry

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chong, Sanggyu, Bigi, Filippo, Grasselli, Federico, Loche, Philip, Kellner, Matthias, Ceriotti, Michele
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914924165332992
author Chong, Sanggyu
Bigi, Filippo
Grasselli, Federico
Loche, Philip
Kellner, Matthias
Ceriotti, Michele
author_facet Chong, Sanggyu
Bigi, Filippo
Grasselli, Federico
Loche, Philip
Kellner, Matthias
Ceriotti, Michele
contents The widespread application of machine learning (ML) to the chemical sciences is making it very important to understand how the ML models learn to correlate chemical structures with their properties, and what can be done to improve the training efficiency whilst guaranteeing interpretability and transferability. In this work, we demonstrate the wide utility of prediction rigidities, a family of metrics derived from the loss function, in understanding the robustness of ML model predictions. We show that the prediction rigidities allow the assessment of the model not only at the global level, but also on the local or the component-wise level at which the intermediate (e.g. atomic, body-ordered, or range-separated) predictions are made. We leverage these metrics to understand the learning behavior of different ML models, and to guide efficient dataset construction for model training. We finally implement the formalism for a ML model targeting a coarse-grained system to demonstrate the applicability of the prediction rigidities to an even broader class of atomistic modeling problems.
format Preprint
id arxiv_https___arxiv_org_abs_2408_14311
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Prediction rigidities for data-driven chemistry
Chong, Sanggyu
Bigi, Filippo
Grasselli, Federico
Loche, Philip
Kellner, Matthias
Ceriotti, Michele
Chemical Physics
The widespread application of machine learning (ML) to the chemical sciences is making it very important to understand how the ML models learn to correlate chemical structures with their properties, and what can be done to improve the training efficiency whilst guaranteeing interpretability and transferability. In this work, we demonstrate the wide utility of prediction rigidities, a family of metrics derived from the loss function, in understanding the robustness of ML model predictions. We show that the prediction rigidities allow the assessment of the model not only at the global level, but also on the local or the component-wise level at which the intermediate (e.g. atomic, body-ordered, or range-separated) predictions are made. We leverage these metrics to understand the learning behavior of different ML models, and to guide efficient dataset construction for model training. We finally implement the formalism for a ML model targeting a coarse-grained system to demonstrate the applicability of the prediction rigidities to an even broader class of atomistic modeling problems.
title Prediction rigidities for data-driven chemistry
topic Chemical Physics
url https://arxiv.org/abs/2408.14311