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 quantum propagators for driven-dissipative quantum dynamics

Jiaji Zhang1, Carlos L. Benavides-Riveros2,*, and Lipeng Chen1,†

  • *Contact author: cl.benavidesriveros@unitn.it
  • †Contact author: chenlp@zhejianglab.com

Phys. Rev. Research 7, L012013 – Published 21 January, 2025

DOI: https://doi.org/10.1103/PhysRevResearch.7.L012013

Abstract

Describing the dynamics of strong-laser driven open quantum systems is a very challenging task that requires the solution of highly involved equations of motion. While machine learning techniques are being applied with some success to simulate the time evolution of individual quantum states, their use to approximate time-dependent operators (that can evolve various states) remains largely unexplored. In this work, we develop driven neural quantum propagators (NQP), a universal neural network framework that solves driven-dissipative quantum dynamics by approximating propagators rather than wave functions or density matrices. NQP can handle arbitrary initial quantum states, adapt to various external fields, and simulate long-time dynamics, even when trained on far shorter time windows. Furthermore, by appropriately configuring the external fields, our trained NQP can be transferred to systems governed by different Hamiltonians. We demonstrate the effectiveness of our approach by studying the spin-boson and the three-state transition Gamma models.

View figure in article

Physics Subject Headings (PhySH)

Article Text

References (57)

  1. D. Jonas, Two-dimensional femtosecond spectroscopy, Annu. Rev. Phys. Chem. 54, 425 (2003).
  2. C. Consani, S. Ruetzel, P. Nuernberger, and T. Brixner, Quantum control spectroscopy of competing reaction pathways in a molecular switch, J. Phys. Chem. A 118, 11364 (2014).
  3. D. Keefer and S. Mukamel, Selective enhancement of spectroscopic features by quantum optimal control, Phys. Rev. Lett. 126, 163202 (2021).
  4. M. F. Gelin, D. Egorova, and W. Domcke, Strong and long makes short: Strong-pump strong-probe spectroscopy, J. Phys. Chem. Lett. 2, 114 (2011).
  5. M. Woerner, A. Ghalgaoui, K. Reimann, and T. Elsaesser, Two-color two-dimensional terahertz spectroscopy: A new approach for exploring even-order nonlinearities in the nonperturbative regime, J. Chem. Phys. 154, 154203 (2021).
  6. D. Egorova, M. F. Gelin, M. Thoss, H. Wang, and W. Domcke, Effects of intense femtosecond pumping on ultrafast electronic-vibrational dynamics in molecular systems with relaxation, J. Chem. Phys. 129, 214303 (2008).
  7. S. Mukamel, Principles of Nonlinear Optical Spectroscopy (Oxford University Press, New York, 1995).
  8. P. Hoff, S. Thallmair, M. Kowalewski, R. Siemering, and R. Vivie-Riedle, Optimal control theory closing the gap between theory and experiment, Phys. Chem. Chem. Phys. 14, 14460 (2012).
  9. A. Norambuena, M. Mattheakis, F. J. González, and R. Coto, Physics-informed neural networks for quantum control, Phys. Rev. Lett. 132, 010801 (2024).
  10. S. Gupta, H. K. Yadalam, M. Kulkarni, and C. Aron, Quantum jumps in driven-dissipative disordered many-body systems, Phys. Rev. A 109, L050201 (2024).
  11. A. Ullah, Y. Huang, M. Yang, and P. O. Dral, Physics-informed neural networks and beyond: Enforcing physical constraints in quantum dissipative dynamics, Digit. Discov. 3, 2052 (2024).
  12. M. Reh, M. Schmitt, and M. Gärttner, Time-dependent variational principle for open quantum systems with artificial neural networks, Phys. Rev. Lett. 127, 230501 (2021).
  13. P. Kramer and M. Saraceno, Geometry of the Time-Dependent Variational Principle in Quantum Mechanics, Lecture Notes in Physics (Springer, Berlin, 1981).
  14. F. Vicentini, A. Biella, N. Regnault, and C. Ciuti, Variational neural-network ansatz for steady states in open quantum systems, Phys. Rev. Lett. 122, 250503 (2019).
  15. J. A. Keith, V. Vassilev-Galindo, B. Cheng, S. Chmiela, M. Gastegger, K.-R. Müller, and A. Tkatchenko, Combining machine learning and computational chemistry for predictive insights into chemical systems, Chem. Rev. 121, 9816 (2021).
  16. M. Ceriotti, C. Clementi, and O. Anatole von Lilienfeld, Machine learning meets chemical physics, J. Chem. Phys. 154, 160401 (2021).
  17. H. Lange, A. V. de Walle, A. Abedinnia, and A. Bohrdt, From architectures to applications: A review of neural quantum states, Quantum Sci. Technol. 9, 040501 (2024).
  18. J. Jumper et al., Highly accurate protein structure prediction with AlphaFold, Nature (London) 596, 583 (2021).
  19. F. Noé, A. Tkatchenko, K. Müller, and C. Clementi, Machine learning for molecular simulation, Annu. Rev. Phys. Chem. 71, 361 (2020).
  20. X. Du, W. Shao, C. Bao, L. Zhang, J. Cheng, and F. Tang, Revealing the molecular structures of α−Al2O3(0001)water interface by machine learning based computational vibrational spectroscopy, J. Chem. Phys. 161, 124702 (2024).
  21. O. T. Unke, S. Chmiela, H. E. Sauceda, M. Gastegger, I. Poltavsky, K. T. Schütt, A. Tkatchenko, and K.-R. Müller, Machine learning force fields, Chem. Rev. 121, 10142 (2021).
  22. V. Martinetto, K. Shah, A. Cangi, and A. Pribram-Jones, Inverting the KohnSham equations with physics-informed machine learning, Mach. Learn.: Sci. Technol. 5, 015050 (2024).
  23. J. Gedeon, J. Schmidt, M. J. P. Hodgson, J. Wetherell, C. L. Benavides-Riveros, and M. A. L. Marques, Machine learning the derivative discontinuity of density-functional theory, Mach. Learn.: Sci. Technol. 3, 015011 (2022).
  24. R. Nagai, R. Akashi, and O. Sugino, Completing density functional theory by machine learning hidden messages from molecules, npj Comput. Mater. 6, 43 (2020).
  25. J. Kirkpatrick et al., Pushing the frontiers of density functionals by solving the fractional electron problem, Science 374, 1385 (2021).
  26. F. Sammüller, S. Robitschko, S. Hermann, and M. Schmidt, Hyperdensity functional theory of soft matter, Phys. Rev. Lett. 133, 098201 (2024).
  27. X. Gong, S. G. Louie, W. Duan, and Y. Xu, Generalizing deep learning electronic structure calculation to the plane-wave basis, Nat. Comput. Sci. 4, 752 (2024).
  28. R. Vinuesa and S. L. Brunton, Enhancing computational fluid dynamics with machine learning, Nat. Comput. Sci. 2, 358 (2022).
  29. S. L. Brunton and J. N. Kutz, Promising directions of machine learning for partial differential equations, Nat. Comput. Sci. 4, 483 (2024).
  30. N. Kovachki, S. Lanthaler, and S. Mishra, On universal approximation and error bounds for fourier neural operators, J. Mach. Learn. Res. 22, 1 (2021).
  31. Z. Li, N. B. Kovachki, K. Azizzadenesheli, B. Liu, K. Bhattacharya, A. Stuart, and A. Anandkumar, Fourier neural operator for parametric partial differential equations, arXiv:2010.08895.
  32. S. Mizera, Scattering with neural operators, Phys. Rev. D 108, L101701 (2023).
  33. V. Niarchos and C. Papageorgakis, Learning s-matrix phases with neural operators, Phys. Rev. D 110, 045020 (2024).
  34. J. Zhang, C. L. Benavides-Riveros, and L. Chen, Artificial-intelligence-based surrogate solution of dissipative quantum dynamics: Physics-informed reconstruction of the universal propagator, J. Phys. Chem. Lett. 15, 3603 (2024).
  35. N. McGreivy and A. Hakim, Weak baselines and reporting biases lead to overoptimism in machine learning for fluid-related partial differential equations, Nat. Mach. Intell. 6, 1256 (2024).
  36. B. Alkin, A. Fürst, S. L. Schmid, L. Gruber, M. Holzleitner, and J. Brandstetter, Universal physics transformers: A framework for efficiently scaling neural operators, in The Thirty-eighth Annual Conference on Neural Information Processing Systems, 2024.
  37. N. Kovachki, Z. Li, B. Liu, K. Azizzadenesheli, K. Bhattacharya, A. Stuart, and A. Anandkumar, Neural operator: Learning maps between function spaces with applications to PDEs, J. Mach. Learn. Res. 24, 1 (2023).
  38. U. Weiss, Quantum Dissipative Systems, 4th ed. (World Scientific, Singapore, 2012).
  39. H.-P. Breuer and F. Petruccione, The Theory of Open Quantum Systems (Oxford University Press, New York, 2007).
  40. F. Campaioli, J. H. Cole, and H. Hapuarachchi, Quantum master equations: tips and tricks for quantum optics, quantum computing, and beyond, PRX Quantum 5, 020202 (2024).
  41. D. Manzano, A short introduction to the Lindblad master equation, AIP Adv. 10, 025106 (2020).
  42. J. Zhang and L. Chen, A non-Markovian neural quantum propagator and its application to the simulation of ultrafast nonlinear spectra, Phys. Chem. Chem. Phys. 27, 182 (2025).
  43. J. Guibas, M. Mardani, Z. Li, A. Tao, A. Anandkumar, and B. Catanzaro, Efficient token mixing for transformers via adaptive Fourier neural operators, in International Conference on Learning Representations (2022).
  44. C. Salvi, M. Lemercier, and A. Gerasimovics, Neural stochastic PDEs: Resolution-invariant learning of continuous spatiotemporal dynamics, in Advances in Neural Information Processing Systems, edited by S. Koyejo, S. Mohamed, A. Agarwal, D. Belgrave, K. Cho, and A. Oh (Curran Associates, Red Hook, New York, 2022), Vol. 35, pp. 1333–1344.
  45. G. E. Karniadakis, I. G. Kevrekidis, L. Lu, P. Perdikaris, S. Wang, and L. Yang, Physics-informed machine learning, Nat. Rev. Phys. 3, 422 (2021).
  46. D. Ran, W.-J. Shan, Z.-C. Shi, Z.-B. Yang, J. Song, and Y. Xia, Pulse reverse engineering for controlling two-level quantum systems, Phys. Rev. A 101, 023822 (2020).
  47. W. Wu and Z.-Z. Zhang, Controllable dynamics of a dissipative two-level system, Sci. Rep. 11, 7188 (2021).
  48. J. R. Kuklinski, U. Gaubatz, F. T. Hioe, and K. Bergmann, Adiabatic population transfer in a three-level system driven by delayed laser pulses, Phys. Rev. A 40, 6741 (1989).
  49. K. Bergmann, H. Theuer, and B. W. Shore, Coherent population transfer among quantum states of atoms and molecules, Rev. Mod. Phys. 70, 1003 (1998).
  50. J. I. Cirac, D. Pérez-García, N. Schuch, and F. Verstraete, Matrix product states and projected entangled pair states: Concepts, symmetries, theorems, Rev. Mod. Phys. 93, 045003 (2021).
  51. J. Ren, W. Li, T. Jiang, Y. Wang, and Z. Shuai, Time- dependent density matrix renormalization group method for quantum dynamics in complex systems, WIREs Comput. Mol. Sci. 12, e1614 (2022).
  52. W. Li, J. Ren, and Z. Shuai, Numerical assessment for accuracy and GPU acceleration of TD-DMRG time evolution schemes, J. Chem. Phys. 152, 024127 (2020).
  53. R. Borrelli and M. F. Gelin, Finite temperature quantum dynamics of complex systems: Integrating thermo-field theories and tensor-train methods, WIREs Comput. Mol. Sci. 11, e1539 (2021).
  54. A. Novikov, D. Podoprikhin, A. Osokin, and D. P. Vetrov, Tensorizing neural networks, in Advances in Neural Information Processing Systems, edited by C. Cortes, N. Lawrence, D. Lee, M. Sugiyama, and R. Garnett (Curran Associates, Red Hook, New York, 2015), Vol. 28.
  55. D. Roberts and A. A. Clerk, Driven-dissipative quantum kerr resonators: New exact solutions, photon blockade and quantum bistability, Phys. Rev. X 10, 021022 (2020).
  56. C. Brif, R. Chakrabarti, and H. Rabitz, Control of quantum phenomena: Past, present and future, New J. Phys. 12, 075008 (2010).
  57. A. Castro, A. García Carrizo, S. Roca, D. Zueco, and F. Luis, Optimal control of molecular spin qudits, Phys. Rev. Appl. 17, 064028 (2022).

Outline

Information

Sign In to Your Journals Account

Filter

Filter

Article Lookup

Enter a citation