Graph neural networks predict energetic and mechanical properties for models of solid solution metal alloy phases

Abstract

We developed a graph convolutional neural network (GCNN) to predict the formation energy and the bulk modulus for models of solid solution alloys for various atomic crystal structures and relaxed volumes. We trained the GCNN model on a dataset for nickel-niobium (NiNb) that was generated for simplicity with the embedded atom model (EAM) empirical interatomic potential. The dataset has been generated by calculating the formation energy and the bulk modulus as a prototypical elastic property for optimized geometries starting from initial body-centered cubic (BCC), face-centered cubic (FCC), and hexagonal compact packed (HCP) crystal structures, with configurations spanning the possible compositional range for each of the three types of initial crystal structure. Numerical results show that the GCNN model effectively predicts both the formation energy and the bulk modulus as functions of the optimized crystal structure, relaxed volume, and configurational entropy of the model structures for solid solution alloys.