Full-data accuracy with fewer labels for training and fine-tuning machine-learning force fields
Sheng Bi, Yi-Ze Wang, Jun Cheng
Abstract
Machine-learning force fields (MLFFs) are reliable only near their training distribution, making efficient construction of diverse training sets a major bottleneck for both train-from-scratch and foundation fine-tuning workflows. Active learning can reduce this cost, but standard model-committee uncertainty is impractical for foundation MLFFs because each committee member requires a separate fine-tuning run. We present an active-learning workflow based on last-layer-projection regression (LLPR), a forward-pass-cheap per-configuration uncertainty estimator. Across molecular, condensed-phase, and electrolyte systems, LLPR identifies compact, high-value training sets that recover full-data accuracy using only a small fraction of electronic-structure labels. In foundation-model fine-tuning, LLPR-selected configurations reach the full-pool fine-tuning ceiling with substantially fewer labels than random selection. In iterative electrolyte fine-tuning, LLPR detects unphysical local coordination before DFT labelling, provides an absolute force-error threshold, and enables automatic termination of the learning loop. The resulting models reproduce reference density and ion-coordination structure, providing a scalable uncertainty-quantification strategy across MLFF training regimes.
Create a lesson
Related papers
Real-Time Emergence of Charge-Transfer-to-Solvent States from Core Excitation
Jiří Suchan, B. Scott Fales, Benjamin G. Levine et al.
ElemCo.jl: A Julia package for electron-correlation methods
Daniel Kats, Charlotte Rickert, Thomas Schraivogel et al.
Franson-Interferometric Bounds on Entangled Two-Photon Absorption
Albin Hedse, Sankaran Ramesh, Luis Matheis et al.
The off-diagonal low rank property: new opportunities for low-scaling computational chemistry methods
Zikuan Wang
Core-valence double ionization of SF6 involving S2p, F1s and S1s inner shells
Veronica Daver Ideböhn, Daniel M. Pereira, Lucas M. Cornetta et al.
Benchmark of Multi-Channel Dyson Equation and Algebraic Diagrammatic Construction Methods for molecules
Mike Keizer, Stefano Paggi, J. Arjan Berger et al.