Tensor-Reduced Atomic Density Representations

James P. Darby, Dávid P. Kovács, Ilyes Batatia, Miguel A. Caro, Gus L.W. Hart, Christoph Ortner, Gábor Csányi

Research output: Contribution to journalArticleScientificpeer-review

9 Citations (Scopus)
34 Downloads (Pure)

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 modeling, and the visualization and analysis of material 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. By exploiting symmetry, we recast this approach as tensor factorization of the standard neighbour-density-based descriptors and, using a new notation, identify connections to existing compression algorithms. In doing so, we form compact tensor-reduced representation of the local atomic environment whose size does not depend on the number of chemical elements, is systematically convergable, and therefore remains applicable to a wide range of data analysis and regression tasks.

Original languageEnglish
Article number028001
Number of pages7
JournalPhysical Review Letters
Volume131
Issue number2
DOIs
Publication statusPublished - 13 Jul 2023
MoE publication typeA1 Journal article-refereed

Fingerprint

Dive into the research topics of 'Tensor-Reduced Atomic Density Representations'. Together they form a unique fingerprint.

Cite this