Prediction Model of Material Properties Based on Feature Fusion and Convolutional Neural Network
石竞琛 1刘霏凝 1王文杰 1赵瑞1
扫码查看
点击上方二维码区域，可以放大扫码查看
作者信息
1. 吉林师范大学 数学与计算机学院,吉林 四平 136000
折叠
摘要
针对目前大多数机器学习模型预测材料性质时需要大量的先验知识以及特征向量筛选困难的问题,基于电子轨道矩阵和元素周期表法两种描述符,通过特征融合的方式,设计了一种卷积神经网络模型OPCNN(Or-bital of electron and Periodic table CNN).实验数据表明,OPCNN与其他预测模型相比,在带隙、生成热以及形成能数据集上都有着更好的性能,平均绝对误差分别为0.26 eV、0.037 KJ/mol和0.073 eV/atom,且R2 都达到了91%以上.OPCNN在保证了预测准确性的同时对先验知识的要求更低,只需要元素周期表中的信息即可预测材料性质,特征融合的思想可以让特征设计更加灵活,有利于新材料体系快速和准确的预测.
Abstract
Aiming at the problem that most machine learning models need a lot of prior knowledge and manual selection of feature vectors in the prediction of material properties,a convolutional neural network model OPCNN(Orbital of Electron and Periodic table CNN)is established by feature fusion based on two descriptors,electronic orbit matrix and periodic table method.The experimental data show that compared with other prediction models,OPCNN has better performance on the bandgap,heat of formation and formation energy datasets,with the mean absolute error of 0.26 eV,0.037 KJ/mol and 0.073 eV/atom,respectively,and the R2 is more than 91%.OPCNN has lower requirements for prior knowledge while ensuring the accuracy of prediction.It only needs the information in the periodic table to predict the material properties.The idea of feature fusion can make feature design more flexible,which is conducive to the rapid and accurate prediction of new material systems.
关键词
神经网络/描述符/预测/元素周期表/电子轨道
Key words
neural networks/descriptor/prediction/periodic table of elements/electron orbital
维普
万方数据