Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

A recipe for scalable attention-based MLIPs: unlocking long-range accuracy with all-to-all node attention

About

Machine-learning interatomic potentials (MLIPs) have advanced rapidly, with many top models relying on strong physics-based inductive biases. However, as models scale to larger systems like biomolecules and electrolytes, they struggle to accurately capture long-range (LR) interactions, leading current approaches to rely on explicit physics-based terms or components. In this work, we propose AllScAIP, a straightforward, attention-based, and energy-conserving MLIP model that scales to O(100 million) training samples. It addresses the long-range challenge using an all-to-all node attention component that is data-driven. Extensive ablations reveal that in low-data/small-model regimes, inductive biases improve sample efficiency. However, as data and model size scale, these benefits diminish or even reverse, while all-to-all attention remains critical for capturing LR interactions. Our model achieves state-of-the-art energy/force accuracy on molecular systems, as well as a number of physics-based evaluations (OMol25), while being competitive on materials (OMat24) and catalysts (OC20). Furthermore, it enables stable, long-timescale MD simulations that accurately recover experimental observables, including density and heat of vaporization predictions.

Eric Qu, Brandon M. Wood, Aditi S. Krishnapriyan, Zachary W. Ulissi• 2026

Related benchmarks

TaskDatasetResultRank
Energy, Force, and Stress PredictionOMat24 (val)
Energy per Atom10.7
21
Molecular energy predictionOMol25 (test)
Average Rank3.43
17
Catalysis PredictionOC20 Total Energy OOD-Both (val)
Energy92.2
9
Molecular property predictionOMol25 (val)
Energy MAE (Biomol.) (meV)22.25
7
Molecular energy and force predictionOMol25 Full 102M 1.0 (All)
Biomolecules Energy MAE (meV)0.15
7
Energy and force predictionSPICE PubChem (test)
Energy MAE (meV/atom)0.13
6
Energy and force predictionSPICE DES370K Monomers (test)
Energy/Atom MAE (meV)0.07
6
Energy and force predictionSPICE DES370K Dimers (test)
Energy MAE / Atom (meV)0.06
6
Energy and force predictionSPICE Dipeptides (test)
Energy/Atom MAE (meV)0.06
6
Energy and force predictionSPICE Solvated Amino Acids (test)
Energy MAE (meV/atom)0.13
6
Showing 10 of 15 rows

Other info

Follow for update