Bayesian Optimization with Gaussian Processes Assisted by Deep Learning for Material Designs.

Abstract

Machine learning (ML) approaches have become ubiquitous in the search for new materials in recent years. Bayesian optimization (BO) based on Gaussian processes (GPs) has become a widely recognized approach in material exploration. However, feature engineering has critical impacts on the efficiency of GP-based BO, because GPs cannot automatically generate descriptors. To address this limitation, this study applies deep kernel learning (DKL), which combines a neural network with a GP, to BO. The efficiency of the DKL model was comparable to or significantly better than that of a standard GP in a data set of 922 oxide data sets, covering band gaps, ionic dielectric constants, and effective masses of electrons, as well as in experimental data sets, the band gaps of 610 hybrid organic-inorganic perovskite alloys. When searching for the alloy with the highest Curie temperature among 4560 alloys, the standard GP outperformed the DKL model because a strongly correlated descriptor of the Curie temperature could be directly utilized. Additionally, DKL supports transfer learning, which further enhances its efficiency. Thus, we believe that BO based on DKL paves the way for exploring diverse material spaces more effectively than GPs.