journal article·2025-06-13Open access

Machine learning on multiple topological materials datasets

Yuqing He, Pierre-Paul De Breuck, Hongming Weng, Matteo Giantomassi, Gian-Marco Rignanese

npj Computational Materials· Volume 11· Issue 1

Abstract

A dataset of 35,608 materials with their topological properties is constructed by combining the density functional theory (DFT) results of Materiae and the Topological Materials Database. Thanks to this, machine-learning approaches are developed to categorize materials into five distinct topological types, with the XGBoost model achieving an impressive 85.2% classification accuracy. By conducting generalization tests on different sub-datasets, differences are identified between the original datasets in terms of topological types, chemical elements, unknown magnetic compounds, and feature space coverage. Their impact on model performance is analyzed. Turning to the simpler binary classification between trivial insulators and nontrivial topological materials, three different approaches are also tested. Key characteristics influencing material topology are identified, with the maximum packing efficiency and the fraction of p valence electrons being highlighted as critical features.
Source: openalex_inverted_index · Quality: baseline

Research context

OpenAlex topics: Machine Learning in Materials Science · X-ray Diffraction in Crystallography · Advanced Graph Neural Networks

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

References (56)

Show source-provided references
1. Kitaev, A. Periodic table for topological insulators and superconductors. In AIP Conference Proceedings, Vol. 1134, 22–30 (American Institute of Physics, 2009).10.1063/1.3149495 ↗
2. Hasan, M. Z. & Kane, C. L. Colloquium: topological insulators. Rev. Mod. Phys. 82, 3045 (2010).10.1103/RevModPhys.82.3045 ↗
3. Qi, X.-L. & Zhang, S.-C. Topological insulators and superconductors. Rev. Mod. Phys. 83, 1057 (2011).10.1103/RevModPhys.83.1057 ↗
4. Po, H. C., Vishwanath, A. & Watanabe, H. Symmetry-based indicators of band topology in the 230 space groups. Nat. Commun. 8, 50 (2017).10.1038/s41467-017-00133-2 ↗
5. Bradlyn, B. et al. Topological quantum chemistry. Nature 547, 298–305 (2017).10.1038/nature23268 ↗
6. Zhang, T. et al. Catalogue of topological electronic materials. Nature 566, 475–479 (2019).10.1038/s41586-019-0944-6 ↗
7. Vergniory, M. G. et al. A complete catalogue of high-quality topological materials. Nature 566, 480–485 (2019).10.1038/s41586-019-0954-4 ↗
8. Tang, F., Po, H. C., Vishwanath, A. & Wan, X. Comprehensive search for topological materials using symmetry indicators. Nature 566, 486–489 (2019).10.1038/s41586-019-0937-5 ↗
9. Kruthoff, J., De Boer, J., Van Wezel, J., Kane, C. L. & Slager, R.-J. Topological classification of crystalline insulators through band structure combinatorics. Phys. Rev. X 7, 041069 (2017).
10. Song, Z., Zhang, T., Fang, Z. & Fang, C. Quantitative mappings between symmetry and topology in solids. Nat. Commun. 9, 3530 (2018).10.1038/s41467-018-06010-w ↗
11. Khalaf, E., Po, H. C., Vishwanath, A. & Watanabe, H. Symmetry indicators and anomalous surface states of topological crystalline insulators. Phys. Rev. X 8, 031070 (2018).
12. Hohenberg, P. & Kohn, W. Inhomogeneous electron gas. Phys. Rev. 136, B864–B871 (1964).10.1103/PhysRev.136.B864 ↗
13. Kohn, W. & Sham, L. J. Self-consistent equations including exchange and correlation effects. Phys. Rev. 140, A1133–A1138 (1965).10.1103/PhysRev.140.A1133 ↗
14. Hellenbrandt, M. The inorganic crystal structure database (ICSD)-present and future. Crystallogr. Rev. 10, 17–22 (2004).10.1080/08893110410001664882 ↗
15. Watanabe, H., Po, H. C. & Vishwanath, A. Structure and topology of band structures in the 1651 magnetic space groups. Sci. Adv. 4, aat8685 (2018).10.1126/sciadv.aat8685 ↗
16. Elcoro, L. et al. Magnetic topological quantum chemistry. Nat. Commun. 12, 5965 (2021).10.1038/s41467-021-26241-8 ↗
17. Peng, B., Jiang, Y., Fang, Z., Weng, H. & Fang, C. Topological classification and diagnosis in magnetically ordered electronic materials. Phys. Rev. B 105, 235138 (2022).10.1103/PhysRevB.105.235138 ↗
18. Samuel, A. L. Some studies in machine learning using the game of checkers. IBM J. Res. Dev. 3, 210–229 (1959).10.1147/rd.33.0210 ↗
19. Zhang, Y. & Kim, E.-A. Quantum loop topography for machine learning. Phys. Rev. Lett. 118, 216401 (2017).10.1103/PhysRevLett.118.216401 ↗
20. Zhang, P., Shen, H. & Zhai, H. Machine learning topological invariants with neural networks. Phys. Rev. Lett. 120, 066401 (2018).10.1103/PhysRevLett.120.066401 ↗
21. Zhang, Y., Ginsparg, P. & Kim, E.-A. Interpreting machine learning of topological quantum phase transitions. Phys. Rev. Res. 2, 023283 (2020).10.1103/PhysRevResearch.2.023283 ↗
22. Scheurer, M. S. & Slager, R.-J. Unsupervised machine learning and band topology. Phys. Rev. Lett. 124, 226401 (2020).10.1103/PhysRevLett.124.226401 ↗
23. Donoho, D. L. Compressed sensing. IEEE Trans. Inf. Theory 52, 1289–1306 (2006).10.1109/TIT.2006.871582 ↗
24. Friedman, J. H. Greedy function approximation: a gradient boosting machine. Ann. Stat. 29, 1189–1232 (2001).10.1214/aos/1013203451 ↗
25. Acosta, C. M. et al. Analysis of topological transitions in two-dimensional materials by compressed sensing. Preprint at arXiv https://doi.org/10.48550/arXiv.1805.10950 (2018).10.48550/arXiv.1805.10950 ↗
26. Claussen, N., Bernevig, B. A. & Regnault, N. Detection of topological materials with machine learning. Phys. Rev. B 101, 245117 (2020).10.1103/PhysRevB.101.245117 ↗
27. Cao, G. et al. Artificial intelligence for high-throughput discovery of topological insulators: the example of alloyed tetradymites. Phys. Rev. Mater. 4, 034204 (2020).10.1103/PhysRevMaterials.4.034204 ↗
28. Liu, J., Cao, G., Zhou, Z. & Liu, H. Screening potential topological insulators in half-heusler compounds via compressed-sensing. J. Phys. Condens. Matter 33, 325501 (2021).10.1088/1361-648X/abba8d ↗
29. Schleder, G. R., Focassio, B. & Fazzio, A. Machine learning for materials discovery: two-dimensional topological insulators. Appl. Phys. Rev. 8, 031409 (2021).10.1063/5.0055035 ↗
30. Andrejevic, N. et al. Machine-learning spectral indicators of topology. Adv. Mater. 34, 2204113 (2022).10.1002/adma.202204113 ↗
31. Ma, A. et al. Topogivity: a machine-learned chemical rule for discovering topological materials. Nano Lett. 23, 772–778 (2023).10.1021/acs.nanolett.2c03307 ↗
32. Vergniory, M. G. et al. All topological bands of all nonmagnetic stoichiometric materials. Science 376, eabg9094 (2022).10.1126/science.abg9094 ↗
33. Cortes, C. & Vapnik, V. Support-vector networks. Mach. Learn. 20, 273–297 (1995).10.1007/BF00994018 ↗
34. Chen, C., Ye, W., Zuo, Y., Zheng, C. & Ong, S. P. Graph networks as a universal machine learning framework for molecules and crystals. Chem. Mater. 31, 3564–3572 (2019).10.1021/acs.chemmater.9b01294 ↗
35. Dunn, A., Wang, Q., Ganose, A., Dopp, D. & Jain, A. Benchmarking materials property prediction methods: the Matbench test set and automatminer reference algorithm. npj Comput. Mater. 6, 138 (2020).10.1038/s41524-020-00406-3 ↗
36. De Breuck, P.-P., Hautier, G. & Rignanese, G.-M. Materials property prediction for limited datasets enabled by feature selection and joint learning with MODNet. npj Comput. Mater. 7, 83 (2021).10.1038/s41524-021-00552-2 ↗
37. Xie, T. & Grossman, J. C. Crystal graph convolutional neural networks for an accurate and interpretable prediction of material properties. Phys. Rev. Lett. 120, 145301 (2018).10.1103/PhysRevLett.120.145301 ↗
38. Stone, M. Cross-validatory choice and assessment of statistical predictions. J. R. Stat. Soc. Ser. B Methodol. 36, 111–133 (1974).10.1111/j.2517-6161.1974.tb00994.x ↗
39. Ho, T. K. Random decision forests. In Proceedings of 3rd International Conference on Document Analysis and Recognition, Vol. 1, 278–282 (IEEE, 1995).10.1109/ICDAR.1995.598994 ↗
40. Chen, T. & Guestrin, C. XGBoost: a scalable tree boosting system. In Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. KDD ’16, 785–794 (ACM, 2016).10.1145/2939672.2939785 ↗
41. De Breuck, P.-P., Evans, M. L. & Rignanese, G.-M. Robust model benchmarking and bias-imbalance in data-driven materials science: a case study on modnet. J. Phys. Condens. Matter 33, 404002 (2021).10.1088/1361-648X/ac1280 ↗
42. Van der Maaten, L. & Hinton, G. Visualizing data using t-SNE. J. Mach. Learn. Res. 9, 2579–2605 (2008).
43. Jain, A. et al. Commentary: The Materials Project: a materials genome approach to accelerating materials innovation. APL Mater. 1, 011002 (2013).10.1063/1.4812323 ↗
44. Ong, S. P. et al. Python materials genomics (pymatgen): a robust, open-source python library for materials analysis. Comput. Mater. Sci. 68, 314–319 (2013).10.1016/j.commatsci.2012.10.028 ↗
45. Ward, L. et al. Including crystal structure attributes in machine learning models of formation energies via Voronoi tessellations. Phys. Rev. B 96, 024104 (2017).10.1103/PhysRevB.96.024104 ↗
46. Ward, L., Agrawal, A., Choudhary, A. & Wolverton, C. A general-purpose machine learning framework for predicting properties of inorganic materials. npj Comput. Mater. 2, 16028 (2016).10.1038/npjcompumats.2016.28 ↗
47. Deml, A. M., O’Hayre, R., Wolverton, C. & Stevanović, V. Predicting density functional theory total energies and enthalpies of formation of metal-nonmetal compounds by linear regression. Phys. Rev. B 93, 085142 (2016).10.1103/PhysRevB.93.085142 ↗
48. Ward, L. et al. Matminer: an open source toolkit for materials data mining. Comput. Mater. Sci. 152, 60–69 (2018).10.1016/j.commatsci.2018.05.018 ↗
49. Olson, R. S. et al. Applications of Evolutionary Computation: 19th European Conference, EvoApplications 2016, Porto, Portugal, Proceedings, Part I, Chap. Automating Biomedical Data Science Through Tree-Based Pipeline Optimization, 123–137 (Springer International Publishing, 2016).10.1007/978-3-319-31204-0_9 ↗
50. Faber, F., Lindmaa, A., von Lilienfeld, O. A. & Armiento, R. Crystal structure representations for machine learning models of formation energies. Int. J. Quantum Chem. 115, 1094–1101 (2015).10.1002/qua.24917 ↗
51. Zhang, R., Zhang, S., He, Z., Jing, J. & Sheng, S. Miedema calculator: a thermodynamic platform for predicting formation enthalpies of alloys within framework of miedema’s theory. Comput. Phys. Commun. 209, 58–69 (2016).10.1016/j.cpc.2016.08.013 ↗
52. Talirz, L. et al. Materials cloud, a platform for open computational science. Sci. Data 7, 299 (2020).10.1038/s41597-020-00637-5 ↗
53. Fraux, G., Cersonsky, R. & Ceriotti, M. Chemiscope: interactive structure-property explorer for materials and molecules. J. Open Source Softw. 5, 2117 (2020).10.21105/joss.02117 ↗
54. Andersen, C. W. et al. OPTIMADE, an API for exchanging materials data. Sci. Data 8, 217 (2021).10.1038/s41597-021-00974-z ↗
55. Evans, M. L. et al. Developments and applications of the optimade API for materials discovery, design, and data exchange. Digit. Discov. 3, 1509–1533 (2024).10.1039/D4DD00039K ↗
56. Kotochigova, S., Levine, Z. H., Shirley, E. L., Stiles, M. D. & Clark, C. W. Local-density-functional calculations of the energy of atoms. Phys. Rev. A 55, 191–199 (1997).10.1103/PhysRevA.55.191 ↗