A recipe for scalable attention-based MLIPs: unlocking long-range accuracy with all-to-all node attention
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 24 of its 39 references that are in the map. First posted on arXiv 6 Mar 2026.
Builds on
24 of 39 references.
References · 24
- 2021E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentialscited by 701
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 569
- 2020Directional Message Passing for Molecular Graphscited by 556
- 2022MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fieldscited by 467
- 2021GemNet: Universal Directional Graph Neural Networks for Moleculescited by 342
- 2022Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphscited by 258
- 2020The Open Catalyst 2020 (OC20) Dataset and Community Challengescited by 246
- 2020Machine Learning Force Fieldscited by 204
- 2023EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representationscited by 201
- 2023Reducing SO(3) Convolutions to SO(2) for Efficient Equivariant GNNscited by 113
- 2025UMA: A Family of Universal Models for Atomscited by 99
- 2022The Open Catalyst 2022 (OC22) Dataset and Challenges for Oxide Electrocatalystscited by 95
12 more
- 2022SPICE, A Dataset of Drug-like Molecules and Peptides for Training Machine Learning Potentialscited by 80
- 2025Learning Smooth and Expressive Interatomic Potentials for Physical Property Predictioncited by 79
- 2022GemNet-OC: Developing Graph Neural Networks for Large and Diverse Molecular Simulation Datasetscited by 61
- 2022Spherical Channels for Modeling Atomic Interactionscited by 47
- 2024The Importance of Being Scalable: Improving the Speed and Accuracy of Neural Network Interatomic Potentials Across Chemical Domainscited by 40
- 2024Latent Ewald summation for machine learning of long-range interactionscited by 26
- 2025PET-MAD, a lightweight universal interatomic potential for advanced materials modelingcited by 23
- 2024A predictive machine learning force field framework for liquid electrolyte developmentcited by 13
- 2024Euclidean Fast Attention -- Machine Learning Global Atomic Representations at Linear Costcited by 12
- 2025A universal augmentation framework for long-range electrostatics in machine learning interatomic potentialscited by 9
- 2025Transformers Discover Molecular Structure Without Graph Priorscited by 6
- 2026From Evaluation to Design: Using Potential Energy Surface Smoothness Metrics to Guide Machine Learning Interatomic Potential Architecturescited by 1
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026Beyond Adam: SOAP and Muon for Faster, Label-Efficient Training of Machine Learning Interatomic Potentialscited by 3
- 2026Knowing when to trust machine-learned interatomic potentialscited by 1
- 2026Compact SO(3) Equivariant Atomistic Foundation Models via Structural Pruningcited by 0
- 2026Distilling latent electrostatics from foundation machine learning interatomic potentialscited by 0
- 2026E3J: An Efficient and Open-Source Backend for Euclidean Equivariant Operations on GPU and TPUcited by 0
- 2026A strategic roadmap for an atomistic machine-learning ecosystemcited by 0
- 2026NEMORA: Neural Equivariant Multipole Operators for Long-Range Atomistic Learningcited by 0
- 2026NEMORA: Neural Equivariant Multipole Operators for Long-Range Atomistic Learning7 Oct 2026
- 2026A strategic roadmap for an atomistic machine-learning ecosystem30 Sep 2026
- 2026E3J: An Efficient and Open-Source Backend for Euclidean Equivariant Operations on GPU and TPU28 Sep 2026
- 2026Beyond Adam: SOAP and Muon for Faster, Label-Efficient Training of Machine Learning Interatomic Potentials2 Jul 2026
- 2026Distilling latent electrostatics from foundation machine learning interatomic potentials12 Jun 2026
- 2026Compact SO(3) Equivariant Atomistic Foundation Models via Structural Pruning9 May 2026
- 2026Knowing when to trust machine-learned interatomic potentials1 May 2026