Predicting Transition Temperature of Superconductors with Graph Neural Networks

Abstract

Predicting high temperature superconductors has long been a great challenge. The difficulty lies in how to predict the transition temperature (Tc) of superconductors. Although recent progress in material informatics has led to a number of machine learning models predicting Tc, prevailing models have not shown adequate generalization ability and physical rationality to find new high temperature superconductors, yet. In this work, a bond sensitive graph neural network (BSGNN) was developed to predict the Tc of various superconductors. In BSGNN, communicative message passing and graph attention methods were utilized to enhance the model's ability to process bonding and interaction information in the crystal lattice, which is crucial for the superconductivity. Consequently, our results revealed the relevance between chemical bond attributes and Tc. It indicates that shorter bond length is favored by high Tc. Meanwhile, some specific chemical elements that have relatively large van der Waals radius is favored by high Tc. It gives a convenient guidance for searching high temperature superconductors in materials database, by ruling out the materials that could never have high Tc