1School of Science, Chongqing University of Posts and Telecommunications, Chongqing 400065, China 2Institute for Advanced Sciences, Chongqing University of Posts and Telecommunications, Chongqing 400065, China 3Southwest Center for Theoretical Physics, Chongqing University, Chongqing 401331, China 4Inspur Electronic Information Industry Co., Ltd, Beijing 100085, China
Abstract:Antiferromagnetic materials are exciting quantum materials with rich physics and great potential for applications. On the other hand, an accurate and efficient theoretical method is highly demanded for determining critical transition temperatures, Néel temperatures, of antiferromagnetic materials. The powerful graph neural networks (GNNs) that succeed in predicting material properties lose their advantage in predicting magnetic properties due to the small dataset of magnetic materials, while conventional machine learning models heavily depend on the quality of material descriptors. We propose a new strategy to extract high-level material representations by utilizing self-supervised training of GNNs on large-scale unlabeled datasets. According to the dimensional reduction analysis, we find that the learned knowledge about elements and magnetism transfers to the generated atomic vector representations. Compared with popular manually constructed descriptors and crystal graph convolutional neural networks, self-supervised material representations can help us to obtain a more accurate and efficient model for Néel temperatures, and the trained model can successfully predict high Néel temperature antiferromagnetic materials. Our self-supervised GNN may serve as a universal pre-training framework for various material properties.
Liu C, Wang Y, Li H, Wu Y, Li Y, Li J, He K, Xu Y, Zhang J, and Wang Y 2020 Nat. Mater.19 522
[5]
Park B G, Wunderlich J, Martı́ X, Holỳ V, Kurosaki Y, Yamada M, Yamamoto H, Nishide A, Hayakawa J, Takahashi H, Shick A B, and Jungwirth T 2011 Nat. Mater.10 347
Wadley P, Howells B, Železnỳ J, Andrews C, Hills V, Campion R P, Novák V, Olejnı́k K, Maccherozzi F, SS D, Martin S Y, Wanger T, Wunderlich J, Freimuth F, Mokrosov Y, Kuneš J, Chauhan J S, Grzybowski M J, Rushforth A W, Edmonds K W, Gallagher B L, and Jungwirth T 2016 Science351 587
Kuo C Y, Drees Y, Fernández-Díaz M T, Zhao L, Vasylechko L, Sheptyakov D, Bell A M T, Pi T W, Lin H J, Wu M K, Pellegrin E, Valvidares S M, Li Z W, Adler P, Todorova A, Küchler R, Steppke A, Tjeng L H, Hu Z, and Komarek A C 2014 Phys. Rev. Lett.113 217203
[47]
Hiley C I, Scanlon D O, Sokol A A, Woodley S M, Ganose A M, Sangiao S, De Teresa J M, Manuel P, Khalyavin D D, Walker M, Lees M R, and Walton R I 2015 Phys. Rev. B92 104413