Pre-training, fine-tuning, and distillation (PFD): Automatically generating machine learning force fields from universal models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Ruoyu, Gao, Yuxiang, Wu, Hongyu, Zhong, Zhicheng
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866908702619992064
author Wang, Ruoyu
Gao, Yuxiang
Wu, Hongyu
Zhong, Zhicheng
author_facet Wang, Ruoyu
Gao, Yuxiang
Wu, Hongyu
Zhong, Zhicheng
contents Universal force fields generalizable across the periodic table represent a new trend in computational materials science. However, the applications of universal force fields in material simulations are limited by their slow inference speed and the lack of first-principles accuracy. Instead of building a single model simultaneously satisfying these characteristics, a strategy that quickly generates material-specific models from the universal model may be more feasible. Here, we propose a new workflow pattern, PFD (Pre-training, Fine-tuning, and Distillation), which automatically generates machine-learning force fields for specific materials from a pre-trained universal model through fine-tuning and distillation. By fine-tuning the pre-trained model, our PFD workflow generates force fields with first-principles accuracy while requiring one to two orders of magnitude less training data compared to traditional methods. The inference speed of the generated force field is further improved through distillation, meeting the requirements of large-scale molecular simulations. Comprehensive testing across diverse materials including complex systems, such as amorphous carbon, interface, etc., reveals marked enhancements in training efficiency, which suggests the PFD workflow a practical and reliable approach for force field generation in computational material sciences.
format Preprint
id arxiv_https___arxiv_org_abs_2502_20809
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Pre-training, fine-tuning, and distillation (PFD): Automatically generating machine learning force fields from universal models
Wang, Ruoyu
Gao, Yuxiang
Wu, Hongyu
Zhong, Zhicheng
Materials Science
Computational Physics
Universal force fields generalizable across the periodic table represent a new trend in computational materials science. However, the applications of universal force fields in material simulations are limited by their slow inference speed and the lack of first-principles accuracy. Instead of building a single model simultaneously satisfying these characteristics, a strategy that quickly generates material-specific models from the universal model may be more feasible. Here, we propose a new workflow pattern, PFD (Pre-training, Fine-tuning, and Distillation), which automatically generates machine-learning force fields for specific materials from a pre-trained universal model through fine-tuning and distillation. By fine-tuning the pre-trained model, our PFD workflow generates force fields with first-principles accuracy while requiring one to two orders of magnitude less training data compared to traditional methods. The inference speed of the generated force field is further improved through distillation, meeting the requirements of large-scale molecular simulations. Comprehensive testing across diverse materials including complex systems, such as amorphous carbon, interface, etc., reveals marked enhancements in training efficiency, which suggests the PFD workflow a practical and reliable approach for force field generation in computational material sciences.
title Pre-training, fine-tuning, and distillation (PFD): Automatically generating machine learning force fields from universal models
topic Materials Science
Computational Physics
url https://arxiv.org/abs/2502.20809