English

Towards Accelerated SCF Workflows with Equivariant Density-Matrix Learning and Analytic Refinement

Chemical Physics 2026-05-01 v1 Artificial Intelligence Machine Learning Computational Physics Quantum Physics

Abstract

We present \textsc{dm-PhiSNet}, a physically constrained \textsc{PhiSNet}-based equivariant model that predicts one-electron reduced density matrices (1-RDMs) directly from molecular geometries in an atomic-orbital (AO) basis for accelerated self-consistent field (SCF) workflows. Training follows a two-stage schedule with progressively introduced physically motivated objectives, and the resulting predictions are refined by a lightweight analytic block. This block enforces electron-number conservation, drives the 1-RDM toward generalized idempotency in the AO metric, and regularizes the occupation spectrum of the L\"owdin-orthogonalized density. Across six closed-shell systems -- H2_2O, CH4_4, NH3_3, HF, ethanol, and NO3_3^- -- the refined 1-RDMs provide SCF initial guesses that substantially reduce iteration steps by 49--81\% relative to standard initializations. Beyond SCF acceleration, the learned 1-RDMs yield accurate one-shot total energies and Hellmann--Feynman atomic forces without force supervision, indicating that the model captures chemically meaningful electronic structure. These results demonstrate that combining equivariant learning with analytic constraint enforcement provides a simple, general route to solver-ready density-matrix initializations and accelerated SCF workflows.

Keywords

Cite

@article{arxiv.2604.27256,
  title  = {Towards Accelerated SCF Workflows with Equivariant Density-Matrix Learning and Analytic Refinement},
  author = {Zuriel Y. Yescas-Ramos and Andrés Álvarez-García and Huziel E. Sauceda},
  journal= {arXiv preprint arXiv:2604.27256},
  year   = {2026}
}

Comments

7 pages, 3 figures

R2 v1 2026-07-01T12:42:31.101Z