Accelerating Electron Dynamics Simulations through Machine Learned Time Propagators

Karan Shah,Attila Cangi
2024-07-26
Abstract:Time-dependent density functional theory (TDDFT) is a widely used method to investigate electron dynamics under various external perturbations such as laser fields. In this work, we present a novel approach to accelerate real time TDDFT based electron dynamics simulations using autoregressive neural operators as time-propagators for the electron density. By leveraging physics-informed constraints and high-resolution training data, our model achieves superior accuracy and computational speed compared to traditional numerical solvers. We demonstrate the effectiveness of our model on a class of one-dimensional diatomic molecules. This method has potential in enabling real-time, on-the-fly modeling of laser-irradiated molecules and materials with varying experimental parameters.
Materials Science,Machine Learning,Computational Physics
What problem does this paper attempt to address?
The paper aims to address the computational efficiency issues of Time-Dependent Density Functional Theory (TDDFT) in simulating electronic dynamics processes. Specifically, the authors propose a new method that utilizes Autoregressive Neural Operators as time propagators for electronic density to accelerate real-time TDDFT electronic dynamics simulations. The main contributions of the paper include: 1. **Proposing a time propagation algorithm based on Fourier Neural Operators (FNOs)**: This method directly evolves the electronic density over time, rather than propagating single-particle orbitals as in traditional methods. By doing so, the computational cost no longer increases with the number of orbitals, and larger time steps can be used, thereby reducing the number of iterations. 2. **Validating the effectiveness of the method**: By demonstrating the model's performance on a class of one-dimensional diatomic molecules, it is shown that this method not only has higher accuracy and faster computational speed compared to traditional numerical solvers but also generalizes well to different ionic configurations and higher resolution grids. 3. **Demonstrating physical feasibility**: The paper discusses whether the predicted electronic density conforms to physical reasonableness and validity, such as properties like density conservation and time-reversal symmetry. The physical significance of the model-predicted density is further validated by calculating observables such as the dipole moment. In summary, this work provides a potential method for real-time, on-the-fly modeling of molecules and materials under laser irradiation, especially under varying experimental parameters. Additionally, this method helps accelerate research on light-matter interactions in fields such as photocatalysis, biochemistry, and nanoscale devices.