Global Plane Waves from Local Gaussians: Periodic Charge Densities in a Blink
Jonas Elsborg, Felix Aertebjerg, Luca Anthony Thiede, Alan Aspuru-Guzik, Tejs Vegge, Arghya Bhowmik
Abstract
We introduce ELECTRAFI, a fast, end-to-end differentiable model for predicting periodic charge densities in crystalline materials. ELECTRAFI constructs anisotropic Gaussians in real space and exploits their closed-form Fourier transforms to analytically evaluate plane-wave coefficients via the Poisson summation formula. This formulation delegates non-local and periodic behavior to analytic transforms, enabling reconstruction of the full periodic charge density with a single inverse FFT. By avoiding explicit real-space grid probing, periodic image summation, and spherical harmonic expansions, ELECTRAFI matches or exceeds state-of-the-art accuracy across periodic benchmarks while being up to faster than the strongest competing method, reconstructing crystal charge densities in a fraction of a second. When used to initialize DFT calculations, ELECTRAFI reduces total DFT compute cost by up to 20 %, whereas slower charge density models negate savings due to high inference times. Our results show that accuracy and inference cost jointly determine end-to-end DFT speedups, and motivate our focus on efficiency.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 1ee66c6b-825e-4ff6-99d6-4f9a5a3b231dBuilds on14
- Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
- Directional Message Passing for Molecular GraphsJohannes Klicpera, Janek Groß, Stephan GünnemannICLR 2020 · 1,079 citations
- GemNet: Universal Directional Graph Neural Networks for MoleculesJohannes Gasteiger, Florian Becker, Stephan GünnemannNeurIPS 2021 · 665 citations
- EquiformerV2: Improved Equivariant Transformer for Scaling to Higher-Degree RepresentationsYi-Lun Liao, Brandon M. Wood, Abhishek Das, Tess E. SmidtICLR 2024 · 311 citations
- Reducing SO(3) Convolutions to SO(2) for Efficient Equivariant GNNsSaro Passaro, C. Lawrence ZitnickICML 2023 · 157 citations
Related papers
- ELECTRA: A Cartesian Network for 3D Charge Density Prediction with Floating OrbitalsJonas Elsborg, Luca A. Thiede, Alán Aspuru-Guzik, Tejs Vegge et al.NeurIPS 2025 · 15 citations
- A Recipe for Charge Density PredictionXiang Fu, Andrew S. Rosen, Kyle Bystrom, Rui Wang et al.NeurIPS 2024 · 26 citations
- Equivariant Atomic and Lattice Modeling Using Geometric Deep Learning for Crystal Structure OptimizationZiduo Yang, Yiming Zhao, Xian Wang, Wei Zhuo et al.AAAI 2026 · 1 citation
- SE(3)-equivariant prediction of molecular wavefunctions and electronic densitiesOliver T. Unke, Mihail Bogojeski, Michael Gastegger, Mario Geiger et al.NeurIPS 2021 · 135 citations
- Towards a Transferable Acceleration Method for Density Functional TheoryZhe Liu, Yuyan Ni, Zhichen Pu, Qiming Sun et al.ICLR 2026 · 3 citations
