Abstract:The band gap is a key physical quantity in material design. First-principles calculations based on density functional theory can approximately predict the band gap, which often require significant computational resources and time. Deep learning models have the advantages of good fitting ability and automatic feature extraction from the data, and are gradually being applied to predict the band gap. In this paper, aiming at the problem of quickly obtaining the band gap value of perovskite materials, a feature fusion neural network model named CGCrabNet is established, and the transfer learning strategy is used to predict the band gap of perovskite materials. CGCrabNet extracts features from both chemical equation and crystal structure of materials, and fits the mapping between features and band gaps. It is an end-to-end neural network model. Based on the pre-training data obtained from the Open Quantum Materials Database (OQMD dataset), the CGCrabNet parameters can be fine-tuned by using only 175 perovskite material data to improve the robustness of the model. The numerical experimental results show that the prediction error of the CGCrabNet model for band gap prediciton based on the OQMD dataset is 0.014eV, which is lower than that obtained from the prediction based on Compositionally restricted attention-based network (CrabNet). The mean absolute error of the model developed in this paper for the prediction of perovskite materials is 0.374eV, which is lower 0.304eV, 0.441eV and 0.194eV than that obtained from random forest regression, support vector machine regression and gradient boosting regression, respectively. The mean absolute error of the test set of CGCrabNet trained only using perovskite data is 0.536 eV, and the mean absolute error of the pre-trained CGCrabNet has decreased by 0.162 eV, which indicates that the transfer learning strategy has significant role in improving the prediction accuracy of small data sets (perovskite material data sets). The difference between the predicted band gap of some perovskite materials such as SrHfO 3 and RbPaO 3 by the model and the band gap calculated by first-principles is less than 0.05eV, which indicates that the CGCrabNet can quickly and accurately predict the properties of new materials and accelerate the development process of new materials.

Band gap prediction of perovskite materials based on transfer learning

Highly versatile and accurate machine learning methods for predicting perovskite properties

Predicting the band gap of lead-free inorganic double perovskites using modified parallel residual network

Predicting the properties of perovskite materials by improved compositionally restricted attention-based networks and explainable machine learning

Leveraging Large-scale Computational Database and Deep Learning for Accurate Prediction of Material Properties

Machine-Learning Prediction of the Computed Band Gaps of Double Perovskite Materials

A Progressive Learning Method for Predicting the Band Gap of ABO3 Perovskites Using an Instrumental Variable

Study on Bandgap Predications of ABX3-type Perovskites by Machine Learning

Predicting band gaps and band-edge positions of oxide perovskites using DFT and machine learning

Deep learning for ultra-fast and high precision screening of energy materials

Accurate and rapid predictions with explainable graph neural networks for small high-fidelity bandgap datasets

Employing the Interpretable Ensemble Learning Approach to Predict the Bandgaps of the Halide Perovskites

Accelerated prediction of perovskite material properties with classical machine learning and graph neural network

Comparative Analysis of Conventional Machine Learning and Graph Neural Network Models for Perovskite Property Prediction

Machine learning-enhanced band gaps prediction for low-symmetry double and layered perovskites

Predicting band gaps of MOFs on small data by deep transfer learning with data augmentation strategies

Predicting the bandgap and efficiency of perovskite solar cells using machine learning methods

Band gap predictions of double perovskite oxides using machine learning

Band gap prediction for large organic crystal structures with machine learning

Data Mining and Graph Network Deep Learning for Band Gap Prediction in Crystalline Borate Materials