Hybrid Graph-Machine Learning Framework for Accurate and Interpretable Band Gap Prediction.
Journal:
Journal of chemical information and modeling
Published Date:
Mar 19, 2026
Abstract
Accurate prediction of the electronic band gap is essential for accelerating the discovery and design of semiconducting and energy materials. Conventional density functional theory (DFT) methods, while physically rigorous, remain computationally expensive and limited in scalability. In this study, we propose a hybrid artificial intelligence framework that combines graph-based deep learning embeddings with classical machine learning algorithms to achieve high-accuracy, interpretable, and computationally efficient band gap prediction. The model integrates embeddings obtained from CGCNN, MEGNet, and SchNet architectures with physically meaningful crystal descriptors─including maximum electronegativity, crystal system, space group, and spin-orbit coupling─and trains them using optimized gradient-boosting and neural architectures. Trained on 136,000 crystal structures from the Materials Project database, the hybrid model achieves R2 = 0.921, MAE = 0.191, and MSE = 0.155, outperforming both classical models (Ward et al., 2016) and standalone graph neural networks such as CGCNN (Xie and Grossman, 2018). The achieved accuracy is statistically comparable to the state-of-the-art ALIGNN model (Choudhary et al., 2021), while requiring lower computational resources and offering enhanced generalization due to the integration of multisource structural information. SHAP-based interpretability analysis highlights that the model captures physically consistent relationships, with metallicity and magnetic site features emerging as dominant factors in band gap prediction. These findings demonstrate that the synergy between deep structural embeddings and classical algorithms provides a powerful, scalable approach for materials informatics. The proposed framework establishes a foundation for multiproperty prediction, transfer learning across databases, and inverse materials design driven by interpretable artificial intelligence.
Authors
Keywords
No keywords available for this article.