V2Rho-FNO: Fourier Neural Operator for Electronic Density Prediction
- URL: http://arxiv.org/abs/2603.15669v1
- Date: Fri, 13 Mar 2026 06:53:13 GMT
- Title: V2Rho-FNO: Fourier Neural Operator for Electronic Density Prediction
- Abstract summary: We present a universal framework based on Fourier Neural Operators (FNOs) that directly learns the mapping from external potentials to electron density distributions.<n>Our work establishes neural operator learning as a promising route for fast, accurate, and transferable electronic structure prediction.
- Score: 10.028886116912622
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Density functional theory (DFT) is a cornerstone of computational chemistry and materials science, but its computational cost limits its use in large-scale and high-throughput applications. While machine learning has accelerated energy prediction for specific molecular classes, transferable prediction of electron density across diverse chemical spaces remains challenging. Here, we present a universal framework based on Fourier Neural Operators (FNOs) that directly learns the mapping from external potentials to electron density distributions. Unlike conventional approaches that rely on explicit atomic orbitals, basis sets, or handcrafted descriptors, the proposed method captures global electronic interactions and long-range correlations through operator learning in the spatial-frequency domain. Trained on datasets spanning multiple elements and molecular geometries, the model achieves zero-shot generalization to entirely unseen molecular systems and accurately predicts their electron densities without retraining. This transferability arises from the intrinsic ability of FNOs to represent global structure in continuous fields. Our work establishes neural operator learning as a promising route for fast, accurate, and transferable electronic structure prediction, with potential applications in high-throughput screening and chemical space exploration.
Related papers
- Learning the Kohn-Sham map with neural operators for quasi-linear scaling density functional theory [57.90994363143731]
Kohn--Sham density functional theory (DFT) underpins electronic-structure simulations.<n>Repeated orbital diagonalizations lead to cubic scaling, restricting quantum calculations to modest scales only.<n>We develop a learning target for orbital-free DFT that maps a Kohn--Sham potential directly to the corresponding density and noninteracting kinetic energy.
arXiv Detail & Related papers (2026-08-24T23:05:42Z) - ED-DiT: Physics-Guided Diffusion Pretraining for Transferable Molecular Representations from Electron Density [14.29876606291099]
We propose ED-DiT, a physics-guided Diffusion Transformer for self-supervised pretraining on electron-density point clouds.<n> ED-DiT learns reusable representations by reconstructing corrupted and partially masked log-density fields across diffusion noise levels.<n>Experiments on six EDBench tasks show that ED-DiT consistently outperforms the same architecture trained from scratch, especially under limited supervision.
arXiv Detail & Related papers (2026-08-04T07:33:35Z) - Enhancing molecular dynamics with equivariant machine-learned densities [10.247839738079698]
We introduce DenSNet, a density-first approach to machine-learned electronic structure.<n>We validate DenSNet on ethanol, ethanethiol, and resorcinol, where infrared spectra from machine-learned trajectories show excellent agreement with experimental gas-phase measurements.<n>We show that reinstating the electron density as the central learned quantity opens a practical route to transferable prediction of spectroscopic and electronic observables in large-scale molecular simulations.
arXiv Detail & Related papers (2026-04-27T14:49:53Z) - Coupled Cluster con MōLe: Molecular Orbital Learning for Neural Wavefunctions [34.53033480184221]
Density functional theory (DFT) is the most widely used method for calculating molecular properties.<n> Coupled-cluster (CC) theory is the most successful method for achieving accuracy beyond DFT.<n>We present the Molecular Orbital Learning (MLe) architecture, an equivariant machine learning model that directly predicts CC's core mathematical objects.
arXiv Detail & Related papers (2026-02-23T18:54:46Z) - Acceleration of Atomistic NEGF: Algorithms, Parallelization, and Machine Learning [61.12861060232382]
The Non-equilibrium Green's function (NEGF) formalism is a powerful method to simulate the quantum transport properties of nanoscale devices.<n>This paper summarizes key (algorithmic) achievements that have allowed us to bring DFT+NEGF simulations closer to the dimensions and functionality of realistic systems.
arXiv Detail & Related papers (2026-02-03T12:01:39Z) - Foundation Models for Discovery and Exploration in Chemical Space [57.97784111110166]
MIST is a family of molecular foundation models trained on large unlabeled datasets.<n>We demonstrate the ability of these models to solve real-world problems across chemical space.
arXiv Detail & Related papers (2025-10-20T17:56:01Z) - OrbitAll: A Unified Quantum Mechanical Representation Deep Learning Framework for All Molecular Systems [64.69217059173184]
OrbitAll is a geometry- and physics-informed deep learning framework.<n>It can represent and process any molecular system with arbitrary charges, spins, and environmental effects.<n>It achieves chemical accuracy using 10 times fewer training data than competing AI models, with a speedup of approximately $103$ - $104$ compared to density functional theory.
arXiv Detail & Related papers (2025-07-05T01:21:56Z) - Constructing accurate machine-learned potentials and performing highly efficient atomistic simulations to predict structural and thermal properties [6.875235178607604]
We introduce a neuroevolution potential (NEP) trained on a dataset generated from ab initio molecular dynamics (AIMD) simulations.
We calculate the phonon density of states (DOS) and radial distribution function (RDF) using both machine learning potentials.
While the MTP potential offers slightly higher accuracy, the NEP achieves a remarkable 41-fold increase in computational speed.
arXiv Detail & Related papers (2024-11-16T23:16:59Z) - Predicting ionic conductivity in solids from the machine-learned potential energy landscape [68.25662704255433]
We propose an approach for the quick and reliable screening of ionic conductors through the analysis of a universal interatomic potential.<n>Eight out of the ten highest-ranked materials are confirmed to be superionic at room temperature in first-principles calculations.<n>Our method achieves a speed-up factor of approximately 50 compared to molecular dynamics driven by a machine-learning potential, and is at least 3,000 times faster compared to first-principles molecular dynamics.
arXiv Detail & Related papers (2024-11-11T09:01:36Z) - Transfer learning for chemically accurate interatomic neural network
potentials [0.0]
We show that pre-training the network parameters on data obtained from density functional calculations improves the sample efficiency of models trained on more accurate ab-initio data.
We provide GM-NN potentials pre-trained and fine-tuned on the ANI-1x and ANI-1ccx data sets, which can easily be fine-tuned on and applied to organic molecules.
arXiv Detail & Related papers (2022-12-07T19:21:01Z) - Gaussian Moments as Physically Inspired Molecular Descriptors for
Accurate and Scalable Machine Learning Potentials [0.0]
We propose a machine learning method for constructing high-dimensional potential energy surfaces based on feed-forward neural networks.
The accuracy of the developed approach in representing both chemical and configurational spaces is comparable to the one of several established machine learning models.
arXiv Detail & Related papers (2021-09-15T16:46:46Z) - BIGDML: Towards Exact Machine Learning Force Fields for Materials [55.944221055171276]
Machine-learning force fields (MLFF) should be accurate, computationally and data efficient, and applicable to molecules, materials, and interfaces thereof.
Here, we introduce the Bravais-Inspired Gradient-Domain Machine Learning approach and demonstrate its ability to construct reliable force fields using a training set with just 10-200 atoms.
arXiv Detail & Related papers (2021-06-08T10:14:57Z) - Multi-task learning for electronic structure to predict and explore
molecular potential energy surfaces [39.228041052681526]
We refine the OrbNet model to accurately predict energy, forces, and other response properties for molecules.
The model is end-to-end differentiable due to the derivation of analytic gradients for all electronic structure terms.
It is shown to be transferable across chemical space due to the use of domain-specific features.
arXiv Detail & Related papers (2020-11-05T06:48:46Z) - Graph Neural Network for Hamiltonian-Based Material Property Prediction [56.94118357003096]
We present and compare several different graph convolution networks that are able to predict the band gap for inorganic materials.
The models are developed to incorporate two different features: the information of each orbital itself and the interaction between each other.
The results show that our model can get a promising prediction accuracy with cross-validation.
arXiv Detail & Related papers (2020-05-27T13:32:10Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.