Steerable Wavelet Scattering for 3D Atomic Systems with Application to Li-Si Energy Prediction

Abstract

A general machine learning architecture is introduced that uses wavelet scattering coefficients of an inputted three dimensional signal as features. Solid harmonic wavelet scattering transforms of three dimensional signals were previously introduced in a machine learning framework for the regression of properties of small organic molecules. Here this approach is extended for general steerable wavelets which are equivariant to translations and rotations, resulting in a sparse model of the target function. The scattering coefficients inherit from the wavelets invariance to translations and rotations. As an illustration of this approach a linear regression model is learned for the formation energy of amorphous lithium-silicon material states trained over a database generated using plane-wave Density Functional Theory methods. State-of-the-art results are produced as compared to other machine learning approaches over similarly generated databases.

NIPS 2018 Workshop on Machine Learning for Molecules and Materials, Montreal, Canada

References in corpus (11)

• Gaussian Approximation Potentials: the accuracy of quantum mechanics, without the electrons
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• ANI-1: An extensible neural network potential with DFT accuracy at force field computational cost
• Machine Learning of Accurate Energy-Conserving Molecular Force Fields
• Comparing molecules and solids across structural and alchemical space
• Group Equivariant Convolutional Networks
• On the Generalization of Equivariance and Convolution in Neural Networks to the Action of Compact Groups
• Constructing first-principles phase diagrams of amorphous LixSi using machine-learning-assisted sampling with an evolutionary algorithm
• Covariant Compositional Networks For Learning Graphs
• Solid Harmonic Wavelet Scattering for Predictions of Molecule Properties
• Intermittent process analysis with scattering moments

Cited by in corpus (4)

• Wavelet Scattering Networks for Atomistic Systems with Extrapolation of Material Properties
• Geometric Scattering for Graph Data Analysis
• Geometric Scattering on Manifolds
• Geometric Wavelet Scattering Networks on Compact Riemannian Manifolds