Best practices for machine-learned force fields

Using the machine-learning–force-fields method, VASP can construct force fields based on ab-initio simulations. There are many aspects to carefully consider while constructing, testing, retraining, and applying a force field. Here, we list some best practices but bare in mind that the list is incomplete and the method has not been applied to a large number of systems. Therefore, we recommend using the usual rigorous checking that is necessary for any research project.

Training

It is generally possible to train force fields on a smaller unit cell and then apply it to a larger system.
It is important to learn the correct forces. This requires converging the forces with respect to the number of k points.
Switch off the symmetry as in MD runs.
If possible, heat up the system gradually, i.e., use temperature ramping from a low temperature to a temperature approximately 30% higher than the desired application temperature. Be careful not to induce phase transitions when heating above the target temperature.
If possible, prefer MD training runs in the NpT ensemble, the additional cell fluctuations improve the robustness of the resulting force field. However, for liquids allow only scaling of the supercell (no change of angles, no individual lattice vector scaling) because otherwise the cell may collapse. This can be achieved with the ICONST file, see 3) and 4) here.
For simulations without fixed lattice, restart often to reinitialize the PAW basis of the KS orbitals and avoid Pulay stress.
When the system contains different components, train them separately first. For instance, when the system has a surface of a crystal and a molecule binding on that surface. First, train the bulk crystal, then the surface, next the isolated molecule, and finally the entire system.

Monitoring

Always monitor the current prediction errors, Bayesian error estimation and error threshold by searching the ML_LOGFILE for ERR (4th column), BEEF (4th column) and THRUPD (4th column), respectively.

Tuning on-the-fly parameters

In case too many or too few training structures are selected there are some on-the-fly parameters which can be tuned:

If not enough or too many local reference structures are extracted from the training data there are additional tags for tuning:

ML_EPS_LOW

Testing

Set up an independent test set of random configurations. Then, check the average errors comparing forces, stresses and energy differences of two structures based on DFT and predicted by machine-learned force fields.

If you have both, ab-initio reference data and a calculation using force fields, check the agreement of some physical properties. For instance, you might check the relaxed lattice parameters, phonons, relative energies of different phases, the elastic constant, the formation of defects etc.

plot blocked averages

Retraining

Application

set the ab-initio parameters to small values. VASP cannot circumvent the initialization of KS orbitals although they are not used during the MD run with MLFF