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

被引:45
作者
Gong, Sheng [1 ]
Xie, Tian [1 ]
Zhu, Taishan [1 ]
Wang, Shuo [2 ]
Fadel, Eric R. [1 ]
Li, Yawei [3 ]
Grossman, Jeffrey C. [1 ]
机构
[1] MIT, Dept Mat Sci & Engn, Cambridge, MA 02139 USA
[2] Univ Maryland, Dept Mat Sci & Engn, College Pk, MD 20742 USA
[3] Penn State Univ, Dept Chem Engn, University Pk, PA 16802 USA
基金
美国国家科学基金会;
关键词
MACHINE LEARNING-MODELS; REACTIVITY; STATE;
D O I
10.1103/PhysRevB.100.184103
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
The 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 the 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 that 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.
引用
收藏
页数:9
相关论文
共 46 条
[1]   Machine Learning Enabled Computational Screening of Inorganic Solid Electrolytes for Suppression of Dendrite Formation in Lithium Metal Anodes [J].
Ahmad, Zeeshan ;
Xie, Tian ;
Maheshwari, Chinmay ;
Grossman, Jeffrey C. ;
Viswanathan, Venkatasubramanian .
ACS CENTRAL SCIENCE, 2018, 4 (08) :996-1006
[2]   TOWARD A THEORY OF CHEMICAL-REACTIVITY BASED ON THE CHARGE-DENSITY [J].
BADER, RFW ;
MACDOUGALL, PJ .
JOURNAL OF THE AMERICAN CHEMICAL SOCIETY, 1985, 107 (24) :6788-6795
[3]   Transferable Atomic Multipole Machine Learning Models for Small Organic Molecules [J].
Bereau, Tristan ;
Andrienko, Denis ;
von Lilienfeld, O. Anatole .
JOURNAL OF CHEMICAL THEORY AND COMPUTATION, 2015, 11 (07) :3225-3233
[4]   PROJECTOR AUGMENTED-WAVE METHOD [J].
BLOCHL, PE .
PHYSICAL REVIEW B, 1994, 50 (24) :17953-17979
[5]   Bypassing the Kohn-Sham equations with machine learning [J].
Brockherde, Felix ;
Vogt, Leslie ;
Li, Li ;
Tuckerman, Mark E. ;
Burke, Kieron ;
Mueller, Klaus-Robert .
NATURE COMMUNICATIONS, 2017, 8
[6]   Solving the electronic structure problem with machine learning [J].
Chandrasekaran, Anand ;
Kamal, Deepak ;
Batra, Rohit ;
Kim, Chiho ;
Chen, Lihua ;
Ramprasad, Rampi .
NPJ COMPUTATIONAL MATERIALS, 2019, 5 (1)
[7]   Predicting the Electrochemical Properties of Lithium-Ion Battery Electrode Materials with the Quantum Neural Network Algorithm [J].
Choi, Hwanho ;
Sohn, Kee-Sun ;
Pyo, Myoungho ;
Chung, Kee-Choo ;
Park, Hwangseo .
JOURNAL OF PHYSICAL CHEMISTRY C, 2019, 123 (08) :4682-4690
[8]   EMBEDDED-ATOM METHOD - DERIVATION AND APPLICATION TO IMPURITIES, SURFACES, AND OTHER DEFECTS IN METALS [J].
DAW, MS ;
BASKES, MI .
PHYSICAL REVIEW B, 1984, 29 (12) :6443-6453
[9]   Reaction: The Near Future of Artificial Intelligence in Materials Discovery [J].
Gomez-Bombarelli, Rafael .
CHEM, 2018, 4 (06) :1189-1190
[10]   Automatic Chemical Design Using a Data-Driven Continuous Representation of Molecules [J].
Gomez-Bombarelli, Rafael ;
Wei, Jennifer N. ;
Duvenaud, David ;
Hernandez-Lobato, Jose Miguel ;
Sanchez-Lengeling, Benjamin ;
Sheberla, Dennis ;
Aguilera-Iparraguirre, Jorge ;
Hirzel, Timothy D. ;
Adams, Ryan P. ;
Aspuru-Guzik, Alan .
ACS CENTRAL SCIENCE, 2018, 4 (02) :268-276