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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Energy, Force, and Stress Prediction | OMat24 (val) | Energy per Atom10.7 | 21 | |
| Molecular energy prediction | OMol25 (test) | Average Rank3.43 | 17 | |
| Catalysis Prediction | OC20 Total Energy OOD-Both (val) | Energy92.2 | 9 | |
| Molecular property prediction | OMol25 (val) | Energy MAE (Biomol.) (meV)22.25 | 7 | |
| Molecular energy and force prediction | OMol25 Full 102M 1.0 (All) | Biomolecules Energy MAE (meV)0.15 | 7 | |
| Energy and force prediction | SPICE PubChem (test) | Energy MAE (meV/atom)0.13 | 6 | |
| Energy and force prediction | SPICE DES370K Monomers (test) | Energy/Atom MAE (meV)0.07 | 6 | |
| Energy and force prediction | SPICE DES370K Dimers (test) | Energy MAE / Atom (meV)0.06 | 6 | |
| Energy and force prediction | SPICE Dipeptides (test) | Energy/Atom MAE (meV)0.06 | 6 | |
| Energy and force prediction | SPICE Solvated Amino Acids (test) | Energy MAE (meV/atom)0.13 | 6 |