Package: mbGDML
mbGDMLCreate, use, and analyze machine learning potentials within the many-body expansion framework.DocumentationMotivation•Approach•Features•Installation•LicenseMotivationMachine learning potentials (i.e., force fields) often rely on local descriptors for size transferability.
These descriptors partition total properties into atomic contributions; however, they inherently neglect complicated long-range interactions by enforcing atomic radial cutoffs.
Global descriptors encode the entire structure with no cutoffs and can capture interactions at all scales.
However, they are restricted to systems with the same number of atoms.Gradient-domain machine learning(GDML) is one example of a ML potential with a global descriptor.
GDML is unique because it trains directly on forces and recovers total energy through analytical integration.
This provides substantially more information about the potential energy surface (PES) and allows for better interpolation between training data.
As a result, GDML typically only needs 1000 structures to accurately learn energies and forces.To date, GDML has been limited to the exact system it was trained on.
This makes simulations on arbitrarily size systems, like solvents, futile.ApproachMany-body expansions (MBEs) rigorously decomposes total (i.e., supersystem) energies into fundamentaln-body interactions.
This expansion is formally exact when allN-body interactions are accounted for.
In practice, however, it is typically truncated to the third order.
One can then model any system by summing up 1-, 2-, and 3-body contributions.MBEs driven by GDML potentials trained onn-body interactions is a promising approach for size-transferable potentials.
Furthermore, GDML model's remarkable data efficiency enables training on highly accurate quantum chemical methods.FeaturesTrainTrain GDML models using grid searches, Bayesian optimization, or both on CPUs.Custom loss functions.Iterative training procedure for automated curation of optimal training sets.PredictMany-body predictions with GDML,SchNetandGAPpotentials.Parallel GDML predictions withrayfrom a laptop to multiple nodes.Periodic structures with the minimum-image convention.Alchemical predictions by tuning out 2- or 3-body contributions of specific entities.AnalysisPrediction sets that store decomposed predictions for further analysis.Radial distribution functions.Cluster and identify problematic (i.e., high error) structures usingsklearn.InterfacesAtomic Simulation Environment(ASE) for geometry optimizations, molecular dynamics simulations, and more.InstallationYou can install mbGDML fromPyPIby usingpip install mbGDML.
Or, the latest development version can be installed directly from theGitHub repositoryor fromTestPyPI.git clone https://github.com/keithgroup/mbGDML
cd mbGDML
pip install .Citing this workIf you find this code helpful in your research or project, please consider citing the following paper:Maldonado, A. M.; Poltavsky, I.; Vassilev-Galindo, V.; Tkatchenko, A.; Keith, J. A. Modeling molecular ensembles with gradient-domain machine learning force fields.Digital Discovery2023,2(3), 871-880. DOI:10.1039/D3DD00011G.@article{maldonado2023modeling,title={Modeling molecular ensembles with gradient-domain machine learning force fields},author={Maldonado, Alex M and Poltavsky, Igor and Vassilev-Galindo, Valentin and Tkatchenko, Alexandre and Keith, John A},journal={Digital Discovery},volume={2},number={3},pages={871--880},year={2023},publisher={Royal Society of Chemistry},doi={10.1039/D3DD00011G}}Citing the paper helps acknowledge the effort put into developing and maintaining this codebase, and it provides a way to support further research and development.
Thank you for your support!LicenseDistributed under the MIT License. SeeLICENSEfor more information.