Learning atomic forces from uncertainty-calibrated adversarial attacks

Abstract

Adversarial approaches, which intentionally challenge machine learning models by generating difficult examples, are increasingly being adopted to improve machine learning interatomic potentials (MLIPs). While already providing great practical value, little is known about the actual prediction errors of MLIPs on adversarial structures and whether these errors can be controlled. We propose the Calibrated Adversarial Geometry Optimization (CAGO) algorithm to discover adversarial structures with user-assigned errors. Through uncertainty calibration, the estimated uncertainty of MLIPs is unified with real errors. By performing geometry optimization for calibrated uncertainty, we reach adversarial structures with the user-assigned target MLIP prediction error. Integrating with active learning pipelines, we benchmark CAGO, demonstrating stable MLIPs that systematically converge structural, dynamical, and thermodynamical properties for liquid water and water adsorption in a metal-organic framework within only hundreds of training structures, where previously many thousands were typically required.

References in corpus (11)

• SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
• E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentials
• DP-GEN: A concurrent learning platform for the generation of reliable deep learning based potential energy models
• Less is more: sampling chemical space with active learning
• DeePMD-kit v2: A software package for Deep Potential models
• Neural Network Based in Silico Simulation of Combustion Reactions
• Forces are not Enough: Benchmark and Critical Evaluation for Machine Learning Force Fields with Molecular Simulations
• Fast and Accurate Uncertainty Estimation in Chemical Machine Learning
• Uncertainty estimation for molecular dynamics and sampling
• Differentiable sampling of molecular geometries with uncertainty-based adversarial attacks
• Uncertainty-biased molecular dynamics for learning uniformly accurate interatomic potentials