PyXtal FF: a Python Library for Automated Force Field Generation

Abstract

We present PyXtal FF, a package based on Python programming language, for developing machine learning potentials (MLPs). The aim of PyXtal FF is to promote the application of atomistic simulations by providing several choices of structural descriptors and machine learning regressions in one platform. Based on the given choice of structural descriptors (including the atom-centered symmetry functions, embedded atom density, SO4 bispectrum, and smooth SO3 power spectrum), PyXtal FF can train the MLPs with either the generalized linear regression or neural networks model, by simultaneously minimizing the errors of energy/forces/stress tensors in comparison with the data from the ab-initio simulation. The trained MLP model from PyXtal FF is interfaced with the Atomic Simulation Environment (ASE) package, which allows different types of light-weight simulations such as geometry optimization, molecular dynamics simulation, and physical properties prediction. Finally, we will illustrate the performance of PyXtal FF by applying it to investigate several material systems, including the bulk SiO2, high entropy alloy NbMoTaW, and elemental Pt for general purposes. Full documentation of PyXtal FF is available at https://pyxtal-ff.readthedocs.io.

arXiv comment13 pages, 4 figures

References in corpus (23)

• Adam: A Method for Stochastic Optimization
• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• The NumPy array: a structure for efficient numerical computation
• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• On representing chemical environments
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• SchNet - a deep learning architecture for molecules and materials
• Moment Tensor Potentials: a class of systematically improvable interatomic potentials
• A Spectral Analysis Method for Automated Generation of Quantum-Accurate Interatomic Potentials
• Band structure diagram paths based on crystallography
• DScribe: Library of Descriptors for Machine Learning in Materials Science
• Accelerating crystal structure prediction by machine-learning interatomic potentials with active learning
• Automatic Selection of Atomic Fingerprints and Reference Configurations for Machine-Learning Potentials
• WACSF - Weighted Atom-Centered Symmetry Functions as Descriptors in Machine Learning Potentials
• Embedded Atom Neural Network Potentials: Efficient and Accurate Machine Learning with a Physically Inspired Representation
• Data-driven learning of total and local energies in elemental boron
• Accurate Force Field for Molybdenum by Machine Learning Large Materials Data
• Quantum-Accurate Spectral Neighbor Analysis Potential Models for Ni-Mo Binary Alloys and FCC Metals
• Stratified construction of neural network based interatomic models for multicomponent materials
• PiNN: A Python Library for Building Atomic Neural Networks of Molecules and Materials
• Neural Networks Potential from the Bispectrum Component: A Case Study on Crystalline Silicon
• Complex Strengthening Mechanisms in the NbMoTaW Multi-Principal Element Alloy
• Spectral Neural Network Potentials for Binary Alloys

Cited by in corpus (7)

• Neuroevolution machine learning potentials: Combining high accuracy and low cost in atomistic simulations and application to heat transport
• GPUMD: A package for constructing accurate machine-learned potentials and performing highly efficient atomistic simulations
• Improving the accuracy of the neuroevolution machine learning potential for multi-component systems
• Short-range order and its impacts on the BCC NbMoTaW multi-principal element alloy by the machine-learning potential
• KLIFF: A framework to develop physics-based and machine learning interatomic potentials
• Size-Dependent Nucleation in Crystal Phase Transition from Machine Learning Metadynamics
• Developing Potential Energy Surfaces for Graphene-based 2D-3D Interfaces from Modified High Dimensional Neural Networks for Applications in Energy Storage