Crystal twins:self-supervised learning for crystalline material property prediction  

作　　者：Rishikesh Magar Yuyang Wang Amir Barati Farimani 

机构地区：[1]Department of Mechanical Engineering,Carnegie Mellon University,Pittsburgh,PA,15213,USA [2]Department of Materials Science and Engineering,Carnegie Mellon University,Pittsburgh,PA,15213,USA [3]Department of Chemical Engineering,Carnegie Mellon University,Pittsburgh,PA,15213,USA

出　　处：《npj Computational Materials》2022年第1期2210-2217,共8页计算材料学（英文）

基　　金：The work has been supported by the start-up fund provided by Department of Mechanical Engineering at Carnegie Mellon University.

摘　　要：Machine learning(ML)models have been widely successful in the prediction of material properties.However,large labeled datasets required for training accurate ML models are elusive and computationally expensive to generate.Recent advances in Self-Supervised Learning(SSL)frameworks capable of training ML models on unlabeled data mitigate this problem and demonstrate superior performance in computer vision and natural language processing.Drawing inspiration from the developments in SSL,we introduce Crystal Twins(CT):a generic SSL method for crystalline materials property prediction that can leverage large unlabeled datasets.CT adapts a twin Graph Neural Network(GNN)and learns representations by forcing graph latent embeddings of augmented instances obtained from the same crystalline system to be similar.We implement Barlow Twins and SimSiam frameworks in CT.By sharing the pre-trained weights when fine-tuning the GNN for downstream tasks,we significantly improve the performance of GNN on 14 challenging material property prediction benchmarks.

关 键 词：PREDICTION PROPERTY CRYSTALLINE 

分 类 号：TB34[一般工业技术—材料科学与工程]