Operator Lanczos Approach enabling Neural Quantum States as Real-Frequency Impurity Solvers
Jonas B. Rigo, Markus Schmitt
Abstract
Abstract To understand the intricate exchange between electrons of different bands in strongly correlated materials, it is essential to treat multi-orbital models accurately. For this purpose, dynamical mean-field theory (DMFT) provides an established framework, whose scope crucially hinges on the availability of efficient quantum impurity solvers. Here we present a real-frequency impurity solver based on neural quantum states (NQS) combined with an operator-Lanczos construction. NQS are an asymptotically unbiased variational ground-state ansatz that employs neural networks to capture long-range correlations on complicated graph structures. We leverage this ability to solve multi-orbital impurity problems using a systematically improvable Segmented Commutator Operator-Lanczos (SCOL) construction. Our benchmarks on both the single-orbital Anderson model and the multi-orbital Hubbard–Kanamori impurity Hamiltonian reveal excellent ground-state precision and the capacity to accurately resolve zero temperature spectral functions and self-energies. These results open avenues for extending DMFT to more challenging problems. Repository contents This repository contains the figures, data, and plotting script used to recreate Fig. 2 and Fig. 3 of “Operator Lanczos Approach enabling Neural Quantum States as Real Frequency Impurity Solvers”. The notebook reproduces spectral functions and self-energies for the Single Anderson Impurity Model (SIAM) and a Hubbard–Kanamori system with two and three orbitals (MO2, MO3). The files in this repsitory are: `plot.ipynb` – matplotlib script that reads the HDF5 files and generate the publication figures. `data/SIAM_plot.h5` – SIAM spectral/self-energy data grouped into three panels (`panel_left`, `panel_mid`, `panel_right`) with metadata for interaction strength `U`, segment depth `l_s`, and method `kind` (`NQS` or `NRG` -> for reference). `data/MO_plot.h5` – MO2 and MO3 results. `MO2` contains frequency `omega`, spectral function `A` and self-energy arrays (`omega_Sigma`, `Sigma_im_nqs` and NRG references). `MO3` holds NRG spectra plus frequncies `omega` and spectral function `A_nqs` for `l_s = 2,3,4`. `siam_square.pdf` – rendered SIAM figure (Fig. 2 in the paper). `mo.pdf` – rendered MO figure (Fig. 3 in the paper). Requirements Python 3 with `numpy`, `matplotlib`, and `h5py`. Jupyter (if you want to run the notebook interactively). A LaTeX installation (e.g., TeX Live) so matplotlib can render TeX labels (`text.usetex=True` is enabled). Install the Python packages locally (examples): pip install numpy matplotlib h5py jupyter Notes - The notebook uses Computer Modern fonts via LaTeX; if TeX is missing you can disable `text.usetex` inside the first code cell. - Data are already curated for plotting, no additional preprocessing is needed beyond loading the HDF5 datasets described above. - Tested versions (newer should also work): `numpy==1.26.4`, `matplotlib==3.10.0`, `h5py==3.13.0`, `jupyter==1.1.1`.