Accelerating Material Property Prediction using Generically Complete Isometry Invariants

Abstract

Periodic material or crystal property prediction using machine learning has grown popular in recent years as it provides a computationally efficient replacement for classical simulation methods. A crucial first step for any of these algorithms is the representation used for a periodic crystal. While similar objects like molecules and proteins have a finite number of atoms and their representation can be built based upon a finite point cloud interpretation, periodic crystals are unbounded in size, making their representation more challenging. In the present work, we adapt the Pointwise Distance Distribution (PDD), a continuous and generically complete isometry invariant for periodic point sets, as a representation for our learning algorithm. The PDD distinguished all (more than 660 thousand) periodic crystals in the Cambridge Structural Database as purely periodic sets of points without atomic types. We develop a transformer model with a modified self-attention mechanism that combines PDD with compositional information via a spatial encoding method. This model is tested on the crystals of the Materials Project and Jarvis-DFT databases and shown to produce accuracy on par with state-of-the-art methods while being several times faster in both training and prediction time.

References in corpus (13)

• Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties
• On representing chemical environments
• SchNet - a deep learning architecture for molecules and materials
• Graph Networks as a Universal Machine Learning Framework for Molecules and Crystals
• Atomistic Line Graph Neural Network for Improved Materials Property Predictions
• The Joint Automated Repository for Various Integrated Simulations (JARVIS) for data-driven materials design
• Benchmarking Materials Property Prediction Methods: The Matbench Test Set and Automatminer Reference Algorithm
• Developing an improved Crystal Graph Convolutional Neural Network framework for accelerated materials discovery
• Deep Neural Networks for Accurate Predictions of Garnet Stability
• Revisiting Data Augmentation for Rotational Invariance in Convolutional Neural Networks
• SE(3) Equivariant Graph Neural Networks with Complete Local Frames
• Efficient Approximations of Complete Interatomic Potentials for Crystal Property Prediction
• Material Property Prediction using Graphs based on Generically Complete Isometry Invariants

Cited by in corpus (3)

• The importance of definitions in crystallography
• Generic families of finite metric spaces with identical or trivial 1-dimensional persistence
• Geographic-style maps with a local novelty distance help navigate the materials space