Enhancing molecular dynamics with equivariant machine-learned densities

arXiv stat.ML / 4/28/2026

💬 OpinionDeveloper Stack & InfrastructureModels & Research

Key Points

  • The paper introduces DenSNet, a “density-first” machine-learning approach that learns the Hohenberg–Kohn mapping from nuclear configurations to ground-state electron densities, enabling electronic observables beyond energies and forces.
  • DenSNet uses an SE(3)-equivariant neural network to predict coefficients of an atom-centered Gaussian density basis, and a Δ-learning scheme that leverages superposed atomic densities as a prior to speed up training.
  • A second SE(3)-equivariant network converts the predicted electron density into total energy, creating a unified framework for molecular dynamics and electronic-structure prediction.
  • Validation on ethanol, ethanethiol, and resorcinol shows that infrared spectra derived from ML-driven trajectories match experimental gas-phase measurements closely.
  • For scalability, the method is trained on polythiophene oligomers (1–6 monomers) and extrapolated to longer chains (up to 12 monomers), producing stable long-time trajectories whose infrared spectra agree with reference DFT calculations.

Abstract

Machine-learning interatomic potentials (MLIPs) have enabled molecular dynamics at near ab initio accuracy, yet remain limited to energies and forces by construction, leaving electronic observables such as dipole moments and polarizabilities inaccessible. We introduce DenSNet, a density-first approach to machine-learned electronic structure that learns the Hohenberg--Kohn map from nuclear configurations to the ground-state electron density. Our approach employs an SE(3)-equivariant neural network to predict density coefficients of a flexible atom-centered Gaussian basis, combined with a \Delta-learning strategy that uses superposed atomic densities as a prior to accelerate training. A second equivariant network then maps the predicted density to the total energy, providing a unified framework for molecular dynamics and electronic structure. We validate DenSNet on ethanol, ethanethiol, and resorcinol, where infrared spectra from machine-learned trajectories show excellent agreement with experimental gas-phase measurements. To test scalability, we train on polythiophene oligomers with 1--6 monomers and extrapolate to chains of up to 12 monomers, generating stable long-time trajectories whose infrared spectra agree with reference density functional theory calculations. Here, 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.