DOI: 10.14278/rodare.3995
title: Dataset for Machine Learning Time Propagators for Time-Dependent Density Functional Theory Simulations

description: # Dataset for "Machine Learning Time Propagators for Time-Dependent Density Functional Theory Simulations"

 

This repository contains the dataset supporting the paper "Machine Learning Time Propagators for Time-Dependent Density Functional Theory Simulations" by Karan Shah and Attila Cangi. It comprises time-dependent density functional theory (TDDFT) simulations of one-dimensional diatomic molecules under laser excitation. The data is used to train and evaluate autoregressive Fourier Neural Operator (FNO) models that serve as machine learning time propagators for electron density evolution.

 

## Overview

 

This dataset comprises time-dependent density functional theory (TDDFT) simulations of one-dimensional diatomic molecules under laser excitation. The data is used to train and evaluate autoregressive Fourier Neural Operator (FNO) models that serve as machine learning time propagators for electron density evolution.

 

## Physical System

 

The simulations model diatomic molecules with:

 

- Soft-Coulomb ionic potential:

$$

v_{\text{ion}}(x) = -\frac{Z_{1}}{\sqrt{(x - d/2)^{2} + a^{2}}} \;-\; \frac{Z_{2}}{\sqrt{(x + d/2)^{2} + a^{2}}}

$$

- Sinusoidal laser excitation:

$v_{\text{las}}(t) = A \sin(\omega t)$ in dipole approximation

- Two-electron systems under adiabatic local density approximation (ALDA)

- Fixed boundary conditions at domain edges

 

## Dataset Specifications

 

- **Spatial domain**: $[-9.0, 9.0]$ atomic units with spacing $\Delta x = 0.05$ a.u. (361 grid points)

- **Temporal domain**: $[0, 5.0]$ femtoseconds with ML time step $\Delta t = 0.1$ fs (51 time steps)

- **Reference resolution**: $\Delta t = 0.01$ fs for high-accuracy ground truth

- **Total systems**: 2048 independent simulations

- **System parameters**:

- Nuclear charges $(Z_{1}, Z_{2})$: $1.0$–$4.0$ a.u.

- Internuclear distances $(d)$: $1.0$–$4.0$ a.u.

- Laser wavelengths: $400$–$750$ nm (optical range)

- Laser intensities: $10^{12}$–$10^{14}$ W/cm$^{2}$

- Softening parameter $(a)$: $1.0$ a.u.

 

## Data Format

 

Each `combined_data.npz` file contains:

 

- `densities`: Electron densities [systems, spatial_points, time_steps]

- `lasers_sliced`: Laser field values during simulation period

- `lasers_val`: Full laser field temporal profile

- `lasers_t`: Time grid for laser fields

- `x`: Spatial coordinate grid

- `t`: Temporal coordinate grid

 

## Baseline Directory

 

```

baseline/

├── combined_data.npz

├── combined_static_energy.npz

├── data_exclude.yaml

├── data_indices.npy

├── data_static_exclude.yaml

├── data_static_exclude_40_percentile.yaml

├── data_static_exclude_60_percentile.yaml

├── inp_gs.yaml

├── inp_td.yaml

├── param_set.yaml

├── params.csv

└── summary_statistics.md

```

 

- `combined_data.npz` — consolidated float32 arrays for 2,048 TDDFT trajectories, including spatial grid (`x`, 361 points), temporal grid (`t`, 51 steps across 0–5 fs ≈ 0–206.7 a.u.), density snapshots, and laser waveforms.

- `combined_static_energy.npz` — derived observables such as particle-number integrals, dipole moments, and Thomas–Fermi energies corresponding to each trajectory and time slice.

- `data_exclude.yaml` — explicit indices removed from the dataset because of boundary reflections or low temporal variation.

- `data_indices.npy` — NumPy array of retained example indices matching the rows in `params.csv`; use it to align parameter metadata with data tensors without reparsing the NPZ archive.

- `data_static_exclude*.yaml` — helper masks listing systems to omit when screening for near-static densities; percentile-specific files (`40`, `60`) filter systems with low temporal variation.

- `inp_gs.yaml`, `inp_td.yaml` — Octopus ground-state and real-time input templates used for the simulations (Crank–Nicolson propagator, 0.01 fs internal time step, custom diatomic potential). Parameters are overriden according to `params.csv`

- `param_set.yaml` — YAML description of the parameter sweep ranges (nuclear charges, internuclear distance, driving-field amplitudes, carrier frequencies).

- `params.csv` — resolved parameter combinations with units, suitable for quick inspection or tabular joins.

- `summary_statistics.md` — autogenerated report validating float32 conversion, spatial/temporal ranges, and observable metrics for every stored quantity.

 

## Dataset Variants

 

- `baseline/` — fine-grid reference generated at 0.01 fs Crank–Nicolson resolution and downsampled to 0.1 fs outputs; used for training the model and to get baseline results.

- `octopus_coarse/` — numerical TDDFT rollouts computed directly on the coarser 0.1 fs grid for solver-versus-model comparisons.

- `spatial_superresolution/` — trajectories evaluated on a doubled spatial resolution (Δx = 0.025 a.u., 721 grid points) to test Fourier Neural Operator generalization without retraining.

- `time_extension/` — long-horizon rollouts propagated to 10 fs for assessing error accumulation over time domains outside of training dataset.