journal article·2025-07-01Open access

Data-driven low-rank approximation for the electron-hole kernel and acceleration of time-dependent GW calculations

Bowen Hou, Jinyuan Wu, Victor Chang Lee, Jiaxuan Guo, Luna Y. Liu, Diana Y. Qiu

npj Computational Materials· Volume 11· Issue 1

Abstract

Many-body interactions are essential for understanding non-linear optics and ultrafast spectroscopy of materials. Recent first principles approaches based on nonequilibrium Green’s function formalisms, such as the time-dependent adiabatic GW (TD-aGW) approach, can predict nonequilibrium dynamics of excited states including electron-hole interactions. However, the high-dimensionality of the electron-hole kernel poses significant computational challenges. Here, we develop a data-driven low-rank approximation for the electron-hole kernel, leveraging localized excitonic effects in the Hilbert space of crystalline systems to achieve significant data compression through singular value decomposition (SVD). We show that the subspace of non-zero singular values remains small even as the k-grid grows, ensuring computational tractability with extremely dense k-grids. This low-rank property enables at least 95% data compression and an order-of-magnitude speedup of TD-aGW calculations. Our approach avoids intensive training processes and eliminates time-accumulated errors, seen in previous approaches, providing a general framework for high-throughput, nonequilibrium simulation of light-driven dynamics in materials.
Source: openalex_inverted_index · Quality: baseline

Research context

OpenAlex topics: Sparse and Compressive Sensing Techniques · Model Reduction and Neural Networks · Electromagnetic Scattering and Analysis

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

References (83)

Show source-provided references
1. Leone, S. R. et al. What will it take to observe processes in ’real time’? Nat. Photonics 8, 162–166 (2014).10.1038/nphoton.2014.48 ↗
2. Ramasesha, K., Leone, S. R. & Neumark, D. M. Real-time probing of electron dynamics using attosecond time-resolved spectroscopy. Annu. Rev. Phys. Chem. 67, 41–63 (2016).10.1146/annurev-physchem-040215-112025 ↗
3. Liu, H. et al. High-harmonic generation from an atomically thin semiconductor. Nat. Phys. 13, 262–265 (2017).10.1038/nphys3946 ↗
4. Ulbricht, R., Hendry, E., Shan, J., Heinz, T. F. & Bonn, M. Carrier dynamics in semiconductors studied with time-resolved terahertz spectroscopy. Rev. Mod. Phys. 83, 543–586 (2011).10.1103/RevModPhys.83.543 ↗
5. Boschini, F., Zonno, M. & Damascelli, A. Time-resolved ARPES studies of quantum materials. Rev. Mod. Phys. 96, 015003 (2024).10.1103/RevModPhys.96.015003 ↗
6. Cavalleri, A. Photo-induced superconductivity. Contemp. Phys. 59, 31–46 (2018).10.1080/00107514.2017.1406623 ↗
7. Chang Lee, V., Yue, L., Gaarde, M. B., Chan, Y.-h & Qiu, D. Y. Many-body enhancement of high-harmonic generation in monolayer MoS2. Nat. Commun. 15, 6228 (2024).10.1038/s41467-024-50534-3 ↗
8. Chan, Y.-H., Qiu, D. Y., da Jornada, F. H. & Louie, S. G. Giant exciton-enhanced shift currents and direct current conduction with subbandgap photo excitations produced by many-electron interactions. Proc. Natl. Acad. Sci. USA 118, e1906938118 (2021).10.1073/pnas.1906938118 ↗
9. Chan, Y.-H., Qiu, D. Y., da Jornada, F. H. & Louie, S. G. Giant self-driven exciton-floquet signatures in time-resolved photoemission spectroscopy of MoS2 from time-dependent GW approach. Proc. Natl. Acad. Sci. USA 120, e2301957120 (2023).10.1073/pnas.2301957120 ↗
10. Man, M. K. et al. Experimental measurement of the intrinsic excitonic wave function. Sci. Adv. 7, eabg0192 (2021).10.1126/sciadv.abg0192 ↗
11. Madéo, J. et al. Directly visualizing the momentum-forbidden dark excitons and their dynamics in atomically thin semiconductors. Science 370, 1199–1204 (2020).10.1126/science.aba1029 ↗
12. Lin, Y. et al. Exciton-driven renormalization of quasiparticle band structure in monolayer MoS2. Phys. Rev. B 106, L081117 (2022).10.1103/PhysRevB.106.L081117 ↗
13. Attaccalite, C., Grüning, M. & Marini, A. Real-time approach to the optical properties of solids and nanostructures: time-dependent Bethe-Salpeter equation. Phys. Rev. B Condens. Matter Mater. Phys. 84, 245110 (2011).10.1103/PhysRevB.84.245110 ↗
14. Perfetto, E., Sangalli, D., Marini, A. & Stefanucci, G. Nonequilibrium Bethe-Salpeter equation for transient photoabsorption spectroscopy. Phys. Rev. B 92, 205304 (2015).10.1103/PhysRevB.92.205304 ↗
15. Perfetto, E., Pavlyukh, Y. & Stefanucci, G. Real-time GW: toward an ab initio description of the ultrafast carrier and exciton dynamics in two-dimensional materials. Phys. Rev. Lett. 128, 016801 (2022).10.1103/PhysRevLett.128.016801 ↗
16. Marques, M. A. & Gross, E. K. Time-dependent density functional theory. Annu. Rev. Phys. Chem. 55, 427–455 (2004).10.1146/annurev.physchem.55.091602.094449 ↗
17. Reining, L., Olevano, V., Rubio, A. & Onida, G. Excitonic effects in solids described by time-dependent density-functional theory. Phys. Rev. Lett. 88, 066404 (2002).10.1103/PhysRevLett.88.066404 ↗
18. Romaniello, P. et al. Double excitations in finite systems. J. Chem. Phys. 130, 044108 (2009).10.1063/1.3065669 ↗
19. Sangalli, D. Excitons and carriers in transient absorption and time-resolved ARPES spectroscopy: an ab initio approach. Phys. Rev. Mater. 5, 083803 (2021).10.1103/PhysRevMaterials.5.083803 ↗
20. Luo, Y., Desai, D., Chang, B. K., Park, J. & Bernardi, M. Data-driven compression of electron-phonon interactions. Phys. Rev. X 14, 021023 (2024).
21. Lu, J. & Ying, L. Compression of the electron repulsion integral tensor in tensor hypercontraction format with cubic scaling cost. J. Computational Phys. 302, 329–335 (2015).10.1016/j.jcp.2015.09.014 ↗
22. Shao, M. et al. Low rank approximation in G0W0 calculations. Sci. China Math. 59, 1593–1612 (2016).10.1007/s11425-016-0296-x ↗
23. Kaye, J., Chen, K. & Parcollet, O. Discrete Lehmann representation of imaginary time Green’s functions. Phys. Rev. B 105, 235115 (2022).10.1103/PhysRevB.105.235115 ↗
24. Dunlap, B. I. Robust and variational fitting. Phys. Chem. Chem. Phys. 2, 2113–2116 (2000).10.1039/b000027m ↗
25. Pham, B. Q. & Gordon, M. S. Compressing the four-index two-electron repulsion integral matrix using the resolution-of-the-identity approximation combined with the rank factorization approximation. J. Chem. Theory Comput. 15, 2254–2264 (2019).10.1021/acs.jctc.8b01256 ↗
26. Del Ben, M. et al. Static subspace approximation for the evaluation of G0W0 quasiparticle energies within a sum-over-bands approach. Phys. Rev. B 99, 125128 (2019).10.1103/PhysRevB.99.125128 ↗
27. Baratz, A., Cohen, G. & Refaely-Abramson, S. (2024) Unsupervised learning approach to quantum wave-packet dynamics from coupled temporal-spatial correlations. Physical Review B 110, https://doi.org/10.1103/PhysRevB.110.13430410.1103/PhysRevB.110.134304 ↗
28. Zang, J, et al. Machine learning-based compression of quantum many body physics: PCA and autoencoder representation of the vertex function. Machine Learning: Science and Technology 5.4, 045076 (2024).10.1088/2632-2153/ad9f20 ↗
29. Shinaoka, H., Otsuki, J., Ohzeki, M. & Yoshimi, K. Compressing Green’s function using intermediate representation between imaginary-time and real-frequency domains. Phys. Rev. B 96, 035147 (2017).10.1103/PhysRevB.96.035147 ↗
30. Zadoks, A., Marrazzo, A. & Marzari, N. Spectral operator representations. npj Comput Mater 10, 278 (2024).10.1038/s41524-024-01446-9 ↗
31. Knøsgaard, N. R. & Thygesen, K. S. Representing individual electronic states for machine learning GW band structures of 2D materials. Nat. Commun. 13, 468 (2022).10.1038/s41467-022-28122-0 ↗
32. Hou, B., Wu, J. & Qiu, D. Y. Unsupervised representation learning of Kohn–Sham states and consequences for downstream predictions of many-body effects. Nat. Commun. 15, 9481 (2024).10.1038/s41467-024-53748-7 ↗
33. Luchnikov, I. A., Ryzhov, A., Stas, P.-J., Filippov, S. N. & Ouerdane, H. Variational autoencoder reconstruction of complex many-body physics. Entropy 21, 1091 (2019).10.3390/e21111091 ↗
34. Batzner, S. et al. E (3)-equivariant graph neural networks for data-efficient and accurate interatomic potentials. Nat. Commun. 13, 2453 (2022).10.1038/s41467-022-29939-5 ↗
35. Thomas, N. et al. Tensor field networks: rotation-and translation-equivariant neural networks for 3D point clouds. arXiv preprint. https://arxiv.org/abs/1802.08219 (2018).
36. Li, H. et al. Deep-learning density functional perturbation theory. Phys. Rev. Lett. 132, 096401 (2024).10.1103/PhysRevLett.132.096401 ↗
37. Yang, L. et al. Deep learning-enhanced variational Monte Carlo method for quantum many-body physics. Phys. Rev. Res. 2, 012039 (2020).10.1103/PhysRevResearch.2.012039 ↗
38. Medvidović, M., Moreno, J.R. Neural-network quantum states for many-body physics. Eur. Phys. J. Plus 139, 631 (2024).10.1140/epjp/s13360-024-05311-y ↗
39. Noé, F., Olsson, S., Köhler, J. & Wu, H. Boltzmann generators: Sampling equilibrium states of many-body systems with deep learning. Science 365, eaaw1147 (2019).10.1126/science.aaw1147 ↗
40. Carrasquilla, J. & Torlai, G. How to use neural networks to investigate quantum many-body physics. PRX Quantum 2, 040201 (2021).10.1103/PRXQuantum.2.040201 ↗
41. Yin, J. et al. Analyzing and predicting non-equilibrium many-body dynamics via dynamic mode decomposition. J. Comput. Phys. 477, 111909 (2023).10.1016/j.jcp.2023.111909 ↗
42. Maliyov, I., Yin, J., Yao, J., Yang, C. & Bernardi, M. Dynamic mode decomposition of nonequilibrium electron-phonon dynamics: accelerating the first-principles real-time Boltzmann equation. npj Comput. Mater. 10, 123 (2024).10.1038/s41524-024-01308-4 ↗
43. Reeves, C. C. et al. Dynamic mode decomposition for extrapolating nonequilibrium Green’s-function dynamics. Phys. Rev. B 107, 075107 (2023).10.1103/PhysRevB.107.075107 ↗
44. Gu, M., Lin, Y., Lee, V. C. & Qiu, D. Y. Probabilistic forecast of nonlinear dynamical systems with uncertainty quantification. Phys. D: Nonlinear Phenom. 457, 133938 (2024).10.1016/j.physd.2023.133938 ↗
45. Lu, L., Jin, P., Pang, G., Zhang, Z. & Karniadakis, G. E. Learning nonlinear operators via DeepONet based on the universal approximation theorem of operators. Nat. Mach. Intell. 3, 218–229 (2021).10.1038/s42256-021-00302-5 ↗
46. Li, Z. et al. Fourier neural operator for parametric partial differential equations. arXiv preprint. https://arxiv.org/abs/2010.08895 (2020).
47. Kovachki, N. et al. Neural operator: learning maps between function spaces with applications to PDEs. J. Mach. Learn. Res. 24, 1–97 (2023).
48. Li, Z. et al. Neural operator: Graph kernel network for partial differential equations. arXiv preprint. https://arxiv.org/abs/2003.03485 (2020).
49. Li, Z., Meidani, K. & Farimani, A. B. Transformer for partial differential equations’ operator learning. arXiv preprint. https://arxiv.org/abs/2205.13671 (2022).
50. Zhu, Y. et al. Predicting nonequilibrium Green’s function dynamics and photoemission spectra via nonlinear integral operator learning. Machine Learning: Science and Technology 6.1, 015027 (2025).
51. Brunton, S. L. & Kutz, J. N. Data-driven Science and Engineering: Machine Learning, Dynamical Systems, and Control (Cambridge University Press, 2022).10.1017/9781009089517 ↗
52. Shwartz-Ziv, R. & Tishby, N. Opening the black box of deep neural networks via information. arXiv preprint. https://arxiv.org/abs/1703.00810 (2017).
53. Kremp, D., Schlanges, M. & Kraeft, W.-D. Quantum Statistics of Nonideal Plasmas, Vol. 25 (Springer Science & Business Media, 2005).
54. Kadanoff, L. P. Quantum Statistical Mechanics (CRC Press, 2018).10.1201/9780429493218 ↗
55. Kadanoff, L., Baym, G. & Mechanics, Q. S. Green’s function methods in equilibrium and nonequilibrium problems. Front. Phys. (1962).
56. Hedin, L. New method for calculating the one-particle Green's function with application to the electron-gas problem. Phys. Rev. 139, A796 (1965).10.1103/PhysRev.139.A796 ↗
57. Hedin, L. & Lundqvist, S. Effects of electron-electron and electron-phonon interactions on the one-electron states of solids. in Solid State Physics, Vol. 23, 1–181 (Elsevier, 1970).10.1016/S0081-1947(08)60615-3 ↗
58. Hybertsen, M. S. & Louie, S. G. Electron correlation in semiconductors and insulators: band gaps and quasiparticle energies. Phys. Rev. B 34, 5390 (1986).10.1103/PhysRevB.34.5390 ↗
59. Adler, S. L. Quantum theory of the dielectric constant in real solids. Phys. Rev. 126, 413 (1962).10.1103/PhysRev.126.413 ↗
60. Wiser, N. Dielectric constant with local field effects included. Phys. Rev. 129, 62 (1963).10.1103/PhysRev.129.62 ↗
61. Aversa, C. & Sipe, J. E. Nonlinear optical susceptibilities of semiconductors: results with a length-gauge analysis. Phys. Rev. B 52, 14636 (1995).10.1103/PhysRevB.52.14636 ↗
62. Virk, K. S. & Sipe, J. Semiconductor optics in length gauge: a general numerical approach. Phys. Rev. B Condens. Matter Mater. Phys. 76, 035213 (2007).10.1103/PhysRevB.76.035213 ↗
63. Rocca, D., Ping, Y., Gebauer, R. & Galli, G. Solution of the Bethe-Salpeter equation without empty electronic states: application to the absorption spectra of bulk systems. Phys. Rev. B—Condens. Matter Mater. Phys. 85, 045116 (2012).10.1103/PhysRevB.85.045116 ↗
64. Rohlfing, M. & Louie, S. G. Electron-hole excitations and optical spectra from first principles. Phys. Rev. B 62, 4927 (2000).10.1103/PhysRevB.62.4927 ↗
65. Albrecht, S., Reining, L., Del Sole, R. & Onida, G. Ab initio calculation of excitonic effects in the optical spectra of semiconductors. Phys. Rev. Lett. 80, 4510 (1998).10.1103/PhysRevLett.80.4510 ↗
66. Deslippe, J. et al. BerkeleyGW: a massively parallel computer package for the calculation of the quasiparticle and optical properties of materials and nanostructures. Comput. Phys. Commun. 183, 1269–1289 (2012).10.1016/j.cpc.2011.12.006 ↗
67. Rohlfing, M. & Louie, S. G. Electron-hole excitations and optical spectra from first principles. Phys. Rev. B 62, 4927–4944 (2000).10.1103/PhysRevB.62.4927 ↗
68. Strinati, G. Application of the Green’s functions method to the study of the optical properties of semiconductors. La Riv. del. Nuovo Cim. 11, 1–86 (1988).10.1007/BF02725962 ↗
69. Giannozzi, P. et al. Quantum espresso: a modular and open-source software project for quantum simulations of materials. J. Phys. Condens. Matter 21, 395502 (2009).10.1088/0953-8984/21/39/395502 ↗
70. Kohn, W. & Sham, L. J. Self-consistent equations including exchange and correlation effects. Phys. Rev. 140, A1133 (1965).10.1103/PhysRev.140.A1133 ↗
71. Kammerlander, D., Botti, S., Marques, M. A. L., Marini, A. & Attaccalite, C. Speeding up the solution of the Bethe-Salpeter equation by a double-grid method and Wannier interpolation. Phys. Rev. B Condens. Matter Mater. Phys. 86, 125203 (2012).10.1103/PhysRevB.86.125203 ↗
72. Qiu, D. Y., Da Jornada, F. H. & Louie, S. G. Optical spectrum of MoS2: many-body effects and diversity of exciton states. Phys. Rev. Lett. 111, 216805 (2013).10.1103/PhysRevLett.111.216805 ↗
73. Hou, B., Wang, D., Barker, B. A. & Qiu, D. Y. Exchange-driven intermixing of bulk and topological surface states by chiral excitons in Bi2Se3. Phys. Rev. Lett. 130, 216402 (2023).10.1103/PhysRevLett.130.216402 ↗
74. Alvertis, A. M. et al. Importance of nonuniform Brillouin zone sampling for ab initio Bethe-Salpeter equation calculations of exciton binding energies in crystalline solids. Phys. Rev. B 108, 235117 (2023).10.1103/PhysRevB.108.235117 ↗
75. Qiu, D. Y., da Jornada, F. H. & Louie, S. G. Screening and many-body effects in two-dimensional crystals: monolayer MoS2. Phys. Rev. B 93, 235435 (2016).10.1103/PhysRevB.93.235435 ↗
76. Li, L. et al. Direct observation of the layer-dependent electronic structure in phosphorene. Nat. Nanotechnol. 12, 21–25 (2017).10.1038/nnano.2016.171 ↗
77. Tempelaar, R. & Berkelbach, T. C. Many-body simulation of two-dimensional electronic spectroscopy of excitons and trions in monolayer transition metal dichalcogenides. Nat. Commun. 10, 3419 (2019).10.1038/s41467-019-11497-y ↗
78. Wu, J., Hou, B., Li, W., He, Y. & Qiu, D. Y. Quasiparticle and excitonic properties of monolayer $$1{T}^{{\prime} }\,{{\rm{WTe}}}2$$ within many-body perturbation theory. Phys. Rev. B 110, 075133 (2024).10.1103/PhysRevB.110.075133 ↗
79. Cao, T., Li, Z., Qiu, D. Y. & Louie, S. G. Gate switchable transport and optical anisotropy in 90 twisted bilayer black phosphorus. Nano Lett. 16, 5542–5546 (2016).10.1021/acs.nanolett.6b02084 ↗
80. Qiu, D. Y., da Jornada, F. H. & Louie, S. G. Environmental screening effects in 2D materials: renormalization of the bandgap, electronic structure, and optical spectra of few-layer black phosphorus. Nano Lett. 17, 4706–4712 (2017).10.1021/acs.nanolett.7b01365 ↗
81. Qiu, D. Y., da Jornada, F. H. & Louie, S. G. Solving the Bethe-Salpeter equation on a subspace: approximations and consequences for low-dimensional materials. Phys. Rev. B 103, 045117 (2021).10.1103/PhysRevB.103.045117 ↗
82. Perdew, J. P., Burke, K. & Ernzerhof, M. Generalized gradient approximation made simple. Phys. Rev. Lett. 77, 3865–3868 (1996).10.1103/PhysRevLett.77.3865 ↗
83. Hybertsen, M. S. & Louie, S. G. Electron correlation in semiconductors and insulators: Band gaps and quasiparticle energies. Phys. Rev. B 34, 5390–5413 (1986).10.1103/PhysRevB.34.5390 ↗