mlsanityData to 7 Oct 2026

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

arXiv 2603.06567 ↗First posted 6 Mar 2026cs.LG cond-mat.mtrl-sci cs.CEcited by 7

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

  1. 2021E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentialscited by 701
  2. 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 569
  3. 2020Directional Message Passing for Molecular Graphscited by 556
  4. 2022MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fieldscited by 467
  5. 2021GemNet: Universal Directional Graph Neural Networks for Moleculescited by 342
  6. 2022Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphscited by 258
  7. 2020The Open Catalyst 2020 (OC20) Dataset and Community Challengescited by 246
  8. 2020Machine Learning Force Fieldscited by 204
  9. 2023EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree Representationscited by 201
  10. 2023Reducing SO(3) Convolutions to SO(2) for Efficient Equivariant GNNscited by 113
  11. 2025UMA: A Family of Universal Models for Atomscited by 99
  12. 2022The Open Catalyst 2022 (OC22) Dataset and Challenges for Oxide Electrocatalystscited by 95
12 more
  1. 2022SPICE, A Dataset of Drug-like Molecules and Peptides for Training Machine Learning Potentialscited by 80
  2. 2025Learning Smooth and Expressive Interatomic Potentials for Physical Property Predictioncited by 79
  3. 2022GemNet-OC: Developing Graph Neural Networks for Large and Diverse Molecular Simulation Datasetscited by 61
  4. 2022Spherical Channels for Modeling Atomic Interactionscited by 47
  5. 2024The Importance of Being Scalable: Improving the Speed and Accuracy of Neural Network Interatomic Potentials Across Chemical Domainscited by 40
  6. 2024Latent Ewald summation for machine learning of long-range interactionscited by 26
  7. 2025PET-MAD, a lightweight universal interatomic potential for advanced materials modelingcited by 23
  8. 2024A predictive machine learning force field framework for liquid electrolyte developmentcited by 13
  9. 2024Euclidean Fast Attention -- Machine Learning Global Atomic Representations at Linear Costcited by 12
  10. 2025A universal augmentation framework for long-range electrostatics in machine learning interatomic potentialscited by 9
  11. 2025Transformers Discover Molecular Structure Without Graph Priorscited by 6
  12. 2026From Evaluation to Design: Using Potential Energy Surface Smoothness Metrics to Guide Machine Learning Interatomic Potential Architecturescited by 1

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 7.

7 papers.