Tripartite interaction representation algorithm for crystal graph neural networks

Abstract

Abstract Driven by emerging research paradigms, the application of artificial intelligence models presents innovative tools for designing materials and optimizing their performance. In the field of materials science, there is a current research emphasis on exploring techniques for characterizing material structures to achieve precise descriptions. This paper proposes a crystal graph convolution neural network model that incorporates a tripartite interaction approach. The model not only incorporates atomic information, bond lengths, and bond angles but also offers a method for updating atoms and bond lengths, facilitating accurate descriptions of crystal structures by capturing implicit structural information. Focusing on predicting the formation energy of crystalline compounds, our results demonstrate improved predictive accuracy compared to existing representation algorithms. The average error of the formation energy in the random dataset, demonstrating robust generalization, is merely 0.048eV/atom, with an impressive $$R^{2}$$ R 2 value of 0.994. Additionally, this paper establishes a crystal graph neural network framework for predicting algorithm performance. By integrating automatic parallel algorithms and an automated process, we achieve a synthesis of these techniques, enhancing computational efficiency and streamlining algorithm usage.