Active learning for accelerated design of layered materials

Bassman, Lindsay; Rajak, Pankaj; Kalia, Rajiv K.; Nakano, Aiichiro; Sha, Fei; Sun, Jifeng; Singh, David J.; Aykol, Muratahan; Huck, Patrick; Persson, Kristin; Vashishta, Priya

doi:10.1038/s41524-018-0129-0

Title: Active learning for accelerated design of layered materials

Journal Article · Mon Dec 10 00:00:00 EST 2018 · npj Computational Materials

DOI:https://doi.org/10.1038/s41524-018-0129-0· OSTI ID:1494091

Bassman, Lindsay ^[1];

^[2]; Kalia, Rajiv K. ^[3];

^[3]; Sha, Fei ^[4]; Sun, Jifeng ^[5];

^[5]; Aykol, Muratahan ^[6]; Huck, Patrick ^[6]; Persson, Kristin ^[6]; Vashishta, Priya ^[3]

Univ. of Southern California, Los Angeles, CA (United States). Collaboratory for Advanced Computing and Simulations, Dept. of Physics and Astronomy
Univ. of Southern California, Los Angeles, CA (United States). Collaboratory for Advanced Computing and Simulations, and Dept. of Computer Science
Univ. of Southern California, Los Angeles, CA (United States). Collaboratory for Advanced Computing and Simulations, Dept. of Physics and Astronomy, and Dept. of Chemical Engineering and Material Science, and Dept. of Computer Science
Univ. of Southern California, Los Angeles, CA (United States). Dept. of Chemical Engineering and Material Science, and Dept. of Computer Science
Univ. of Missouri, Columbia, MO (United States). Dept. of Physics and Astronomy
Lawrence Berkeley National Lab. (LBNL), Berkeley, CA (United States)

Hetero-structures made from vertically stacked monolayers of transition metal dichalcogenides hold great potential for optoelectronic and thermoelectric devices. Discovery of the optimal layered material for specific applications necessitates the estimation of key material properties, such as electronic band structure and thermal transport coefficients. However, screening of material properties via brute force ab initio calculations of the entire material structure space exceeds the limits of current computing resources. Moreover, the functional dependence of material properties on the structures is often complicated, making simplistic statistical procedures for prediction difficult to employ without large amounts of data collection. Here, we present a Gaussian process regression model, which predicts material properties of an input hetero-structure, as well as an active learning model based on Bayesian optimization, which can efficiently discover the optimal hetero-structure using a minimal number of ab initio calculations. The electronic band gap, conduction/valence band dispersions, and thermoelectric performance are used as representative material properties for prediction and optimization. The Materials Project platform is used for electronic structure computation, while the BoltzTraP code is used to compute thermoelectric properties. Bayesian optimization is shown to significantly reduce the computational cost of discovering the optimal structure when compared with finding an optimal structure by building a regression model to predict material properties. The models can be used for predictions with respect to any material property and our software, including data preparation code based on the Python Materials Genomics (PyMatGen) library as well as python-based machine learning code, is available open source.

View Accepted Manuscript (DOE)

Cite

Export

Save

Research Organization:: Lawrence Berkeley National Laboratory (LBNL), Berkeley, CA (United States)

Sponsoring Organization:: USDOE Office of Science (SC), Basic Energy Sciences (BES)

Grant/Contract Number:: AC02-05CH11231

OSTI ID:: 1494091

Journal Information:: npj Computational Materials, Vol. 4, Issue 1; ISSN 2057-3960

Publisher:: Nature Publishing GroupCopyright Statement

Country of Publication:: United States

Language:: English

Citation Metrics:

Cited by: 81 works

Citation information provided by
Web of Science

References (55)

From Organized High-Throughput Data to Phenomenological Theory using Machine Learning: The Example of Dielectric Breakdown Kim, Chiho; Pilania, Ghanshyam; Ramprasad, Ramamurthy Chemistry of Materials, Vol. 28, Issue 5 https://doi.org/10.1021/acs.chemmater.5b04109	journal	February 2016
Generalized Gradient Approximation Made Simple Perdew, John P.; Burke, Kieron; Ernzerhof, Matthias Physical Review Letters, Vol. 77, Issue 18, p. 3865-3868 https://doi.org/10.1103/PhysRevLett.77.3865	journal	October 1996
Effect of quantum-well structures on the thermoelectric figure of merit Hicks, L. D.; Dresselhaus, M. S. Physical Review B, Vol. 47, Issue 19, p. 12727-12731 https://doi.org/10.1103/PhysRevB.47.12727	journal	May 1993
Projector augmented-wave method Blöchl, P. E. Physical Review B, Vol. 50, Issue 24, p. 17953-17979 https://doi.org/10.1103/PhysRevB.50.17953	journal	December 1994
Efficiency of ab-initio total energy calculations for metals and semiconductors using a plane-wave basis set Kresse, G.; Furthmüller, J. Computational Materials Science, Vol. 6, Issue 1, p. 15-50 https://doi.org/10.1016/0927-0256(96)00008-0	journal	July 1996
The inorganic crystal structure data base Bergerhoff, G.; Hundt, R.; Sievers, R. Journal of Chemical Information and Modeling, Vol. 23, Issue 2 https://doi.org/10.1021/ci00038a003	journal	May 1983
Machine learning bandgaps of double perovskites Pilania, G.; Mannodi-Kanakkithodi, A.; Uberuaga, B. P. Scientific Reports, Vol. 6, Issue 1 https://doi.org/10.1038/srep19375	journal	January 2016
Accelerated materials property predictions and design using motif-based fingerprints Huan, Tran Doan; Mannodi-Kanakkithodi, Arun; Ramprasad, Rampi Physical Review B, Vol. 92, Issue 1 https://doi.org/10.1103/PhysRevB.92.014106	journal	July 2015
Strong interlayer coupling in van der Waals heterostructures built from single-layer chalcogenides Fang, H.; Battaglia, C.; Carraro, C. Proceedings of the National Academy of Sciences, Vol. 111, Issue 17 https://doi.org/10.1073/pnas.1405435111	journal	April 2014
Identifying Structural Flow Defects in Disordered Solids Using Machine-Learning Methods Cubuk, E. D.; Schoenholz, S. S.; Rieser, J. M. Physical Review Letters, Vol. 114, Issue 10 https://doi.org/10.1103/PhysRevLett.114.108001	journal	March 2015
Van der Waals heterostructures Geim, A. K.; Grigorieva, I. V. Nature, Vol. 499, Issue 7459, p. 419-425 https://doi.org/10.1038/nature12385	journal	July 2013
Taking the Human Out of the Loop: A Review of Bayesian Optimization Shahriari, Bobak; Swersky, Kevin; Wang, Ziyu Proceedings of the IEEE, Vol. 104, Issue 1 https://doi.org/10.1109/JPROC.2015.2494218	journal	January 2016
A New Method of Locating the Maximum Point of an Arbitrary Multipeak Curve in the Presence of Noise Kushner, H. J. Journal of Basic Engineering, Vol. 86, Issue 1 https://doi.org/10.1115/1.3653121	journal	March 1964
Electronics and optoelectronics of two-dimensional transition metal dichalcogenides Wang, Qing Hua; Kalantar-Zadeh, Kourosh; Kis, Andras Nature Nanotechnology, Vol. 7, Issue 11, p. 699-712 https://doi.org/10.1038/nnano.2012.193	journal	November 2012
Toward Effective Utilization of Methane: Machine Learning Prediction of Adsorption Energies on Metal Alloys Toyao, Takashi; Suzuki, Keisuke; Kikuchi, Shoma The Journal of Physical Chemistry C, Vol. 122, Issue 15 https://doi.org/10.1021/acs.jpcc.7b12670	journal	March 2018
Materials informatics: An emerging technology for materials development LeSar, Richard Statistical Analysis and Data Mining, Vol. 1, Issue 6 https://doi.org/10.1002/sam.10034	journal	June 2009
Recent advances in optoelectronic properties and applications of two-dimensional metal chalcogenides Xia, Congxin; Li, Jingbo Journal of Semiconductors, Vol. 37, Issue 5 https://doi.org/10.1088/1674-4926/37/5/051001	journal	May 2016
Efficient iterative schemes for ab initio total-energy calculations using a plane-wave basis set Kresse, G.; Furthmüller, J. Physical Review B, Vol. 54, Issue 16, p. 11169-11186 https://doi.org/10.1103/PhysRevB.54.11169	journal	October 1996
From ultrasoft pseudopotentials to the projector augmented-wave method Kresse, G.; Joubert, D. Physical Review B, Vol. 59, Issue 3, p. 1758-1775 https://doi.org/10.1103/PhysRevB.59.1758	journal	January 1999
Multi-fidelity machine learning models for accurate bandgap predictions of solids Pilania, G.; Gubernatis, J. E.; Lookman, T. Computational Materials Science, Vol. 129 https://doi.org/10.1016/j.commatsci.2016.12.004	journal	March 2017
Modulating Optoelectronic Properties of Two-Dimensional Transition Metal Dichalcogenide Semiconductors by Photoinduced Charge Transfer Choi, Jungwook; Zhang, Hanyu; Choi, Jong Hyun ACS Nano, Vol. 10, Issue 1 https://doi.org/10.1021/acsnano.5b07457	journal	December 2015
Using support vector regression for the prediction of the band gap and melting point of binary and ternary compound semiconductors Gu, Tianhong; Lu, Wencong; Bao, Xinhua Solid State Sciences, Vol. 8, Issue 2 https://doi.org/10.1016/j.solidstatesciences.2005.10.011	journal	February 2006
Commentary: The Materials Project: A materials genome approach to accelerating materials innovation Jain, Anubhav; Ong, Shyue Ping; Hautier, Geoffroy APL Materials, Vol. 1, Issue 1 https://doi.org/10.1063/1.4812323	journal	July 2013
Inhomogeneous Electron Gas Hohenberg, P.; Kohn, W. Physical Review, Vol. 136, Issue 3B, p. B864-B871 https://doi.org/10.1103/PhysRev.136.B864	journal	November 1964
Informatics-aided bandgap engineering for solar materials Dey, Partha; Bible, Joe; Datta, Somnath Computational Materials Science, Vol. 83 https://doi.org/10.1016/j.commatsci.2013.10.016	journal	February 2014
Machine Learning Strategy for Accelerated Design of Polymer Dielectrics Mannodi-Kanakkithodi, Arun; Pilania, Ghanshyam; Huan, Tran Doan Scientific Reports, Vol. 6, Issue 1 https://doi.org/10.1038/srep20952	journal	February 2016
Physical and chemical tuning of two-dimensional transition metal dichalcogenides Wang, Haotian; Yuan, Hongtao; Sae Hong, Seung Chemical Society Reviews, Vol. 44, Issue 9 https://doi.org/10.1039/C4CS00287C	journal	January 2015
Representations in neural network based empirical potentials Cubuk, Ekin D.; Malone, Brad D.; Onat, Berk The Journal of Chemical Physics, Vol. 147, Issue 2 https://doi.org/10.1063/1.4990503	journal	July 2017
Ab initiomolecular dynamics for liquid metals Kresse, G.; Hafner, J. Physical Review B, Vol. 47, Issue 1, p. 558-561 https://doi.org/10.1103/PhysRevB.47.558	journal	January 1993
Recent development in 2D materials beyond graphene Gupta, Ankur; Sakthivel, Tamilselvan; Seal, Sudipta Progress in Materials Science, Vol. 73 https://doi.org/10.1016/j.pmatsci.2015.02.002	journal	August 2015
Machine Learning Assisted Predictions of Intrinsic Dielectric Breakdown Strength of ABX ₃ Perovskites Kim, Chiho; Pilania, Ghanshyam; Ramprasad, Rampi The Journal of Physical Chemistry C, Vol. 120, Issue 27 https://doi.org/10.1021/acs.jpcc.6b05068	journal	June 2016
Detailed Balance Limit of Efficiency of p‐n Junction Solar Cells Shockley, William; Queisser, Hans J. Journal of Applied Physics, Vol. 32, Issue 3, p. 510-519 https://doi.org/10.1063/1.1736034	journal	March 1961
BoltzTraP. A code for calculating band-structure dependent quantities Madsen, Georg K. H.; Singh, David J. Computer Physics Communications, Vol. 175, Issue 1 https://doi.org/10.1016/j.cpc.2006.03.007	journal	July 2006
Photovoltaic Effect in an Electrically Tunable van der Waals Heterojunction Furchi, Marco M.; Pospischil, Andreas; Libisch, Florian Nano Letters, Vol. 14, Issue 8 https://doi.org/10.1021/nl501962c	journal	July 2014
The chemistry of two-dimensional layered transition metal dichalcogenide nanosheets Chhowalla, Manish; Shin, Hyeon Suk; Eda, Goki Nature Chemistry, Vol. 5, Issue 4, p. 263-275 https://doi.org/10.1038/nchem.1589	journal	April 2013
A general-purpose machine learning framework for predicting properties of inorganic materials Ward, Logan; Agrawal, Ankit; Choudhary, Alok npj Computational Materials, Vol. 2, Issue 1 https://doi.org/10.1038/npjcompumats.2016.28	journal	August 2016
On the tuning of electrical and thermal transport in thermoelectrics: an integrated theory–experiment perspective Yang, Jiong; Xi, Lili; Qiu, Wujie npj Computational Materials, Vol. 2, Issue 1 https://doi.org/10.1038/npjcompumats.2015.15	journal	February 2016
Artificial neural network prediction of the band gap and melting point of binary and ternary compound semiconductors Zhaochun, Zhang; Ruiwu, Peng; Nianyi, Chen Materials Science and Engineering: B, Vol. 54, Issue 3 https://doi.org/10.1016/S0921-5107(98)00157-3	journal	June 1998
Atomically-thin layered films for device applications based upon 2D TMDC materials McDonnell, Stephen J.; Wallace, Robert M. Thin Solid Films, Vol. 616 https://doi.org/10.1016/j.tsf.2016.08.068	journal	October 2016
Materials informatics Rajan, Krishna Materials Today, Vol. 8, Issue 10 https://doi.org/10.1016/S1369-7021(05)71123-8	journal	October 2005
Prediction model of band gap for inorganic compounds by combination of density functional theory calculations and machine learning techniques Lee, Joohwi; Seko, Atsuto; Shitara, Kazuki Physical Review B, Vol. 93, Issue 11 https://doi.org/10.1103/PhysRevB.93.115104	journal	March 2016
Electronic fitness function for screening semiconductors as thermoelectric materials Xing, Guangzong; Sun, Jifeng; Li, Yuwei Physical Review Materials, Vol. 1, Issue 6 https://doi.org/10.1103/PhysRevMaterials.1.065405	journal	November 2017
Python Materials Genomics (pymatgen): A robust, open-source python library for materials analysis Ong, Shyue Ping; Richards, William Davidson; Jain, Anubhav Computational Materials Science, Vol. 68 https://doi.org/10.1016/j.commatsci.2012.10.028	journal	February 2013
Self-Consistent Equations Including Exchange and Correlation Effects Kohn, W.; Sham, L. J. Physical Review, Vol. 140, Issue 4A, p. A1133-A1138 https://doi.org/10.1103/PhysRev.140.A1133	journal	November 1965
Inhomogeneous Electron Gas Rajagopal, A. K.; Callaway, J. Physical Review B, Vol. 7, Issue 5 https://doi.org/10.1103/physrevb.7.1912	journal	March 1973
Reducing Dzyaloshinskii-Moriya interaction and field-free spin-orbit torque switching in synthetic antiferromagnets Chen, Ruyi; Cui, Qirui; Liao, Liyang Nature Communications, Vol. 12, Issue 1 https://doi.org/10.1038/s41467-021-23414-3	journal	May 2021
High-resolution X-ray luminescence extension imaging Ou, Xiangyu; Qin, Xian; Huang, Bolong Nature, Vol. 590, Issue 7846 https://doi.org/10.1038/s41586-021-03251-6	journal	February 2021
Electronic structure of AlFeN films exhibiting crystallographic orientation change from c- to a-axis with Fe concentrations and annealing effect Tatemizo, Nobuyuki; Imada, Saki; Okahara, Kizuna Scientific Reports, Vol. 10, Issue 1 https://doi.org/10.1038/s41598-020-58835-5	journal	February 2020
Materials informatics Rodgers, John Materials Science and Technology, Vol. 25, Issue 4 https://doi.org/10.1179/174328409x430456	journal	April 2009
Materials Informatics Rajan, Krishna; Mendez, Patricio Statistical Analysis and Data Mining, Vol. 1, Issue 5 https://doi.org/10.1002/sam.10022	journal	April 2009
Materials informatics Rajan, Krishna Materials Today, Vol. 15, Issue 11 https://doi.org/10.1016/s1369-7021(12)70204-3	journal	November 2012
Identifying structural flow defects in disordered solids using machine learning methods Cubuk, Ekin D.; Schoenholz, Samuel S.; Rieser, Jennifer M. arXiv https://doi.org/10.48550/arxiv.1409.6820	text	January 2014
Accelerated materials property predictions and design using motif-based fingerprints Huan, Tran Doan; Mannodi-Kanakkithodi, Arun; Ramprasad, Rampi arXiv https://doi.org/10.48550/arxiv.1503.07503	text	January 2015
Prediction model of band-gap for AX binary compounds by combination of density functional theory calculations and machine learning techniques Lee, Joohwi; Seko, Atsuto; Shitara, Kazuki arXiv https://doi.org/10.48550/arxiv.1509.00973	text	January 2015
A General-Purpose Machine Learning Framework for Predicting Properties of Inorganic Materials Ward, Logan; Agrawal, Ankit; Choudhary, Alok arXiv https://doi.org/10.48550/arxiv.1606.09551	text	January 2016

Cited By (12)

Machine Learning Approaches for Thermoelectric Materials Research Wang, Tian; Zhang, Cheng; Snoussi, Hichem Advanced Functional Materials, Vol. 30, Issue 5 https://doi.org/10.1002/adfm.201906041	journal	November 2019
Recent advances and applications of machine learning in solid-state materials science Schmidt, Jonathan; Marques, Mário R. G.; Botti, Silvana npj Computational Materials, Vol. 5, Issue 1 https://doi.org/10.1038/s41524-019-0221-0	journal	August 2019
A fast neural network approach for direct covariant forces prediction in complex multi-element extended systems Mailoa, Jonathan P.; Kornbluth, Mordechai; Batzner, Simon Nature Machine Intelligence, Vol. 1, Issue 10 https://doi.org/10.1038/s42256-019-0098-0	journal	September 2019
Screening billions of candidates for solid lithium-ion conductors: A transfer learning approach for small data Cubuk, Ekin D.; Sendek, Austin D.; Reed, Evan J. The Journal of Chemical Physics, Vol. 150, Issue 21 https://doi.org/10.1063/1.5093220	journal	June 2019
From DFT to machine learning: recent approaches to materials science–a review Schleder, Gabriel R.; Padilha, Antonio C. M.; Acosta, Carlos Mera Journal of Physics: Materials, Vol. 2, Issue 3 https://doi.org/10.1088/2515-7639/ab084b	journal	May 2019
Rocketsled: a software library for optimizing high-throughput computational searches Dunn, Alexander; Brenneck, Julien; Jain, Anubhav Journal of Physics: Materials, Vol. 2, Issue 3 https://doi.org/10.1088/2515-7639/ab0c3d	journal	April 2019
Electronic structure as a guide in screening for potential thermoelectrics: Demonstration for half-Heusler compounds Feng, Zhenzhen; Fu, Yuhao; Putatunda, Aditya Physical Review B, Vol. 100, Issue 8 https://doi.org/10.1103/physrevb.100.085202	journal	August 2019
Active-learning and materials design: the example of high glass transition temperature polymers Kim, Chiho; Chandrasekaran, Anand; Jha, Anurag MRS Communications, Vol. 9, Issue 3 https://doi.org/10.1557/mrc.2019.78	journal	June 2019
Materials science in the artificial intelligence age: high-throughput library generation, machine learning, and a pathway from correlations to the underpinning physics Vasudevan, Rama K.; Choudhary, Kamal; Mehta, Apurva MRS Communications, Vol. 9, Issue 3 https://doi.org/10.1557/mrc.2019.95	journal	July 2019
Active learning of deep surrogates for PDEs: application to metasurface design Pestourie, Raphaël; Mroueh, Youssef; Nguyen, Thanh V. npj Computational Materials, Vol. 6, Issue 1 https://doi.org/10.1038/s41524-020-00431-2	journal	October 2020
Scalable Gaussian Processes for Predicting the Properties of Inorganic Glasses with Large Datasets Bishnoi, Suresh; Ravinder, R.; Singh, Hargun arXiv https://doi.org/10.48550/arxiv.2007.02795	preprint	January 2020
Efficient Estimation of Material Property Curves and Surfaces via Active Learning Tian, Yuan; Xue, Dezhen; Yuan, Ruihao arXiv https://doi.org/10.48550/arxiv.2010.06896	text	January 2020