Multi-fidelity machine learning for predicting bandgaps of nonlinear optical crystals
Zhaoxi Yu, Pujie Xue, Bin‐Bin Xie, Lin Shen, Wei‐Hai Fang
Abstract
, 25175-25188) and the gradient boosting regression tree algorithm. The calculated and experimental bandgaps of NLO crystals were collected as the low- and high-fidelity labels, respectively. The experimental values were predicted based on chemical compositions of crystals without prior knowledge about crystal structures. The multi-fidelity ML model overcame the performance of single-fidelity predictor. Furthermore, it was observed that less accurate predictions on the low-fidelity label may result in more accurate prediction on the high-fidelity label, at least in the present case. Using the multi-fidelity ML model with the best performance in this work, the mean absolute error on the test set of experimental bandgaps was 0.293 eV, which is smaller than that using the single-fidelity model (0.355 eV). It is far from perfect but accurate enough as an effective computational tool in the first step to discover novel NLO materials.