Robust and Interpretable Adaptation of Equivariant Materials Foundation Models via Sparsity-promoting Fine-tuning
Not on the map. Why?
Abstract (arXiv)
Pre-trained materials foundation models, or machine learning interatomic potentials, leverage general physicochemical knowledge to effectively approximate potential energy surfaces. However, they often require domain-specific calibration due to physicochemical diversity as well as mismatches between practical computational settings and those used in constructing the pre-training data. To address this, we propose a sparsity-promoting fine-tuning method that selectively updates model parameters by exploiting the structural properties of E(3)-equivariant materials foundation models. On energy and force prediction tasks across molecular and crystalline benchmarks, our method matches or surpasses full fine-tuning and equivariant low-rank adaptation while updating only ~3~% of parameters, and in some cases as little as ~0.5~%. Beyond energy and force calibration, we further demonstrate task generalizability by applying our method to magnetic moment prediction and magnetism-aware total energy modeling. Finally, analysis of sparsity patterns reveals physically interpretable signatures, such as enhanced d-orbital contributions in transition metal systems. Overall, our results establish sparsity-promoting fine-tuning as a flexible and interpretable method for domain specialization of equivariant materials foundation models.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 39 of its 85 references that are in the map. First posted on arXiv 17 Jun 2026.
Builds on
39 of 85 references.
References · 39
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2022BLIP: Bootstrapping Language-Image Pre-training for Unified Vision-Language Understanding and Generationcited by 4,322
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,236
- 2017Neural Message Passing for Quantum Chemistrycited by 3,487
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,262
- 2015Learning both Weights and Connections for Efficient Neural Networkscited by 2,031
- 2016Pruning Filters for Efficient ConvNetscited by 1,143
- 2021E(n) Equivariant Graph Neural Networkscited by 965
- 2018Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point cloudscited by 746
- 2021E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentialscited by 708
- 2018Rethinking the Value of Network Pruningcited by 599
27 more
- 2017Learning Sparse Neural Networks through L0 Regularizationcited by 583
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 571
- 2016Learning Structured Sparsity in Deep Neural Networkscited by 541
- 2021Sparsity in Deep Learning: Pruning and growth for efficient inference and training in neural networkscited by 527
- 2022MACE: Higher Order Equivariant Message Passing Neural Networks for Fast and Accurate Force Fieldscited by 474
- 2022Equiformer: Equivariant Graph Attention Transformer for 3D Atomistic Graphscited by 264
- 2022Learning Local Equivariant Representations for Large-Scale Atomistic Dynamicscited by 256
- 2022e3nn: Euclidean Neural Networkscited by 174
- 2018Clebsch-Gordan Nets: a Fully Fourier Space Spherical Convolutional Neural Networkcited by 171
- 2024The Road Less Scheduledcited by 163
- 2019Drawing Early-Bird Tickets: Towards More Efficient Training of Deep Networkscited by 160
- 2020Soft Threshold Weight Reparameterization for Learnable Sparsitycited by 151
- 2020SINDy-PI: A Robust Algorithm for Parallel Implicit Sparse Identification of Nonlinear Dynamicscited by 125
- 2017Bayesian Compression for Deep Learningcited by 116
- 2015Data-free parameter pruning for Deep Neural Networkscited by 112
- 2019Winning the Lottery with Continuous Sparsificationcited by 100
- 2021Sparse DETR: Efficient End-to-End Object Detection with Learnable Sparsitycited by 98
- 2022SPICE, A Dataset of Drug-like Molecules and Peptides for Training Machine Learning Potentialscited by 82
- 2020Dynamic Sparse Training: Find Efficient Sparse Network From Scratch With Trainable Masked Layerscited by 68
- 2024Orb: A Fast, Scalable Neural Network Potentialcited by 66
- 2021Efficient Lottery Ticket Finding: Less Data is Morecited by 44
- 2020Learned Threshold Pruningcited by 17
- 2025High-performance training and inference for deep equivariant interatomic potentialscited by 17
- 2024Equivariant Graph Network Approximations of High-Degree Polynomials for Force Field Predictioncited by 6
- 2025MatterTune: An Integrated, User-Friendly Platform for Fine-Tuning Atomistic Foundation Models to Accelerate Materials Simulation and Discoverycited by 5
- 2025Scalable Data-Driven Basis Selection for Linear Machine Learning Interatomic Potentialscited by 1
- 2025GeoAda: Efficiently Finetune Geometric Diffusion Models with Equivariant Adapterscited by 1
Built on by
Not cited yet.