arXiv Machine Learning

Equivariant Cellular Sheaves for Molecular Electronic Structure: Bridging Sheaf Cohomology and E(3)-Equivariant Hamiltonian Learning

arXiv AI
Aug 20

Coupled-cluster molecular properties across the main group that extrapolate beyond training size

The paper introduces MEHnet-MG, an equivariant neural network that predicts a one‑electron Hamiltonian from a single inexpensive B3LYP/def2‑SVP calculation and uses it to compute a wide range of molecular properties—energy, optical gap, dipole, quadrupole, polarizability, Mulliken charges, and Mayer bond orders—at coupled‑cluster accuracy for nine main‑group elements, including phosphorus, sulfur, and chlorine. Trained on a new CCSD(T) dataset, the model reduces property errors by factors ranging from 3.8 to 230 compared to various DFT methods while adding only ~25 ms per molecule. Importantly, by deriving properties from a predicted Hamiltonian rather than pooling atomic features, the architecture naturally incorporates correct size‑scaling, enabling accurate extrapolation to large π‑conjugated systems (up to 58 atoms) where traditional pooling‑based models fail.

By Wenhao He, Xu Chen, Noah Song, Haowei Xu, Tim S. Hindges, Bohan Li, Zihan Lin, Yu Yao, Avetik R. Harutyunyan, Fang Liu, Yao Wang, Hao Tang, Ju Li
arXiv Machine Learning
Jun 2

Sheaf Neural Networks on SPD Manifolds: Second-Order Geometric Representation Learning

arXiv:2604. 20308v2 Announce Type: replace Abstract: Graph neural networks face two fundamental challenges rooted in the linear structure of Euclidean vector spaces: (1) Current architectures represent geometry through vectors (directions, gradients), yet many tasks require matrix-valued representations that capture relationships between directions-such as how atomic orientations covary in a molecule.

By Yuhan Peng, Junwen Dong, Yuzhi Zeng, Hao Li, Ce Ju, Huitao Feng, Diaaeldin Taha, Anna Wienhard, Kelin Xia
arXiv Machine Learning
Jul 8

Multimodal Molecular Representation Learning with Graph Neural Networks, Deep & Cross Networks, and SMILES Embeddings

arXiv:2607. 05736v1 Announce Type: new Abstract: Molecular property prediction often relies on isolated data modalities, where continuous 3D graph neural networks (GNNs) struggle to efficiently capture long-range topological dependencies and exact macroscopic heuristics.

By Qiwei Han, Chi Zhou, Ruobing Wang, Zheng Ma
arXiv AI
Aug 26

Learning the Kohn-Sham map with neural operators for quasi-linear scaling density functional theory

The paper presents a neural operator that learns the Kohn–Sham map, directly predicting electron density from the Kohn–Sham potential without orbital diagonalization. Using a domain‑invariant SE(3)‑equivariant Fourier neural operator trained on 8,504 molecules and solids, the model achieves quasi‑linear scaling self‑consistent field (SCF) convergence across diverse systems—including organic molecules, insulators, and metals—while reproducing Kohn–Sham DFT accuracy for densities, spectra, and structural observables. This enables large‑scale simulations, such as magnesium dislocation densities with 82,500 valence electrons, on a single GPU.

By Danish Khan, Maurice D. Hanisch, Nikolai Argatoff, Evan Xie, Sandeep Sharma, Anima Anandkumar
arXiv Machine Learning
Jul 1

ElemeNet: Multiscale Molecular Machine Learning with Uncertainty Quantification Across the Periodic Table

arXiv:2606. 30961v1 Announce Type: cross Abstract: Advances in deep learning architectures and representations have enabled ML-driven chemical property prediction, but state-of-the-art (SOTA) models have remained largely confined to independent codebases and lack support for diverse chemical species.

By Jacob W. Toney, Samir Darouich, Yiran Wang, Aaron G. Garrison, Johannes K\"astner, Heather J. Kulik
arXiv Machine Learning
Jun 30

MALOQ: Massively Accelerated Learning of Operators for Quantum Transport

arXiv:2606. 28911v1 Announce Type: new Abstract: Machine-learned (ML) operator models can be trained to predict density functional theory (DFT) Hamiltonian/density matrices at significantly reduced computational cost, thus extending electronic-structure calculations to previously unfeasible scales.

By Manasa Kaniselvan, Alexander Maeder, Denghui Lu, Alexandros Nikolaos Ziogas, Mathieu Luisier
arXiv Machine Learning
Jul 22

GEqTrain: A Configuration-Driven Framework for Retargeting Equivariant Graph Neural Networks Across 3D Scientific Tasks

arXiv:2607. 19083v1 Announce Type: new Abstract: Equivariant graph neural networks provide a powerful modeling language for three-dimensional scientific data, but their reuse is often limited by implementations tied to specific tasks, outputs, and training regimes.

By Daniele Angioletti, Marco Nobile, Vittorio Limongelli
arXiv AI
Aug 28

Mechanistic Reaction Prediction via Discrete Flow Matching on Graph-Structured Electron Occupation

MAELLE is a mechanistic reaction prediction framework that models chemical reactions as discrete flow matching over graph-structured electron occupation vectors. It formulates the reactant-to-product mapping as a Continuous-time Markov Chain on electron sites and uses Optimal Transport to generate mechanistically interpretable edit trajectories without elementary step annotations. The method achieves competitive accuracy on the USPTO-480K benchmark, remains robust in out-of-distribution scenarios, and can recover mechanistic pathways that align with known chemistry and predict side products.

By Nguyen Xuan-Vu, Octavian Susanu, Daniel Armstrong, Philippe Schwaller