Accelerated Data-Driven Discovery and Screening of Two-Dimensional Magnets Using Graph Neural Networks

Ahmed Elrashidy,James Della-Giustina,Jia-An Yan
2024-02-06
Abstract:In this study, we employ Graph Neural Networks (GNNs) to accelerate the discovery of novel 2D magnetic materials which have transformative potential in spintronics applications. Using data from the Materials Project database and the Computational 2D materials database (C2DB), we train three GNN architectures on a dataset of 1190 magnetic monolayers with energy above the convex hull ($E_{\text{hull}}$) less than 0.3 eV/atom. Our Crystal Diffusion Variational Auto Encoder (CDVAE) generates 11,100 candidate crystals. Subsequent training on two Atomistic Line Graph Neural Networks (ALIGNN) achieves a 93$\%$ accuracy in predicting magnetic monolayers and a mean average error of 0.039 eV/atom for $E_{\text{hull}}$ predictions. After narrowing down candidates based on magnetic likelihood and predicted energy, constraining the atom count in the monolayers to five or fewer, and performing dimensionality checks, we identify 190 candidates. These are validated using Density-Functional Theory (DFT) to confirm their magnetic and energetic favorability resulting in 167 magnetic monolayers with $E_{\text{hull}} < 0.3$ eV/atom and a total magnetization of $\geq$ $0.5 \mu_{B}$. Our methodology offers a way to accelerate exploring and predicting potential 2D magnetic materials, contributing to the ongoing computational and experimental efforts aimed at the discovery of new 2D magnets.
Disordered Systems and Neural Networks,Mesoscale and Nanoscale Physics,Materials Science
What problem does this paper attempt to address?
### Problems the Paper Attempts to Solve This paper aims to accelerate the discovery and screening of novel 2D magnetic materials using Graph Neural Networks (GNNs). Specifically, the research team utilized data from the Materials Project database and the Computational 2D Materials Database (C2DB) to train three GNN architectures to generate and predict magnetic monolayer materials with low energy convex hull (Ehull). These materials have great potential in spintronics applications. ### Main Steps and Methods 1. **Data Generation**: - Used Crystal Diffusion Variational Auto Encoder (CDVAE) to generate 11,100 candidate crystals. 2. **Preliminary Screening**: - Trained two Atomistic Line Graph Neural Networks (ALIGNN) models, one for predicting magnetic monolayer materials and the other for predicting Ehull. These models achieved 93% prediction accuracy and a mean absolute error (MAE) of 0.039 eV/atom, respectively. 3. **Further Screening**: - Based on conditions such as magnetic probability, predicted energy, charge neutrality, and atomic number limit (no more than 5 atoms), 190 candidate materials were screened. 4. **Structure Optimization**: - Used M3GNet IAP for symmetry-constrained structure optimization. 5. **Density Functional Theory (DFT) Validation**: - Validated the magnetism and energy stability of candidate materials through DFT calculations, ultimately confirming 167 magnetic monolayer materials with Ehull less than 0.3 eV/atom and total magnetization ≥0.5µB. ### Background and Significance - **Data-Driven Methods**: In recent years, data-driven methods have played a significant role in the discovery of new materials. Many studies have utilized high-throughput DFT calculations and machine learning (ML) techniques to develop databases of 1D, 2D, and 3D materials. - **Advantages of Graph Neural Networks**: Traditional ML methods have limitations in representing materials, while GNNs can iteratively update node representations through message-passing mechanisms, making them particularly suitable for exploring material properties and inverse design. - **Importance of 2D Magnetic Materials**: 2D magnetic materials have great potential in next-generation nanoelectronic devices, especially in spintronics applications. Despite some challenges, such as poor air stability and achieving semiconductor magnetism above room temperature, research in this field is very active. ### Conclusion This study proposes an accelerated data-driven method combining GNNs and high-throughput DFT calculations to discover and predict potential 2D magnetic materials. This method not only improves the efficiency of discovering new materials but also provides important references for future experimental and theoretical research.