Online test-time adaptation for better generalization of interatomic potentials to out-of-distribution data

Bibliographic Details
Title: Online test-time adaptation for better generalization of interatomic potentials to out-of-distribution data
Authors: Taoyong Cui, Chenyu Tang, Dongzhan Zhou, Yuqiang Li, Xingao Gong, Wanli Ouyang, Mao Su, Shufei Zhang
Source: Nature Communications, Vol 16, Iss 1, Pp 1-11 (2025)
Publisher Information: Nature Portfolio, 2025.
Publication Year: 2025
Collection: LCC:Science
Subject Terms: Science
More Details: Abstract Machine learning interatomic potentials (MLIPs) enable more efficient molecular dynamics (MD) simulations with ab initio accuracy, which have been used in various domains of physical science. However, distribution shift between training and test data causes deterioration of the test performance of MLIPs, and even leads to collapse of MD simulations. In this work, we propose an online Test-time Adaptation Interatomic Potential (TAIP) framework to improve the generalization on test data. Specifically, we design a dual-level self-supervised learning approach that leverages global structure and atomic local environment information to align the model with the test data. Extensive experiments demonstrate TAIP’s capability to bridge the domain gap between training and test dataset without additional data. TAIP enhances the test performance on various benchmarks, from small molecule datasets to complex periodic molecular systems with various types of elements. TAIP also enables stable MD simulations where the corresponding baseline models collapse.
Document Type: article
File Description: electronic resource
Language: English
ISSN: 2041-1723
Relation: https://doaj.org/toc/2041-1723
DOI: 10.1038/s41467-025-57101-4
Access URL: https://doaj.org/article/5e1c6fb61c814ed480ec78e4b04f029d
Accession Number: edsdoj.5e1c6fb61c814ed480ec78e4b04f029d
Database: Directory of Open Access Journals
More Details
ISSN:20411723
DOI:10.1038/s41467-025-57101-4
Published in:Nature Communications
Language:English