Training a Foundation Model for Materials on a Budget
Koker, Teddy, Kotak, Mit, Smidt, Tess
–arXiv.org Artificial Intelligence
Foundation models for materials modeling are advancing quickly, but their training remains expensive, often placing state-of-the-art methods out of reach for many research groups. We introduce Nequix, a compact E(3)-equivariant potential that pairs a simplified NequIP design with modern training practices, including equivariant root-mean-square layer normalization and the Muon optimizer, to retain accuracy while substantially reducing compute requirements. Nequix has 700K parameters and was trained in 100 A100 GPU-hours. On the Matbench-Discovery and MDR Phonon benchmarks, Nequix ranks third overall while requiring a 20 times lower training cost than most other methods, and it delivers two orders of magnitude faster inference speed than the current top-ranked model.
arXiv.org Artificial Intelligence
Oct-10-2025
- Country:
- Asia > Middle East
- Jordan (0.05)
- North America > United States
- Massachusetts > Middlesex County > Cambridge (0.14)
- Asia > Middle East
- Genre:
- Research Report (1.00)
- Technology: