Abstract:The recently proposed crystal graph convolutional neural network (CGCNN) offers a highly versatile and accurate machine learning (ML) framework by learning material properties directly from graph-like representations of crystal structures ("crystal graphs"). Here, we develop an improved variant of the CGCNN model (iCGCNN) that outperforms the original by incorporating information of the Voronoi tessellated crystal structure, explicit 3-body correlations of neighboring constituent atoms, and an optimized chemical representation of interatomic bonds in the crystal graphs. We demonstrate the accuracy of the improved framework in two distinct illustrations: First, when trained/validated on 180,000/20,000 density functional theory (DFT) calculated thermodynamic stability entries taken from the Open Quantum Materials Database (OQMD) and evaluated on a separate test set of 230,000 entries, iCGCNN achieves a predictive accuracy that is significantly improved, i.e., 20% higher than that of the original CGCNN. Second, when used to assist high-throughput search for materials in the ThCr2Si2 structure-type, iCGCNN exhibited a success rate of 31% which is 310 times higher than an undirected high-throughput search and 2.4 times higher than that of the original CGCNN. Using both CGCNN and iCGCNN, we screened 132,600 compounds with elemental decorations of the ThCr2Si2 prototype crystal structure and identified a total of 97 new unique stable compounds by performing 757 DFT calculations, accelerating the computational time of the high-throughput search by a factor of 130. Our results suggest that the iCGCNN can be used to accelerate high-throughput discoveries of new materials by quickly and accurately identifying crystalline compounds with properties of interest.

GCIceNet: A Graph Convolutional Network for Accurate Classification of Water Phases

Topology-based Phase Identification of Bulk, Interface, and Confined Water using Edge-Conditioned Convolutional Graph Neural Network

Ice phase classification made easy with score-based denoising

Classification of liquid crystal textures using convolutional neural networks

GalaxyWater-CNN: Prediction of Water Positions on the Protein Structure by a 3D-Convolutional Neural Network

Novel approach for designing order parameters of clathrate hydrate structures by graph neural network

IceCoder: Identification of Ice phases in molecular simulation using variational autoencoder

Graph convolutional network for water network partitioning

Developing an improved Crystal Graph Convolutional Neural Network framework for accelerated materials discovery

Graph Neural Networks and Structural Information on Ionic Liquids: A Cheminformatics Study on Molecular Physicochemical Property Prediction

Accurate Predictions of Aqueous Solubility of Drug Moleculesviathe Multilevel Graph Convolutional Network (MGCN) and SchNet Architectures

An Interpretable Convolutional Neural Network Framework for Analyzing Molecular Dynamics Trajectories: a Case Study on Functional States for G-Protein-Coupled Receptors

A coarse-grained deep neural network model for liquid water

High-dimensional order parameters and neural network classifiers applied to amorphous ices

Multi-order graph attention network for water solubility prediction and interpretation

Phase Equilibrium of Water with Hexagonal and Cubic Ice Using the SCAN Functional

Rotation-equivariant Graph Neural Networks for Learning Glassy Liquids Representations

Learning physical properties of liquid crystals with deep convolutional neural networks

Physical information-enhanced graph neural network for predicting phase separation

IceGCN: An Interactive Sea Ice Classification Pipeline for SAR Imagery Based on Graph Convolutional Network