Orbital Transformers for Predicting Wavefunctions in Time-Dependent Density Functional Theory

• URL: http://arxiv.org/abs/2603.03511v1
• Date: Tue, 03 Mar 2026 20:34:05 GMT
• Title: Orbital Transformers for Predicting Wavefunctions in Time-Dependent Density Functional Theory
• Authors: Xuan Zhang, Haiyang Yu, Chengdong Wang, Jacob Helwig, Shuiwang Ji, Xiaofeng Qian,
• Abstract summary: We aim to learn wavefunctions simulated by time-dependent density functional theory (TDDFT)<n>In real-time TDDFT, the electronic wavefunctions of a molecule evolve over time in response to an external excitation.<n>We propose OrbEvo, which learns to evolve the full electronic wavefunction coefficients across time steps.
• Score: 45.322818939044645
• License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
• Abstract: We aim to learn wavefunctions simulated by time-dependent density functional theory (TDDFT), which can be efficiently represented as linear combination coefficients of atomic orbitals. In real-time TDDFT, the electronic wavefunctions of a molecule evolve over time in response to an external excitation, enabling first-principles predictions of physical properties such as optical absorption, electron dynamics, and high-order response. However, conventional real-time TDDFT relies on time-consuming propagation of all occupied states with fine time steps. In this work, we propose OrbEvo, which is based on an equivariant graph transformer architecture and learns to evolve the full electronic wavefunction coefficients across time steps. First, to account for external field, we design an equivariant conditioning to encode both strength and direction of external electric field and break the symmetry from SO(3) to SO(2). Furthermore, we design two OrbEvo models, OrbEvo-WF and OrbEvo-DM, using wavefunction pooling and density matrix as interaction method, respectively. Motivated by the central role of the density functional in TDDFT, OrbEvo-DM encodes the density matrix aggregated from all occupied electronic states into feature vectors via tensor contraction, providing a more intuitive approach to learn the time evolution operator. We adopt a training strategy specifically tailored to limit the error accumulation of time-dependent wavefunctions over autoregressive rollout. To evaluate our approach, we generate TDDFT datasets consisting of 5,000 different molecules in the QM9 dataset and 1,500 molecular configurations of the malonaldehyde molecule in the MD17 dataset. Results show that our OrbEvo model accurately captures quantum dynamics of excited states under external field, including time-dependent wavefunctions, time-dependent dipole moment, and optical absorption spectra.

Related papers

• Determining Molecular Ground State with Quantum Imaginary Time Evolution using Broken-Symmetry Wave Function [0.44998333629984877]
  We propose replacing the Hartree-Fock wave function with a spin- and spatial-symmetry broken wave function.<n>We show that this approach provides good convergence behavior towards the lowest energy eigenstate.<n>This method provides a pathway for improved energy simulations in open shell systems.
  arXiv  Detail & Related papers  (2025-04-25T08:12:34Z)
• RadioDiff-$k^2$: Helmholtz Equation Informed Generative Diffusion Model for Multi-Path Aware Radio Map Construction [76.24833675757033]
  We propose a physics-informed generative learning approach, named RadioDiff-$k2$, for accurate and efficient multipath-aware radio map (RM) construction.<n>We show that the proposed RadioDiff-$k2$ framework achieves state-of-the-art (SOTA) performance in both image-level RM construction and localization tasks.
  arXiv  Detail & Related papers  (2025-04-22T06:28:13Z)
• Accelerating Electron Dynamics Simulations through Machine Learned Time Propagators [0.9208007322096533]
  We present a novel approach to accelerate real time TDDFT based electron dynamics simulations. By leveraging physics-informed constraints and high-resolution training data, our model achieves superior accuracy and computational speed. This method has potential in enabling real-time, on-the-fly modeling of laser-irradiated molecules and materials.
  arXiv  Detail & Related papers  (2024-07-12T18:29:48Z)
• Ab-initio variational wave functions for the time-dependent many-electron Schrödinger equation [41.94295877935867]
  We introduce a variational approach for fermionic time-dependent wave functions, surpassing mean-field approximations. We use time-dependent Jastrow factors and backflow transformations, which are enhanced through neural networks parameterizations. The results showcase the ability of our variational approach to accurately capture the time evolution, providing insight into the quantum dynamics of interacting electronic systems.
  arXiv  Detail & Related papers  (2024-03-12T09:37:22Z)
• Interpolating many-body wave functions for accelerated molecular dynamics on the near-exact electronic surface [0.0]
  We develop a scheme for the correlated many-electron state through the space of atomic configurations. We demonstrate provable convergence to near-exact potential energy surfaces for subsequent dynamics. We combine this with modern electronic structure approaches to systematically resolve molecular dynamics trajectories.
  arXiv  Detail & Related papers  (2024-02-16T22:03:37Z)
• Transform Once: Efficient Operator Learning in Frequency Domain [69.74509540521397]
  We study deep neural networks designed to harness the structure in frequency domain for efficient learning of long-range correlations in space or time. This work introduces a blueprint for frequency domain learning through a single transform: transform once (T1)
  arXiv  Detail & Related papers  (2022-11-26T01:56:05Z)
• Real-Space, Real-Time Approach to Quantum-Electrodynamical Time-Dependent Density Functional Theory [55.41644538483948]
  The equations are solved by time propagating the wave function on a tensor product of a Fock-space and real-space grid. Examples include the coupling strength and light frequency dependence of the energies, wave functions, optical absorption spectra, and Rabi splitting magnitudes in cavities.
  arXiv  Detail & Related papers  (2022-09-01T18:49:51Z)
• Computing molecular excited states on a D-Wave quantum annealer [52.5289706853773]
  We demonstrate the use of a D-Wave quantum annealer for the calculation of excited electronic states of molecular systems. These simulations play an important role in a number of areas, such as photovoltaics, semiconductor technology and nanoscience.
  arXiv  Detail & Related papers  (2021-07-01T01:02:17Z)
• Controlled coherent dynamics of [VO(TPP)], a prototype molecular nuclear qudit with an electronic ancilla [50.002949299918136]
  We show that [VO(TPP)] (vanadyl tetraphenylporphyrinate) is a promising system suitable to implement quantum computation algorithms. It embeds an electronic spin 1/2 coupled through hyperfine interaction to a nuclear spin 7/2, both characterized by remarkable coherence.
  arXiv  Detail & Related papers  (2021-03-15T21:38:41Z)
• Predicting molecular dipole moments by combining atomic partial charges and atomic dipoles [3.0980025155565376]
  "MuML" models are fitted together to reproduce molecular $boldsymbolmu$ computed using high-level coupled-cluster theory. We demonstrate that the uncertainty in the predictions can be estimated reliably using a calibrated committee model.
  arXiv  Detail & Related papers  (2020-03-27T14:35:37Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.