Reuse & Permissions

It is not necessary to obtain permission to reuse this article or its components as it is available under the terms of the Creative Commons Attribution 4.0 International license. This license permits unrestricted use, distribution, and reproduction in any medium, provided attribution to the author(s) and the published article's title, journal citation, and DOI are maintained. Please note that some figures may have been included with permission from other third parties. It is your responsibility to obtain the proper permission from the rights holder directly for these figures.

Export citation

Export citation

Choose format for download:

Download Citation
  • Letter
  • Open Access

Neural network evolution strategy for solving quantum sign structures

Ao Chen1,*, Kenny Choo2, Nikita Astrakhantsev2, and Titus Neupert2

  • 1Department of Physics, ETH Zurich, CH-8093 Zurich, Switzerland
  • 2Department of Physics, University of Zurich, CH-8057 Zurich, Switzerland

  • *chenao.phys@gmail.com

Phys. Rev. Research 4, L022026 – Published 2 May, 2022

DOI: https://doi.org/10.1103/PhysRevResearch.4.L022026

Abstract

Feed-forward neural networks are a novel class of variational wave functions for correlated many-body quantum systems. Here, we propose a specific neural network ansatz suitable for systems with real-valued wave functions. Its characteristic is to encode the all-important rugged sign structure of a quantum wave function in a convolutional neural network with discrete output. Its training is achieved through an evolutionary algorithm. We test our variational ansatz and training strategy on two spin-1/2 Heisenberg models, one on the two-dimensional square lattice and one on the three-dimensional pyrochlore lattice. In the former, our ansatz converges with high accuracy to the analytically known sign structures of ordered phases. In the latter, where such sign structures are a priori unknown, we obtain better variational energies than with other neural network states. Our results demonstrate the utility of discrete neural networks to solve quantum many-body problems.

View figure in article

Physics Subject Headings (PhySH)

Article Text

Supplemental Material

References (37)

  1. D. Ceperley and B. Alder, Quantum Monte Carlo, Science 231, 555 (1986).
  2. M. Troyer and U.-J. Wiese, Computational Complexity and Fundamental Limitations to Fermionic Quantum Monte Carlo Simulations, Phys. Rev. Lett. 94, 170201 (2005).
  3. F. Verstraete, V. Murg, and J. Cirac, Matrix product states, projected entangled pair states, and variational renormalization group methods for quantum spin systems, Adv. Phys. 57, 143 (2008).
  4. U. Schollwöck, The density-matrix renormalization group in the age of matrix product states, Ann. Phys. (NY) 326, 96 (2011).
  5. F. Becca and S. Sorella, Variational Monte Carlo, in Quantum Monte Carlo Approaches for Correlated Systems (Cambridge University Press, 2017) pp. 103–130.
  6. D. Tahara and M. Imada, Variational Monte Carlo method combined with quantum-number projection and multi-variable optimization, J. Phys. Soc. Jpn. 77, 114701 (2008).
  7. G. Carleo and M. Troyer, Solving the quantum many-body problem with artificial neural networks, Science 355, 602 (2017).
  8. V. Dumoulin and F. Visin, A guide to convolution arithmetic for deep learning, arXiv:1603.07285.
  9. T. Cohen and M. Welling,Group equivariant convolutional networks, Proceedings of The 33rd International Conference on Machine Learning, edited by M. F. Balcan, and K. Q. Weinberger, Proceedings of Machine Learning Research (PMLR, New York, New York, USA, 2016), pp. 2990–2999.
  10. Y. Nomura, Helping restricted Boltzmann machines with quantum-state representation by restoring symmetry, J. Phys.: Condens. Matter 33, 174003 (2021).
  11. K. Choo, T. Neupert, and G. Carleo, Two-dimensional frustrated J1−J2 model studied with neural network quantum states, Phys. Rev. B 100, 125124 (2019).
  12. C. Roth and A. H. MacDonald, Group convolutional neural networks improve quantum state accuracy, arXiv:2104.05085.
  13. T. Vieijra, C. Casert, J. Nys, W. De Neve, J. Haegeman, J. Ryckebusch, and F. Verstraete, Restricted Boltzmann Machines for Quantum States with Non-Abelian or Anyonic Symmetries, Phys. Rev. Lett. 124, 097201 (2020).
  14. T. Vieijra and J. Nys, Many-body quantum states with exact conservation of non-Abelian and lattice symmetries through variational Monte Carlo, Phys. Rev. B 104, 045123 (2021).
  15. O. Sharir, Y. Levine, N. Wies, G. Carleo, and A. Shashua, Deep Autoregressive Models for the Efficient Variational Simulation of Many-Body Quantum Systems, Phys. Rev. Lett. 124, 020503 (2020).
  16. L. Yang, Z. Leng, G. Yu, A. Patel, W.-J. Hu, and H. Pu, Deep learning-enhanced variational Monte Carlo method for quantum many-body physics, Phys. Rev. Research 2, 012039(R) (2020).
  17. M. Li, J. Chen, Q. Xiao, F. Wang, Q. Jiang, X. Zhao, R. Lin, H. An, X. Liang, and L. He, Bridging the gap between deep learning and frustrated quantum spin system for extreme-scale simulations on new generation of Sunway supercomputer, IEEE Transactions on Parallel and Distributed Systems 1 (2022).
  18. Z. Cai and J. Liu, Approximating quantum many-body wave functions using artificial neural networks, Phys. Rev. B 97, 035116 (2018).
  19. A. Szabó and C. Castelnovo, Neural network wave functions and the sign problem, Phys. Rev. Research 2, 033075 (2020).
  20. F. Ferrari, F. Becca, and J. Carrasquilla, Neural Gutzwiller-projected variational wave functions, Phys. Rev. B 100, 125131 (2019).
  21. Y. Nomura, A. S. Darmawan, Y. Yamaji, and M. Imada, Restricted Boltzmann machine learning for solving strongly correlated quantum systems, Phys. Rev. B 96, 205152 (2017).
  22. Y. Nomura and M. Imada, Dirac-Type Nodal Spin Liquid Revealed by Refined Quantum Many-Body Solver Using Neural-Network Wave Function, Correlation Ratio, and Level Spectroscopy, Phys. Rev. X 11, 031034 (2021).
  23. X. Liang, S.-J. Dong, and L. He, Hybrid convolutional neural network and projected entangled pair states wave functions for quantum many-particle states, Phys. Rev. B 103, 035138 (2021).
  24. T. Westerhout, N. Astrakhantsev, K. S. Tikhonov, M. I. Katsnelson, and A. A. Bagrov, Generalization properties of neural network approximations to frustrated magnet ground states, Nat. Commun. 11, 1593 (2020).
  25. N. Astrakhantsev, T. Westerhout, A. Tiwari, K. Choo, A. Chen, M. H. Fischer, G. Carleo, and T. Neupert, Broken-Symmetry Ground States of the Heisenberg Model on the Pyrochlore Lattice, Phys. Rev. X 11, 041021 (2021).
  26. C. Audet and W. Hare, Derivative-Free and Blackbox Optimization (Springer, Gewerbestrasse 11, 6330 Cham, Switzerland, 2017).
  27. N. Hansen, The CMA evolution strategy: A tutorial, arXiv:1604.00772.
  28. I. Hagymási, R. Schäfer, R. Moessner, and D. J. Luitz, Possible Inversion Symmetry Breaking in the s=1/2 Pyrochlore Heisenberg Magnet, Phys. Rev. Lett. 126, 117204 (2021).
  29. See Supplemental Material at http://link.aps.org/supplemental/10.1103/PhysRevResearch.4.L022026 for algorithm details and supplemental data.
  30. G. Torlai, G. Mazzola, J. Carrasquilla, M. Troyer, R. Melko, and G. Carleo, Neural-network quantum state tomography, Nat. Phys. 14, 447 (2018).
  31. S. Sorella, M. Casula, and D. Rocca, Weak binding between two aromatic rings: Feeling the van der Waals attraction by quantum Monte Carlo methods, J. Chem. Phys. 127, 014105 (2007).
  32. N. Hansen, A. Auger, R. Ros, S. Finck, and P. Pošík, Comparing results of 31 algorithms from the black-box optimization benchmarking bbob-2009, in Proceedings of the 12th Annual Conference Companion on Genetic and Evolutionary Computation (Association for Computing Machinery, Portland, Oregon, USA, 2010), pp. 1689–1696.
  33. N. Hansen, Y. Akimoto, and P. Baudis, CMA-ES/pycma.
  34. S.-S. Gong, W. Zhu, D. N. Sheng, O. I. Motrunich, and M. P. A. Fisher, Plaquette Ordered Phase and Quantum Phase Diagram in the Spin-12J1−J2 Square Heisenberg Model, Phys. Rev. Lett. 113, 027201 (2014).
  35. Y. Iqbal, T. Müller, P. Ghosh, M. J. P. Gingras, H. O. Jeschke, S. Rachel, J. Reuther, and R. Thomale, Quantum and Classical Phases of the Pyrochlore Heisenberg Model with Competing Interactions, Phys. Rev. X 9, 011005 (2019).
  36. W. Marshall and R. E. Peierls, Antiferromagnetism, Proc. R. Soc. London, Ser. A 232, 48 (1955).
  37. X. Liang, W.-Y. Liu, P.-Z. Lin, G.-C. Guo, Y.-S. Zhang, and L. He, Solving frustrated quantum many-particle models with convolutional neural networks, Phys. Rev. B 98, 104426 (2018).

Outline

Information

Sign In to Your Journals Account

Filter

Filter

Article Lookup

Enter a citation