arXiv AI By Chenxing Liang, Chengdong Wang, Yuchao Lin, Xiaofeng Qian, Shuiwang Ji

Equivariant Flow Matching for Electron Density Prediction

Read the original on arXiv AI →

The paper introduces OrbFlow, an SE(3)-equivariant generative model that predicts Gaussian-type orbital coefficients for electron density using flow matching. By training with a two-phase trajectory curriculum, OrbFlow improves accuracy on QM9 and MD benchmarks, reducing density errors significantly compared to previous models. The predicted densities also accelerate self-consistent field procedures, cutting iterations by up to 68% and accurately reproducing dipole and quadrupole moments without additional SCF calculations.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

arXiv AI
Aug 26

Learning the Kohn-Sham map with neural operators for quasi-linear scaling density functional theory

The paper presents a neural operator that learns the Kohn–Sham map, directly predicting electron density from the Kohn–Sham potential without orbital diagonalization. Using a domain‑invariant SE(3)‑equivariant Fourier neural operator trained on 8,504 molecules and solids, the model achieves quasi‑linear scaling self‑consistent field (SCF) convergence across diverse systems—including organic molecules, insulators, and metals—while reproducing Kohn–Sham DFT accuracy for densities, spectra, and structural observables. This enables large‑scale simulations, such as magnesium dislocation densities with 82,500 valence electrons, on a single GPU.

By Danish Khan, Maurice D. Hanisch, Nikolai Argatoff, Evan Xie, Sandeep Sharma, Anima Anandkumar
arXiv Machine Learning
Oct 12

Electron-Density-Informed Molecular Property Prediction Using Surrogate Functionals

The paper introduces a machine‑learning approach that incorporates the ground‑state electron density as input while enforcing it to be the minimum of a surrogate functional derived from orbital‑free density functional theory. This method outperforms state‑of‑the‑art models on six of seven extensive properties in the QM9 dataset and shows better generalization to larger molecules in the QM40 dataset. Models that use the density without enforcing the surrogate functional constraints perform worse on all extensive properties of both datasets.

By Manuel V. Klockow, Simon Wagner, Roman Remme, Dominic Plein, Fred A. Hamprecht
arXiv Machine Learning
Sep 29

Scaling Density Functional Theory with Gaussian Splatting

The paper introduces Gaussian Splatting for Density Functional Theory (GS‑DFT), a method that represents molecular orbitals as a cloud of Gaussians optimized via gradient descent. GS‑DFT replaces fixed atom‑centered basis sets with an adaptive, differentiable orthogonalization and efficient two‑electron integral evaluation, achieving accuracy comparable to large conventional bases with far fewer parameters. The solver scales quadratically with cloud size, enabling simulations of up to 2,742 atoms on a single four‑GPU node at triple‑zeta precision.

By Andr\'es Guzm\'an-Cordero, Cindy Zhang, Majdi Hassan, Marta Skreta, Kirill Neklyudov, Matija Medvidovi\'c