arXiv AI

Equivariant Flow Matching for Electron Density Prediction

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.

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
arXiv Machine Learning
Oct 8

Physics-Aligned Electronic Ground-State Learning Improves Generalization

The paper introduces Physics‑Aligned Electronic Ground‑State Models (GSMs) that sit between machine‑learned interatomic potentials and Kohn‑Sham density functional theory in terms of computational cost and accuracy. By aligning GSM learning objectives with KS‑DFT equations and enforcing physical constraints, the authors develop new training techniques—OrthoNormal‑Loss (ON‑Loss), Grassmann Restricted Occupied‑Orbital Training (GROOT), and Residual Optimal‑gauge Conditioning‑aware KS‑Eq. Training (ROCKET)—which dramatically reduce energy and force errors in size‑extrapolation tests from QM9 to QM40. The models also achieve near‑chemical‑accuracy predictions on reactive chemistry benchmarks and demonstrate efficient label‑free self‑consistency fine‑tuning. whyItMatters":"The work advances the cost‑accuracy frontier for electronic structure predictions, enabling more reliable out‑of‑distribution generalization in drug and material discovery."

By Eike S. Eberhard, Xaver Kainz, Viktor Kotsev, Abdulrahman Aldossary, Stephan G\"unnemann
arXiv AI
Jun 15

A Fixed-Point Neural Operator for Size- and Functional-Transferable Hamiltonian Prediction

arXiv:2606. 14498v1 Announce Type: cross Abstract: Predicting the Kohn-Sham Hamiltonian with machine learning can accelerate density functional theory while retaining access to molecular orbitals, energy levels, and electronic-structure observables that energy-only surrogates cannot resolve.

By Yunhong Lou, Xihang Yue, Xinran Wei, Tianqi Deng, Linchao Zhu
arXiv AI
Aug 28

Mechanistic Reaction Prediction via Discrete Flow Matching on Graph-Structured Electron Occupation

MAELLE is a mechanistic reaction prediction framework that models chemical reactions as discrete flow matching over graph-structured electron occupation vectors. It formulates the reactant-to-product mapping as a Continuous-time Markov Chain on electron sites and uses Optimal Transport to generate mechanistically interpretable edit trajectories without elementary step annotations. The method achieves competitive accuracy on the USPTO-480K benchmark, remains robust in out-of-distribution scenarios, and can recover mechanistic pathways that align with known chemistry and predict side products.

By Nguyen Xuan-Vu, Octavian Susanu, Daniel Armstrong, Philippe Schwaller