Prediction of Magnetic Properties in van der Waals Magnets using Graph Neural Networks

Author(s)

Abstract

We study two-dimensional (2D) magnetic materials using state-of-the-art machine learning models that use a graph-theory framework. We find that representing materials as graphs allows us to better learn structure-property relationships by leveraging both the chemical properties of the constituent atoms and the connectivity between those atoms. Graph neural network models are capable of predicting global properties of crystal structure (i.e. graph-wise properties) and local properties of the constituent atoms (i.e. node-wise properties). We embed physical constraints into our model by simultaneously making predictions of local and global properties. In particular, we use the Atomistic Line Graph Neural Network (ALIGNN) architecture. We train the ALIGNN model on data comprising local and global magnetic moments of 314 2D structures of the form CrA$^\textii}$B$^\texti}$B$^\textii}$X$_6$, based on monolayer \ceCr2Ge2Te6}, calculated from first-principles. By learning the relationships between both local and global magnetic properties, we demonstrate an improvement over models that only consider global magnetic properties.

Download Paper

Materials

Citation

Issues

If you have any questions about this publication or are having problems accessing it, please contact [email protected].