A Study on the Fine-Tuning Performance of Universal Machine-Learned Interatomic Potentials (U-MLIPs)

Abstract

Universal machine-learned interatomic potentials (U-MLIPs) have demonstrated effectiveness across diverse atomistic systems but often require fine-tuning for task-specific accuracy. We investigate the fine-tuning of two MACE-based foundation models, MACE-MP-0 and its variant MACE-MP-0b, and identify key insights. Fine-tuning on task-specific datasets enhances accuracy and, in some cases, outperforms models trained from scratch. Additionally, fine-tuned models benefit from faster convergence due to the strong initial predictions provided by the foundation model. The success of fine-tuning also depends on careful dataset selection, which can be optimized through filtering or active learning. We further discuss practical strategies for achieving better fine-tuning foundation models in atomistic simulations and explore future directions for their development and applications.

Abstract: arXiv descriptive metadata (CC0). Thank you to arXiv for use of its open access interoperability. This service was not reviewed or approved by, nor does it necessarily express or reflect the policies or opinions of, arXiv.