论文标题
基于全球注意的图形卷积神经网络,用于改进材料的财产预测
Global Attention based Graph Convolutional Neural Networks for Improved Materials Property Prediction
论文作者
论文摘要
机器学习(ML)方法在探索和开发新材料方面越来越受欢迎。更具体地说,图形神经网络(GNN)已应用于预测材料特性。在这项工作中,我们开发了一种新型模型GATGNN,用于预测基于由多个图形注意层(GAT)和全球注意力层组成的图神经网络的无机材料特性。通过应用GAT层,我们的模型可以有效地学习每个原子本地社区中原子之间共享的复杂键。随后,全球注意力层提供了无机晶体材料中每个原子的重量系数,这些系数用于大大改善模型的性能。值得注意的是,随着GATGNN模型的发展,我们表明我们的方法能够胜过以前的模型的预测,并提供对材料结晶的见解。
Machine learning (ML) methods have gained increasing popularity in exploring and developing new materials. More specifically, graph neural network (GNN) has been applied in predicting material properties. In this work, we develop a novel model, GATGNN, for predicting inorganic material properties based on graph neural networks composed of multiple graph-attention layers (GAT) and a global attention layer. Through the application of the GAT layers, our model can efficiently learn the complex bonds shared among the atoms within each atom's local neighborhood. Subsequently, the global attention layer provides the weight coefficients of each atom in the inorganic crystal material which are used to considerably improve our model's performance. Notably, with the development of our GATGNN model, we show that our method is able to both outperform the previous models' predictions and provide insight into the crystallization of the material.