Machine learning for energy band prediction of halide perovskites

Abstract Halide perovskites have emerged as a class of highly promising photovoltaic materials with exceptional optoelectronic properties. The bandgaps of halide perovskites, along with the energy levels of the conduction band minimum (CBM) and valence band maximum (VBM), play a critical role in determining light absorption, interfacial energy alignment, charge carrier dynamics and photovoltaic performance of the corresponding solar cells. Herein, we developed high-accuracy machine learning models based on state-of-the-art algorithms to predict the CBM, VBM and bandgaps of halide perovskites. In this work, the eXtreme Gradient Boosting Regression (XGB), which outperformed five other shallow machine learning models as well as Transformer and multilayer perceptron (MLP) models, achieved a coefficient of determination (R²) of 0.8298 for CBM prediction (R² of 0.8481 for VBM) and a mean absolute error (MAE) of 0.1510 eV (MAE of 0.1490 eV for VBM) on the test set. For predicting Perdew-Burke-Ernzerhof functional (PBE)-calculated bandgaps, the XGB model demonstrated an R² score of 0.9316 and an MAE of 0.1018 eV on the test set. Finally, we conducted SHapley Additive exPlanations (SHAP) analysis based on the optimal models to identify the key features influencing energy band properties of halide perovskites. Our findings statistically revealed the dominant factors affecting bandgaps, CBM and VBM energy levels in halide materials, which aligned with previous non-machine learning studies. This work provides meaningful insights for the rational design of halide perovskites with tailored energy band properties.