- Open Access
Efficient crystal structure prediction using universal neural network potential with diversity preservation in genetic algorithms
Phys. Rev. Materials 10, 063401 – Published 2 June, 2026
DOI: https://doi.org/10.1103/4knt-ccqg
Abstract
Crystal structure prediction (CSP) is crucial for identifying stable crystal structures in given systems and is a prerequisite for computational atomistic simulations. Recent advances in neural network potentials (NNPs) have reduced the computational cost of CSP. However, searching for stable crystal structures across the entire composition space in multicomponent systems remains a significant challenge. Here, we propose an improvement of a genetic algorithm (GA)-based CSP method using a universal NNP designed to efficiently expand convex hull volumes while preserving the diversity of crystal structures. Our hull-informed filtering and elitist-selection procedures incorporate an aging mechanism that prioritizes recently improved compositions. We also employ niching to prevent convergence to a small set of stoichiometries, thereby preserving a diverse, high-quality population. Our evaluation shows that the present method outperforms the symmetry-aware random structure generation and existing CSP methods, achieving a larger convex hull with fewer trials. We demonstrated that our approach, combined with the developed universal NNP (PFP), can accurately reproduce and explore phase diagrams obtained through DFT calculations; this indicates the validity of PFP across a wide range of crystal structures and element combinations. This study, which integrates a universal NNP with a GA-based CSP method, highlights the promise of these methods in materials discovery.
Physics Subject Headings (PhySH)
Article Text
Supplemental Material
References (88)
- J. Hafner, Ab-initio simulations of materials using VASP: Density-functional theory and beyond, J. Comput. Chem. 29, 2044 (2008).
- A. R. Oganov, Modern Methods of Crystal Structure Prediction (Wiley-VCH, Berlin, 2011).
- A. R. Oganov, C. J. Pickard, Q. Zhu, and R. J. Needs, Structure prediction drives materials discovery, Nat. Rev. Mater. 4, 331 (2019).
- D. J. Wales, Energy Landscapes Applications to Clusters, Biomolecules and Glasses (Cambridge University Press, Cambridge, 2004).
- A. Jain, Y. Shin, and K. A. Persson, Computational predictions of energy materials using density functional theory, Nat. Rev. Mater. 1, 15004 (2016).
- G. Hautier, C. Fischer, V. Ehrlacher, A. Jain, and G. Ceder, Data mined ionic substitutions for the discovery of new compounds, Inorg. Chem. 50, 656 (2011).
- C. J. Pickard and R. J. Needs, Ab initio random structure searching, J. Phys.: Condens. Matter 23, 053201 (2011).
- D. J. Wales and J. P. K. Doye, Global optimization by basin-hopping and the lowest energy structures of lennard-jones clusters containing up to 110 atoms, J. Phys. Chem. A 101, 5111 (1997).
- M. Amsler and S. Goedecker, Crystal structure prediction using the minima hopping method, J. Chem. Phys. 133, 224104 (2010).
- M. Krummenacher, M. Gubler, J. A. Finkler, H. Huber, M. Sommer-Jörgensen, and S. Goedecker, Performing highly efficient minima hopping structure predictions using the atomic simulation environment (ASE), SoftwareX 25, 101632 (2024).
- D. M. Deaven and K. M. Ho, Molecular geometry optimization with a genetic algorithm, Phys. Rev. Lett. 75, 288 (1995).
- A. R. Oganov and C. W. Glass, Crystal structure prediction using ab initio evolutionary techniques: Principles and applications, J. Chem. Phys. 124, 244704 (2006).
- C. W. Glass, A. R. Oganov, and N. Hansen, USPEX—Evolutionary crystal structure prediction, Comput. Phys. Commun. 175, 713 (2006).
- A. R. Oganov, A. O. Lyakhov, and M. Valle, How evolutionary crystal structure prediction works—and why, Acc. Chem. Res. 44, 227 (2011).
- A. O. Lyakhov, A. R. Oganov, H. T. Stokes, and Q. Zhu, New developments in evolutionary structure prediction algorithm USPEX, Comput. Phys. Commun. 184, 1172 (2013).
- D. C. Lonie and E. Zurek, XtalOpt: An open-source evolutionary algorithm for crystal structure prediction, Comput. Phys. Commun. 182, 372 (2011).
- Z. Falls, P. Avery, X. Wang, K. P. Hilleke, and E. Zurek, The XtalOpt evolutionary algorithm for crystal structure prediction, J. Phys. Chem. C 125, 1601 (2021).
- Y. Wang, J. Lv, L. Zhu, and Y. Ma, Crystal structure prediction via particle-swarm optimization, Phys. Rev. B 82, 094116 (2010).
- T. Yamashita, N. Sato, H. Kino, T. Miyake, K. Tsuda, and T. Oguchi, Crystal structure prediction accelerated by Bayesian optimization, Phys. Rev. Mater. 2, 013803 (2018).
- M. K. Bisbo and B. Hammer, Efficient global structure optimization with a machine-learned surrogate model, Phys. Rev. Lett. 124, 086102 (2020).
- J. Behler and M. Parrinello, Generalized neural-network representation of high-dimensional potential-energy surfaces, Phys. Rev. Lett. 98, 146401 (2007).
- A. P. Bartók, M. C. Payne, R. Kondor, and G. Csányi, Gaussian approximation potentials: the accuracy of quantum mechanics, without the electrons, Phys. Rev. Lett. 104, 136403 (2010).
- A. V. Shapeev, Moment tensor potentials: a class of systematically improvable interatomic potentials, Multiscale Model. Simul. 14, 1153 (2016).
- R. Drautz, Atomic cluster expansion for accurate and transferable interatomic potentials, Phys. Rev. B 99, 014104 (2019).
- E. V. Podryabinkin, E. V. Tikhonov, A. V. Shapeev, and A. R. Oganov, Accelerating crystal structure prediction by machine-learning interatomic potentials with active learning, Phys. Rev. B 99, 064114 (2019).
- C. Chen, W. Ye, Y. Zuo, C. Zheng, and S. P. Ong, Graph networks as a universal machine learning framework for molecules and crystals, Chem. Mat. 31, 3564 (2019).
- C. Chen and S. P. Ong, A universal graph deep learning interatomic potential for the periodic table, Nat. Comput. Sci. 2, 718 (2022).
- K. Choudhary, B. DeCost, L. Major, K. Butler, J. Thiyagalingam, and F. Tavazza, Unified graph neural network force-field for the periodic table: Solid state applications, Digit. Discov. 2, 346 (2023).
- B. Deng, P. Zhong, K. Jun, J. Riebesell, K. Han, C. J. Bartel, and G. Ceder, CHGNet as a pretrained universal neural network potential for charge-informed atomistic modelling, Nat. Mach. Intell. 5, 1031 (2023).
- I. Batatia, D. P. Kovacs, G. Simm, C. Ortner, and G. Csanyi, in Advances in Neural Information Processing Systems, Vol. 35, edited by S. Koyejo, S. Mohamed, A. Agarwal, D. Belgrave, K. Cho, and A. Oh (Curran Associates, Inc., 2022), pp. 11423–11436.
- I. Batatia, P. Benner, Y. Chiang, A. M. Elena, D. P. Kovács, J. Riebesell, X. R. Advincula, M. Asta, M. Avaylon, W. J. Baldwin, F. Berger, N. Bernstein, A. Bhowmik, F. Bigi, S. M. Blau, V. Cărare, M. Ceriotti, S. Chong, J. P. Darby, S. De, et al., A foundation model for atomistic materials chemistry, J. Chem. Phys. 163, 184110 (2025).
- A. Merchant, S. Batzner, S. S. Schoenholz, M. Aykol, G. Cheon, and E. D. Cubuk, Scaling deep learning for materials discovery, Nature (London) 624, 80 (2023).
- M. Neumann, J. Gin, B. Rhodes, S. Bennett, Z. Li, H. Choubisa, A. Hussey, and J. Godwin, Orb: A fast, scalable neural network potential, arXiv:2410.22570.
- Y.-L. Liao, B. Wood, A. Das, and T. Smidt, in International Conference on Learning Representations, edited by B. Kim, Y. Yue, S. Chaudhuri, K. Fragki- adaki, M. Khan, and Y. Sun (2024) Vol. 2024, pp. 39282–39309.
- L. Barroso-Luque, S. Muhammed, X. Fu, B. Wood, M. Dzamba, M. Gao, A. Rizvi, C. L. Zitnick, and Z. W. Ulissi, Open materials 2024 (OMat24) inorganic materials dataset and models, arXiv:2410.12771.
- H. Yang, C. Hu, Y. Zhou, X. Liu, Y. Shi, J. Li, G. Li, Z. Chen, S. Chen, C. Zeni, M. Horton, R. Pinsler, A. Fowler, D. Zügner, T. Xie, J. Smith, L. Sun, Q. Wang, L. Kong, C. Liu, et al., MatterSim: A deep learning atomistic model across elements, temperatures and pressures, arXiv:2405.04967.
- S. Takamoto, S. Izumi, and J. Li, TeaNet: Universal neural network interatomic potential inspired by iterative electronic relaxations, Comput. Mater. Sci. 207, 111280 (2022).
- S. Takamoto, C. Shinagawa, D. Motoki, K. Nakago, W. Li, I. Kurata, T. Watanabe, Y. Yayama, H. Iriguchi, Y. Asano, T. Onodera, T. Ishii, T. Kudo, H. Ono, R. Sawada, R. Ishitani, M. Ong, T. Yamaguchi, T. Kataoka, A. Hayashi, et al., Towards universal neural network potential for material discovery applicable to arbitrary combination of 45 elements, Nat. Commun. 13, 2991 (2022).
- R. Jacobs, D. Morgan, S. Attarian, J. Meng, C. Shen, Z. Wu, C. Y. Xie, J. H. Yang, N. Artrith, B. Blaiszik, G. Ceder, K. Choudhary, G. Csanyi, E. D. Cubuk, B. Deng, R. Drautz, X. Fu, J. Godwin, V. Honavar, O. Isayev, et al., A practical guide to machine learning interatomic potentials – Status and future, Curr. Opin. Solid State Mater. Sci. 35, 101214 (2025).
- T. L. Jacobsen, M. S. Jørgensen, and B. Hammer, On-the-fly machine learning of atomic potential in density functional theory structure optimization, Phys. Rev. Lett. 120, 026102 (2018).
- Q. Tong, L. Xue, J. Lv, Y. Wang, and Y. Ma, Accelerating CALYPSO structure prediction by data-driven learning of a potential energy surface, Faraday Discuss. 211, 31 (2018).
- H. Wang, Y. Zhang, L. Zhang, and H. Wang, Crystal structure prediction of binary alloys via deep potential, Front. Chem. 8, (2020).
- C. W. Park and C. Wolverton, Developing an improved crystal graph convolutional neural network framework for accelerated materials discovery, Phys. Rev. Mater. 4, 063801 (2020).
- S. Kang, W. Jeong, C. Hong, S. Hwang, Y. Yoon, and S. Han, Accelerated identification of equilibrium structures of multicomponent inorganic crystals using machine learning potentials, npj Comput. Mater. 8, 108 (2022).
- G. Cheng, X.-G. Gong, and W.-J. Yin, Crystal structure prediction by combining graph network and optimization algorithm, Nat. Commun. 13, 1492 (2022).
- J. Schmidt, N. Hoffmann, H.-C. Wang, P. Borlido, P. J. M. A. Carriço, T. F. T. Cerqueira, S. Botti, and M. A. L. Marques, Machine-learning-assisted determination of the global zero-temperature phase diagram of materials, Adv. Mater. 35, 2210788 (2023).
- L. Chang, H. Tamaki, T. Yokoyama, K. Wakasugi, S. Yotsuhashi, M. Kusaba, A. R. Oganov, and R. Yoshida, Downfolding from ab initio to interacting model Hamiltonians: comprehensive analysis and benchmarking of the DFT+cRPA approach, npj Comput. Mater. 10, 129 (2024).
- S. Donaldson, R. A. Lawrence, and M. I. J. Probert, A genetic algorithm for convex hull optimisation, arXiv:2404.14354.
- P. E. Blöchl, Projector augmented-wave method, Phys. Rev. B 50, 17953 (1994).
- G. Kresse and D. Joubert, From ultrasoft pseudopotentials to the projector augmented-wave method, Phys. Rev. B 59, 1758 (1999).
- J. P. Perdew, K. Burke, and M. Ernzerhof, Generalized gradient approximation made simple, Phys. Rev. Lett. 77, 3865 (1996).
- G. Kresse and J. Hafner, Ab initio molecular dynamics for liquid metals, Phys. Rev. B 47, 558 (1993).
- G. Kresse and J. Furthmüller, Efficient iterative schemes for ab initio total-energy calculations using a plane-wave basis set, Phys. Rev. B 54, 11169 (1996).
- G. Kresse and J. Furthmüller, Efficiency of ab-initio total energy calculations for metals and semiconductors using a plane-wave basis set, Comput. Mater. Sci. 6, 15 (1996).
- S. L. Dudarev, G. A. Botton, S. Y. Savrasov, C. J. Humphreys, and A. P. Sutton, Electron-energy-loss spectra and the structural stability of nickel oxide: An LSDA+U study, Phys. Rev. B 57, 1505 (1998).
- A. Jain, S. P. Ong, G. Hautier, W. Chen, W. D. Richards, S. Dacek, S. Cholia, D. Gunter, D. Skinner, G. Ceder, and K. A. Persson, Commentary: The Materials Project: A materials genome approach to accelerating materials innovation, APL Mater. 1, 011002 (2013).
- L. Wang, T. Maxisch, and G. Ceder, Oxidation energies of transition metal oxides within the framework, Phys. Rev. B 73, 195107 (2006).
- M. J. Mehl, D. Hicks, C. Toher, O. Levy, R. M. Hanson, G. Hart, and S. Curtarolo, The AFLOW library of crystallographic prototypes: part 1, Comput. Mater. Sci. 136, S1 (2017).
- S. P. Ong, W. D. Richards, A. Jain, G. Hautier, M. Kocher, S. Cholia, D. Gunter, V. L. Chevrier, K. A. Persson, and G. Ceder, Python Materials Genomics (pymatgen): A robust, open-source python library for materials analysis, Comput. Mater. Sci. 68, 314 (2013).
- A. Wang, R. Kingsbury, M. McDermott, M. Horton, A. Jain, S. P. Ong, S. Dwaraknath, and K. A. Persson, A framework for quantifying uncertainty in DFT energy corrections, Sci. Rep. 11, 15496 (2021).
- A. Jain, G. Hautier, S. P. Ong, C. J. Moore, C. C. Fischer, K. A. Persson, and G. Ceder, Formation enthalpies by mixing GGA and calculations, Phys. Rev. B 84, 045115 (2011).
- G. Trimarchi, A. J. Freeman, and A. Zunger, Predicting stable stoichiometries of compounds via evolutionary global space-group optimization, Phys. Rev. B 80, 092101 (2009).
- C. J. Bartel, Review of computational approaches to predict the thermodynamic stability of inorganic solids, J. Mater. Sci. 57, 10475 (2022).
- S. S. Omee, L. Wei, M. Hu, and J. Hu, Crystal structure prediction using neural network potential and age-fitness Pareto genetic algorithm, J. Mater. Inform. 4, (2024).
- S. S. Omee, L. Wei, S. Dey, and J. Hu, Polymorphism crystal structure prediction with adaptive space group diversity control, Adv. Sci. 12, e10792 (2025).
- D. E. Goldberg, J. Richardson, et al., in Genetic Algorithms and their Applications: Proceedings of the Second International Conference on Genetic Algorithms, Vol. 4149 (Lawrence Erlbaum, Hillsdale, NJ, 1987), pp. 414–425.
- K. Deb, A. Pratap, S. Agarwal, and T. Meyarivan, A fast and elitist multiobjective genetic algorithm: NSGA-II, IEEE Trans. Evol. Comput. 6, 182 (2002).
- K. Deb and H. Jain, An evolutionary many-objective optimization algorithm using reference-point-based nondominated sorting approach, part i: Solving problems with box constraints, IEEE Trans. Evol. Comput. 18, 577 (2014).
- T. Akiba, S. Sano, T. Yanase, T. Ohta, and M. Koyama, in Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining (Association for Computing Machinery, New York, 2019), pp. 2623–2631.
- A. H. Larsen, J. J. Mortensen, J. Blomqvist, I. E. Castelli, R. Christensen, M. Dułak, J. Friis, M. N. Groves, B. Hammer, C. Hargus, E. D. Hermes, P. C. Jennings, P. B. Jensen, J. Kermode, J. R. Kitchin, E. L. Kolsbjerg, J. Kubal, K. Kaasbjerg, S. Lysgaard, J. B. Maronsson, et al., The atomic simulation environment—a Python library for working with atoms, J. Phys. Condens. Matter 29, 273002 (2017).
- A. O. Lyakhov, A. R. Oganov, and M. Valle, How to predict very large and complex crystal structures, Comput. Phys. Commun. 181, 1623 (2010).
- S. Fredericks, K. Parrish, D. Sayre, and Q. Zhu, PyXtal: A Python library for crystal structure generation and symmetry analysis, Comput. Phys. Commun. 261, 107810 (2021).
- N. L. Abraham and M. I. J. Probert, Improved real-space genetic algorithm for crystal structure and polymorph prediction, Phys. Rev. B 77, 134117 (2008).
- M. Valle and A. R. Oganov, in 2008 IEEE Symposium on Visual Analytics Science and Technology (IEEE, 2008), pp. 11–18.
- A. R. Oganov and M. Valle, How to quantify energy landscapes of solids, J. Chem. Phys. 130, 104504 (2009).
- A. R. Oganov, Y. Ma, A. O. Lyakhov, M. Valle, and C. Gatti, Evolutionary crystal structure prediction as a method for the discovery of minerals and materials, Rev. Mineral. Geochem. 71, 271 (2010).
- Q. Zhu, A. R. Oganov, and A. O. Lyakhov, Evolutionary metadynamics: A novel method to predict crystal structures, CrystEngComm 14, 3596 (2012).
- E. Real, A. Aggarwal, Y. Huang, and Q. V. Le, in in Pro- ceedings of the Thirty-Third AAAI Conference on Ar- tificial Intelligence and Thirty-First Innovative Appli- cations of Artificial Intelligence Conference and Ninth AAAI Symposium on Educational Advances in Artificial Intelligence (AAAI Press, Honolulu, 2019).
- The modified crossover and mutation methods are released and available at https://pypi.org/project/pfn-ase-extras/.
- L. B. Vilhelmsen and B. Hammer, Systematic study of to gold clusters on MgO(100) centers using density-functional theory, Phys. Rev. Lett. 108, 126101 (2012).
- P. J. Clark and F. C. Evans, Distance to nearest neighbor as a measure of spatial relationships in populations, Ecology 35, 445 (1954).
- C. Zeni, R. Pinsler, D. Zügner, A. Fowler, M. Horton, X. Fu, Z. Wang, A. Shysheya, J. Crabbé, S. Ueda, R. Sordillo, L. Sun, J. Smith, B. Nguyen, H. Schulz, S. Lewis, C.-W. Huang, Z. Lu, Y. Zhou, H. Yang, et al., A generative model for inorganic materials design, Nature (London) 639, 624 (2025).
- See Supplemental Material at http://link.aps.org/supplemental/10.1103/4knt-ccqg for crystal structures in xyz formats.
- A. Togo, K. Shinohara, and T. Isao, Spglib a software library for crystal symmetry search, Sci. Technol. Adv. Mater., Meth. 4, 2384822 (2024).
- D. Hicks, M. J. Mehl, E. Gossett, C. Toher, O. Levy, R. M. Hanson, G. Hart, and S. Curtarolo, The AFLOW library of crystallographic prototypes: Part 2, Comput. Mater. Sci. 161, S1 (2019).
- D. Hicks, M. J. Mehl, M. Esters, C. Oses, O. Levy, G. L. Hart, C. Toher, and S. Curtarolo, The AFLOW library of crystallographic prototypes: Part 3, Comput. Mater. Sci. 199, 110450 (2021).
- H. Eckert, S. Divilov, M. J. Mehl, D. Hicks, A. C. Zettel, M. Esters, X. Campilongo, and S. Curtarolo, The AFLOW library of crystallographic prototypes: Part 4, Comput. Mater. Sci. 240, 112988 (2024).
- R. E. Goodall, A. S. Parackal, F. A. Faber, R. Armiento, and A. A. Lee, Rapid discovery of stable materials by coordinate-free coarse graining, Sci. Adv. 8, eabn4117 (2022).