Leveraging Orbital Information and Atomic Feature in Deep Learning Model
This work addresses material property prediction for materials science researchers, but it appears incremental as it builds on existing deep learning methods with hybrid enhancements.
The authors tackled material property prediction by proposing Orbital CrystalNet, a deep learning framework that integrates orbital field matrix and atomic features with graph representation learning, achieving superior performance over state-of-the-art models on Material Project and JARVIS datasets.
Predicting material properties base on micro structure of materials has long been a challenging problem. Recently many deep learning methods have been developed for material property prediction. In this study, we propose a crystal representation learning framework, Orbital CrystalNet, OCrystalNet, which consists of two parts: atomic descriptor generation and graph representation learning. In OCrystalNet, we first incorporate orbital field matrix (OFM) and atomic features to construct OFM-feature atomic descriptor, and then the atomic descriptor is used as atom embedding in the atom-bond message passing module which takes advantage of the topological structure of crystal graphs to learn crystal representation. To demonstrate the capabilities of OCrystalNet we performed a number of prediction tasks on Material Project dataset and JARVIS dataset and compared our model with other baselines and state of art methods. To further present the effectiveness of OCrystalNet, we conducted ablation study and case study of our model. The results show that our model have various advantages over other state of art models.