特征融合的卷积神经网络材料性质预测模型  

Prediction Model of Material Properties Based on Feature Fusion and Convolutional Neural Network

作　　者：石竞琛 刘霏凝 王文杰 赵瑞[1] SHI Jingchen;LIU Feining;WANG Wenjie;ZHAO Rui(College of Mathematics and Computer,Jilin Normal University,Siping 136000,China)

机构地区：[1]吉林师范大学数学与计算机学院,吉林四平136000

出　　处：《哈尔滨理工大学学报》2024年第3期116-124,共9页Journal of Harbin University of Science and Technology

基　　金：吉林省科技发展计划项目(YDZJ202201ZYTS629);国家自然科学基金(22078124)。

摘　　要：针对目前大多数机器学习模型预测材料性质时需要大量的先验知识以及特征向量筛选困难的问题,基于电子轨道矩阵和元素周期表法两种描述符,通过特征融合的方式,设计了一种卷积神经网络模型OPCNN(Orbital of electron and Periodic table CNN)。实验数据表明,OPCNN与其他预测模型相比,在带隙、生成热以及形成能数据集上都有着更好的性能,平均绝对误差分别为0.26 eV、0.037 KJ/mol和0.073 eV/atom,且R^(2)都达到了91%以上。OPCNN在保证了预测准确性的同时对先验知识的要求更低,只需要元素周期表中的信息即可预测材料性质,特征融合的思想可以让特征设计更加灵活,有利于新材料体系快速和准确的预测。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 R^(2) 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.

关 键 词：神经网络 描述符 预测 元素周期表 电子轨道 

分 类 号：TP399[自动化与计算机技术—计算机应用技术]