Transfer learning for chemically accurate interatomic neural network potentials
Viktor Zaverkin, David Holzmüller, Luca Bonfirraro, Johannes Kästner
Abstract
data. Additionally, we show that fine-tuning with energy labels alone can suffice to obtain accurate atomic forces and run large-scale atomistic simulations, provided a well-designed fine-tuning data set. We also investigate possible limitations of transfer learning, especially regarding the design and size of the pre-training and fine-tuning data sets. Finally, we provide GM-NN potentials pre-trained and fine-tuned on the ANI-1x and ANI-1ccx data sets, which can easily be fine-tuned on and applied to organic molecules.
Topics & Concepts
Artificial neural networkComputer scienceDiscriminative modelTransfer of learningSet (abstract data type)Ab initioInteratomic potentialData setWork (physics)Artificial intelligenceMachine learningBiological systemChemistryMolecular dynamicsComputational chemistryPhysicsQuantum mechanicsOrganic chemistryProgramming languageBiologyMachine Learning in Materials ScienceComputational Drug Discovery MethodsCrystallography and molecular interactions