Predicting charge density distribution of materials using a local-environment-based graph convolutional network

Abstract

Electron charge density distribution of materials is one of the key quantities in computational materials science as theoretically it determines the ground state energy and practically it is used in many materials analyses. However, the scaling of density functional theory calculations with number of atoms limits the usage of charge-density-based calculations and analyses. Here we introduce a machine learning scheme with local-environment-based graphs and graph convolutional neural networks to predict charge density on grid-points from crystal structure. We show the accuracy of this scheme through a comparison of predicted charge densities as well as properties derived from the charge density, and the scaling is O(N). More importantly, the transferability is shown to be high with respect to different compositions and structures, which results from the explicit encoding of geometry.

References in corpus (10)

• Automatic chemical design using a data-driven continuous representation of molecules
• Crystal Graph Convolutional Neural Networks for an Accurate and Interpretable Prediction of Material Properties
• SchNet - a deep learning architecture for molecules and materials
• By-passing the Kohn-Sham equations with machine learning
• Prediction model of band-gap for AX binary compounds by combination of density functional theory calculations and machine learning techniques
• Machine Learning Enabled Computational Screening of Inorganic Solid Electrolytes for Dendrite Suppression with Li Metal Anode
• A Transferable Machine-Learning Model of the Electron Density
• Laplacian-level density functionals for the kinetic energy density and exchange-correlation energy
• Transferable atomic multipole machine learning models for small organic molecules
• Hierarchical Visualization of Materials Space with Graph Convolutional Neural Networks

Cited by in corpus (13)

• The Open Catalyst 2020 (OC20) Dataset and Community Challenges
• Quantum deep field: data-driven wave function, electron density generation, and atomization energy prediction and extrapolation with machine learning
• Equivariant graph neural networks for fast electron density estimation of molecules, liquids, and solids
• Calibrating DFT formation enthalpy calculations by multi-fidelity machine learning
• Predicting Elastic Properties of Materials from Electronic Charge Density Using 3D Deep Convolutional Neural Networks
• Structure motif centric learning framework for inorganic crystalline systems
• First-Principles Prediction of Electronic Transport in Experimental Semiconductor Heterostructures via Physics-Based Machine Learning
• Screening and understanding Li adsorption on 2-dimensional metallic materials by learning physics
• Spiers Memorial Lecture: How to do impactful research in artificial intelligence for chemistry and materials science
• Convolutional network learning of self-consistent electron density via grid-projected atomic fingerprints
• Shifting computational boundaries for complex organic materials
• On the equivalence of molecular graph convolution and molecular wave function with poor basis set
• Machine learning based prediction of the electronic structure of quasi-one-dimensional materials under strain