期刊
CRYSTALS
卷 9, 期 4, 页码 -出版社
MDPI
DOI: 10.3390/cryst9040191
关键词
material informatics; material descriptor; convolutional neural networks; features extraction; formation energy
资金
- National Natural Science Foundation of China [51741101]
- National Science Foundation EPSCoR Program [OIA-1655740]
Computational prediction of crystal materials properties can help to do large-scale in-silicon screening. Recent studies of material informatics have focused on expert design of multi-dimensional interpretable material descriptors/features. However, successes of deep learning such as Convolutional Neural Networks (CNN) in image recognition and speech recognition have demonstrated their automated feature extraction capability to effectively capture the characteristics of the data and achieve superior prediction performance. Here, we propose CNN-OFM-Magpie, a CNN model with OFM (Orbital-field Matrix) and Magpie descriptors to predict the formation energy of 4030 crystal material by exploiting the complementarity of two-dimensional OFM features and Magpie features. Experiments showed that our method achieves better performance than conventional regression algorithms such as support vector machines and Random Forest. It is also better than CNN models using only the OFM features, the Magpie features, or the basic one-hot encodings. This demonstrates the advantages of CNN and feature fusion for materials property prediction. Finally, we visualized the two-dimensional OFM descriptors and analyzed the features extracted by the CNN to obtain greater understanding of the CNN-OFM model.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据