# Equivariant Flow Matching for Electron Density Prediction

arXiv: https://arxiv.org/abs/2610.02651 · HTML: https://mlsanity.com/paper/2610.02651/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Placed in Flow matching, time series and science, one of 14 directions in Diffusion models.
- Builds on 30 of its 59 references that are in the map.
- First posted on arXiv 2 Oct 2026.

## Placed in

- Diffusion models / [Flow matching, time series and science](https://mlsanity.com/diffusion/flow-matching-time-series-and-science/)

## Builds on (most cited first, 30 of 30)

- [Flow Matching for Generative Modeling](https://mlsanity.com/paper/2210.02747/) (2022)
- [Consistency Models](https://mlsanity.com/paper/2303.01469/) (2023)
- [Improving and generalizing flow-based generative models with minibatch optimal transport](https://mlsanity.com/paper/2302.00482/) (2023)
- [Scheduled Sampling for Sequence Prediction with Recurrent Neural Networks](https://mlsanity.com/paper/1506.03099/) (2015)
- [Sequence Level Training with Recurrent Neural Networks](https://mlsanity.com/paper/1511.06732/) (2015)
- [One Step Diffusion via Shortcut Models](https://mlsanity.com/paper/2410.12557/) (2024)
- [e3nn: Euclidean Neural Networks](https://mlsanity.com/paper/2207.09453/) (2022)
- [Artificial Intelligence for Science in Quantum, Atomistic, and Continuum Systems](https://mlsanity.com/paper/2307.08423/) (2023)
- [Reducing SO(3) Convolutions to SO(2) for Efficient Equivariant GNNs](https://mlsanity.com/paper/2302.03655/) (2023)
- [AlphaFold Meets Flow Matching for Generating Protein Ensembles](https://mlsanity.com/paper/2402.04845/) (2024)
- [FlowMM: Generating Materials with Riemannian Flow Matching](https://mlsanity.com/paper/2406.04713/) (2024)
- [By-passing the Kohn-Sham equations with machine learning](https://mlsanity.com/paper/1609.02815/) (2016)
- [Efficient and Equivariant Graph Networks for Predicting Quantum Hamiltonian](https://mlsanity.com/paper/2306.04922/) (2023)
- [Informing Geometric Deep Learning with Electronic Interactions to Accelerate Quantum Chemistry](https://mlsanity.com/paper/2105.14655/) (2021)
- [Harmonic Self-Conditioned Flow Matching for Multi-Ligand Docking and Binding Site Design](https://mlsanity.com/paper/2310.05764/) (2023)
- [Equivariant graph neural networks for fast electron density estimation of molecules, liquids, and solids](https://mlsanity.com/paper/2112.00652/) (2021)
- [MOFFlow: Flow Matching for Structure Prediction of Metal-Organic Frameworks](https://mlsanity.com/paper/2410.17270/) (2024)
- [Enhancing the Scalability and Applicability of Kohn-Sham Hamiltonians for Molecular Systems](https://mlsanity.com/paper/2502.19227/) (2025)
- [Higher-Order Equivariant Neural Networks for Charge Density Prediction in Materials](https://mlsanity.com/paper/2312.05388/) (2023)
- [Equivariant Neural Operator Learning with Graphon Convolution](https://mlsanity.com/paper/2311.10908/) (2023)
- [Efficient and Scalable Density Functional Theory Hamiltonian Prediction through Adaptive Sparsity](https://mlsanity.com/paper/2502.01171/) (2025)
- [High-order Equivariant Flow Matching for Density Functional Theory Hamiltonian Prediction](https://mlsanity.com/paper/2505.18817/) (2025)
- [EquiformerV3: Scaling Efficient, Expressive, and General SE(3)-Equivariant Graph Attention Transformers](https://mlsanity.com/paper/2604.09130/) (2026)
- [ELECTRA: A Cartesian Network for 3D Charge Density Prediction with Floating Orbitals](https://mlsanity.com/paper/2503.08305/) (2025)
- [Gaussian Plane-Wave Neural Operator for Electron Density Estimation](https://mlsanity.com/paper/2402.04278/) (2024)
- [A Recipe for Charge Density Prediction](https://mlsanity.com/paper/2405.19276/) (2024)
- [Transferable E(3) equivariant parameterization for Hamiltonian of molecules and solids](https://mlsanity.com/paper/2210.16190/) (2022)
- [Learning local equivariant representations for quantum operators](https://mlsanity.com/paper/2407.06053/) (2024)
- [Efficient Prediction of SO(3)-Equivariant Hamiltonian Matrices via SO(2) Local Frames](https://mlsanity.com/paper/2506.09398/) (2025)
- [Towards A Transferable Acceleration Method for Density Functional Theory](https://mlsanity.com/paper/2509.25724/) (2025)

## Abstract (arXiv)

Machine learning surrogates for density functional theory (DFT) have been increasingly used to reduce the cost of first-principles calculations. In this arena, predicting real-space electron densities offers a scalable and transferable initialization for self-consistent field (SCF) procedures. However, current methods face a clear dilemma. That is, grid-based architectures incur a high computational cost, while basis-set methods fail to capture the structural correlations inherent in the coefficient space. Here, we develop OrbFlow, an SE(3)-equivariant generative model that predicts Gaussian-type orbital (GTO) coefficients via flow matching. OrbFlow retains the efficiency of a compact atom-centered basis while replacing pointwise regression with a learned probability path over the full coefficient space. It is trained through a two-phase trajectory curriculum that mitigates discretization drift during numerical integration. OrbFlow achieves state-of-the-art accuracy on QM9, reducing density error by 13.6% relative to the previous best model, and reduces error by 51% to 63% on every molecule of the MD benchmark relative to the strongest prior method sharing its basis. The predicted density also cuts SCF iterations by up to 68% with zero-shot transfer to unseen exchange-correlation functionals and recovers dipole and quadrupole moments to within a few percent of DFT references without any SCF calculation.