Tensor-reduced atomic density representations

Abstract

Density based representations of atomic environments that are invariant under Euclidean symmetries have become a widely used tool in the machine learning of interatomic potentials, broader data-driven atomistic modelling and the visualisation and analysis of materials datasets.The standard mechanism used to incorporate chemical element information is to create separate densities for each element and form tensor products between them. This leads to a steep scaling in the size of the representation as the number of elements increases. Graph neural networks, which do not explicitly use density representations, escape this scaling by mapping the chemical element information into a fixed dimensional space in a learnable way. We recast this approach as tensor factorisation by exploiting the tensor structure of standard neighbour density based descriptors. In doing so, we form compact tensor-reduced representations whose size does not depend on the number of chemical elements, but remain systematically convergeable and are therefore applicable to a wide range of data analysis and regression tasks.

6 pages, 3 figures

References in corpus (6)

• A Spectral Analysis Method for Automated Generation of Quantum-Accurate Interatomic Potentials
• Sketching as a Tool for Numerical Linear Algebra
• Efficient and Accurate Machine-Learning Interpolation of Atomic Energies in Compositions with Many Species
• Gaussian Moments as Physically Inspired Molecular Descriptors for Accurate and Scalable Machine Learning Potentials
• Recursive evaluation and iterative contraction of $N$-body equivariant features
• GPU-Accelerated Approximate Kernel Method for Quantum Machine Learning

Cited by in corpus (11)

• A practical guide to machine learning interatomic potentials -- Status and future
• Evaluation of the MACE Force Field Architecture: from Medicinal Chemistry to Materials Science
• Roadmap on machine learning glassy dynamics
• Neural Network Kinetics for Exploring Diffusion Multiplicity and Chemical Ordering in Compositionally Complex Materials
• Automatic feature selection and weighting in molecular systems using Differentiable Information Imbalance
• A collinear-spin machine learned interatomic potential for Fe${}_7$Cr${}_2$Ni alloy
• Universal machine learning interatomic potentials poised to supplant DFT in modeling general defects in metals and random alloys
• How Graph Neural Network Interatomic Potentials Extrapolate: Role of the Message-Passing Algorithm
• Uncertainty Quantification in Atomistic Simulations of Silicon using Interatomic Potentials
• Equivariant Tensor Network Potentials
• Scalable Data-Driven Basis Selection for Linear Machine Learning Interatomic Potentials