On The Finetuning of MLIPs Through the Lens of Iterated Maps With BPTT

Evan Dramko, Yizhi Zhu, Geoffroy Hautier, Anastasios Kyrillidis, Chris Jermaine, Thomas Reps, Aleksandar Krivokapic

Abstract:

Accurate structural relaxation is critical for advanced materials design.
Traditional approaches built on physics-derived first-principles calculations are computationally expensive, motivating the creation of machine-learning interatomic potentials (MLIPs), which strive to faithfully reproduce first-principles computed forces.
We propose a fine-tuning method to be used on a pretrained MLIP in which we create a fully-differentiable end-to-end simulation loop that optimizes the predicted final structures directly.
Trajectories are unrolled and gradients are tracked through the entire relaxation.
We show that this method consistently improves performance across all evaluated pretrained models; resulting in an average of roughly $32 \%$ reduction in prediction error.
Interestingly, we show the process is robust to substantial variation in the relaxation setup, achieving negligibly different results across varied hyperparameter and procedural modifications.