Orbital Graph Convolutional Neural Network for Material Property Prediction

Abstract

Material representations that are compatible with machine learning models play a key role in developing models that exhibit high accuracy for property prediction. Atomic orbital interactions are one of the important factors that govern the properties of crystalline materials, from which the local chemical environments of atoms is inferred. Therefore, to develop robust machine learningmodels for material properties prediction, it is imperative to include features representing such chemical attributes. Here, we propose the Orbital Graph Convolutional Neural Network (OGCNN), a crystal graph convolutional neural network framework that includes atomic orbital interaction features that learns material properties in a robust way. In addition, we embedded an encoder-decoder network into the OGCNN enabling it to learn important features among basic atomic (elemental features), orbital-orbital interactions, and topological features. We examined the performance of this model on a broad range of crystalline material data to predict different properties. We benchmarked the performance of the OGCNN model with that of: 1) the crystal graph convolutional neural network (CGCNN), 2) other state-of-the-art descriptors for material representations including Many-body Tensor Representation (MBTR) and the Smooth Overlap of Atomic Positions (SOAP), and 3) other conventional regression machine learning algorithms where different crystal featurization methods have been used. We find that OGCNN significantly outperforms them. The OGCNN model with high predictive accuracy can be used to discover new materials among the immense phase and compound spaces of materials

3 figures in main text. 7 figures in SI. Accepted for publication by Physical Review Materials. The template used for the latex has been taken from: https://github.com/brenhinkeller/preprint-template.tex

References in corpus (13)

• Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties
• On representing chemical environments
• Fast and Accurate Modeling of Molecular Atomization Energies with Machine Learning
• A General-Purpose Machine Learning Framework for Predicting Properties of Inorganic Materials
• Big Data of Materials Science - Critical Role of the Descriptor
• DScribe: Library of Descriptors for Machine Learning in Materials Science
• Universal Fragment Descriptors for Predicting Electronic Properties of Inorganic Crystals
• How to represent crystal structures for machine learning: towards fast prediction of electronic properties
• Discovery of low thermal conductivity compounds with first-principles anharmonic lattice dynamics calculations and Bayesian optimization
• Materials Cartography: Representing and Mining Material Space Using Structural and Electronic Fingerprints
• Unified Representation of Molecules and Crystals for Machine Learning
• Deep Neural Networks for Accurate Predictions of Garnet Stability
• Atom2Vec: learning atoms for materials discovery

Cited by in corpus (21)

• Molecular Contrastive Learning of Representations via Graph Neural Networks
• TransPolymer: a Transformer-based language model for polymer property predictions
• High-throughput determination of Hubbard U and Hund J values for transition metal oxides via linear response formalism
• Deep Reinforcement Learning Optimizes Graphene Nanopores for Efficient Desalination
• Graph Neural Networks for Molecules
• AI-driven inverse design of materials: Past, present and future
• Artificial Intelligence in Material Engineering: A review on applications of AI in Material Engineering
• Graph Neural Networks Accelerated Molecular Dynamics
• Machine-learning enhanced dark soliton detection in Bose-Einstein condensates
• An Introduction to Electrocatalyst Design using Machine Learning for Renewable Energy Storage
• Transferable and Robust Machine Learning Model for Predicting Stability of Si Anodes for Multivalent Cation Batteries
• Towards Complex Dynamic Physics System Simulation with Graph Neural ODEs
• A Tailored Convolutional Neural Network for Nonlinear Manifold Learning of Computational Physics Data using Unstructured Spatial Discretizations
• Self-supervised graph neural networks for accurate prediction of Néel temperature
• Coarse-grained crystal graph neural networks for reticular materials design
• Learning the Crystal Structure Genome for Property Classification
• PINK: physical-informed machine learning for lattice thermal conductivity
• Self-supervised Representations and Node Embedding Graph Neural Networks for Accurate and Multi-scale Analysis of Materials
• Deep learning of material transport in complex neurite networks
• Machine learning-driven elasticity prediction in advanced inorganic materials via convolutional neural networks
• Predicting Many Crystal Properties via an Adaptive Transformer-based Framework