Efficient and Equivariant Graph Networks for Predicting Quantum Hamiltonian
Not on the map. Why?
Abstract (arXiv)
We consider the prediction of the Hamiltonian matrix, which finds use in quantum chemistry and condensed matter physics. Efficiency and equivariance are two important, but conflicting factors. In this work, we propose a SE(3)-equivariant network, named QHNet, that achieves efficiency and equivariance. Our key advance lies at the innovative design of QHNet architecture, which not only obeys the underlying symmetries, but also enables the reduction of number of tensor products by 92%. In addition, QHNet prevents the exponential growth of channel dimension when more atom types are involved. We perform experiments on MD17 datasets, including four molecular systems. Experimental results show that our QHNet can achieve comparable performance to the state of the art methods at a significantly faster speed. Besides, our QHNet consumes 50% less memory due to its streamlined architecture. Our code is publicly available as part of the AIRS library (urlhttps://github.com/divelab/AIRS).
Key facts. Cited by 38 papers in the map (data through 7 Oct 2026). Builds on 33 of its 51 references that are in the map. First posted on arXiv 8 Jun 2023.
Builds on
33 of 51 references.
References · 33
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2017Neural Message Passing for Quantum Chemistrycited by 3,460
- 2019Fast Graph Representation Learning with PyTorch Geometriccited by 2,414
- 2021E(n) Equivariant Graph Neural Networkscited by 951
- 2018Tensor field networks: Rotation- and translation-equivariant neural networks for 3D point cloudscited by 735
- 2021E(3)-Equivariant Graph Neural Networks for Data-Efficient and Accurate Interatomic Potentialscited by 701
- 2020SE(3)-Transformers: 3D Roto-Translation Equivariant Attention Networkscited by 602
- 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 569
- 2020Directional Message Passing for Molecular Graphscited by 556
- 2021Equivariant message passing for the prediction of tensorial properties and molecular spectracited by 438
- 2021GemNet: Universal Directional Graph Neural Networks for Moleculescited by 342
- 2022DiffDock: Diffusion Steps, Twists, and Turns for Molecular Dockingcited by 307
21 more
- 2019Cormorant: Covariant Molecular Neural Networkscited by 262
- 2020Fast and Uncertainty-Aware Directional Message Passing for Non-Equilibrium Moleculescited by 254
- 2021Geometric and Physical Quantities Improve E(3) Equivariant Message Passingcited by 225
- 2021Spherical Message Passing for 3D Graph Networkscited by 166
- 2021Learning Gradient Fields for Molecular Conformation Generationcited by 162
- 2022EquiBind: Geometric Deep Learning for Drug Binding Structure Predictioncited by 143
- 2021GeoMol: Torsional Geometric Generation of Molecular 3D Conformer Ensemblescited by 106
- 2022Generating 3D Molecules for Target Protein Bindingcited by 104
- 2019Molecular geometry prediction using a deep generative graph neural networkcited by 91
- 2021Learning Neural Generative Dynamics for Molecular Conformation Generationcited by 87
- 2022Periodic Graph Transformers for Crystal Material Property Predictioncited by 87
- 2021Independent SE(3)-Equivariant Models for End-to-End Rigid Protein Dockingcited by 81
- 2019Unifying machine learning and quantum chemistry -- a deep neural network for molecular wavefunctionscited by 73
- 2019A Generative Model for Molecular Distance Geometrycited by 73
- 2022ComENet: Towards Complete and Efficient Message Passing for 3D Molecular Graphscited by 69
- 2021SE(3)-equivariant prediction of molecular wavefunctions and electronic densitiescited by 67
- 2020OrbNet: Deep Learning for Quantum Chemistry Using Symmetry-Adapted Atomic-Orbital Featurescited by 63
- 2021An End-to-End Framework for Molecular Conformation Generation via Bilevel Programmingcited by 59
- 2020Advanced Graph and Sequence Neural Networks for Molecular Property Prediction and Drug Discoverycited by 46
- 2022E3Bind: An End-to-End Equivariant Network for Protein-Ligand Dockingcited by 35
- 2021Fast Quantum Property Prediction via Deeper 2D and 3D Graph Networkscited by 10
Built on by
Citing papers per year: 2023: 2; 2024: 12; 2025: 12; 2026: 12.
38 papers.
Sort:
By area:
- 2024Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Productscited by 35
- 2023QH9: A Quantum Hamiltonian Prediction Benchmark for QM9 Moleculescited by 26
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiencycited by 18
- 2025Enhancing the Scalability and Applicability of Kohn-Sham Hamiltonians for Molecular Systemscited by 13
- 2024Self-Consistency Training for Density-Functional-Theory Hamiltonian Predictioncited by 12
- 2025HIP: Hessian Interatomic Potentials without derivativescited by 11
- 2025Autonomous Agents for Scientific Discovery: Orchestrating Scientists, Language, Code, and Physicscited by 11
- 2025Efficient and Scalable Density Functional Theory Hamiltonian Prediction through Adaptive Sparsitycited by 10
- 2025High-order Equivariant Flow Matching for Density Functional Theory Hamiltonian Predictioncited by 10
- 2024A Space Group Symmetry Informed Network for O(3) Equivariant Crystal Tensor Predictioncited by 7
- 2024Learning local equivariant representations for quantum operatorscited by 7
- 2025Learning from the electronic structure of molecules across the periodic tablecited by 7
26 more
- 2024Infusing Self-Consistency into Density Functional Theory Hamiltonian Prediction via Deep Equilibrium Modelscited by 6
- 2024Equivariant Graph Network Approximations of High-Degree Polynomials for Force Field Predictioncited by 6
- 2025Efficient Prediction of SO(3)-Equivariant Hamiltonian Matrices via SO(2) Local Framescited by 5
- 2024Reducing the Cost of Quantum Chemical Data By Backpropagating Through Density Functional Theorycited by 4
- 2024TraceGrad: a Framework Learning Expressive SO(3)-equivariant Non-linear Representations for Electronic-Structure Hamiltonian Predictioncited by 4
- 2026Transferable SCF-Acceleration through Solver-Aligned Initialization Learningcited by 3
- 2023Electronic excited states from physically-constrained machine learningcited by 2
- 2024Towards Harmonization of SO(3)-Equivariance and Expressiveness: a Hybrid Deep Learning Framework for Electronic-Structure Hamiltonian Predictioncited by 2
- 2024nabla2DFT: A Universal Quantum Chemistry Dataset of Drug-Like Molecules and a Benchmark for Neural Network Potentialscited by 2
- 2025Advancing Universal Deep Learning for Electronic-Structure Hamiltonian Prediction of Materialscited by 2
- 2025Towards A Transferable Acceleration Method for Density Functional Theorycited by 2
- 2026Derivative Informed Learning of Exchange-Correlation Functionalscited by 2
- 2024NeuralSCF: Neural network self-consistent fields for density functional theorycited by 1
- 2025Self-Refining Training for Amortized Density Functional Theorycited by 1
- 2025Tensor Decomposition Networks for Fast Machine Learning Interatomic Potential Computationscited by 1
- 2026M=oLe-Lambda: Learning the Coupled-Cluster Response State for Energies, Gradients, and Propertiescited by 1
- 2025Distributed Equivariant Graph Neural Networks for Large-Scale Electronic Structure Predictioncited by 0
- 2026Coupled Cluster con M=oLe: Molecular Orbital Learning for Neural Wavefunctionscited by 0
- 2026Orbital Transformers for Predicting Wavefunctions in Time-Dependent Density Functional Theorycited by 0
- 2026A Fixed-Point Neural Operator for Size- and Functional-Transferable Hamiltonian Predictioncited by 0
- 2026MALOQ: Massively Accelerated Learning of Operators for Quantum Transportcited by 0
- 2026A Density-Matrix Framework for Electronic-Structure Analysis of Electrolytes for Lithium Batteriescited by 0
- 2026Coupled-cluster molecular properties across the main group that extrapolate beyond training sizecited by 0
- 2026Continuous-Time Quantum Walks based Graph Neural Networkcited by 0
- 2026Equivariant Cellular Sheaves for Molecular Electronic Structure: Bridging Sheaf Cohomology and E(3)-Equivariant Hamiltonian Learningcited by 0
- 2026Physics-Aligned Electronic Ground-State Learning Improves Generalizationcited by 0
- 2026Physics-Aligned Electronic Ground-State Learning Improves Generalization7 Oct 2026
- 2026Continuous-Time Quantum Walks based Graph Neural Network21 Aug 2026
- 2026Coupled-cluster molecular properties across the main group that extrapolate beyond training size18 Aug 2026
- 2026A Density-Matrix Framework for Electronic-Structure Analysis of Electrolytes for Lithium Batteries28 Jul 2026
- 2026MALOQ: Massively Accelerated Learning of Operators for Quantum Transport27 Jun 2026
- 2026Equivariant Cellular Sheaves for Molecular Electronic Structure: Bridging Sheaf Cohomology and E(3)-Equivariant Hamiltonian Learning14 Jun 2026
- 2026A Fixed-Point Neural Operator for Size- and Functional-Transferable Hamiltonian Prediction12 Jun 2026
- 2026Derivative Informed Learning of Exchange-Correlation Functionals2 Jun 2026
- 2026M=oLe-Lambda: Learning the Coupled-Cluster Response State for Energies, Gradients, and Properties28 May 2026
- 2026Transferable SCF-Acceleration through Solver-Aligned Initialization Learning23 Apr 2026
- 2026Orbital Transformers for Predicting Wavefunctions in Time-Dependent Density Functional Theory3 Mar 2026
- 2026Coupled Cluster con M=oLe: Molecular Orbital Learning for Neural Wavefunctions23 Feb 2026
26 more
- 2025Autonomous Agents for Scientific Discovery: Orchestrating Scientists, Language, Code, and Physics10 Oct 2025
- 2025Learning from the electronic structure of molecules across the periodic table30 Sep 2025
- 2025Towards A Transferable Acceleration Method for Density Functional Theory30 Sep 2025
- 2025HIP: Hessian Interatomic Potentials without derivatives25 Sep 2025
- 2025Advancing Universal Deep Learning for Electronic-Structure Hamiltonian Prediction of Materials24 Sep 2025
- 2025Distributed Equivariant Graph Neural Networks for Large-Scale Electronic Structure Prediction4 Jul 2025
- 2025Tensor Decomposition Networks for Fast Machine Learning Interatomic Potential Computations1 Jul 2025
- 2025Efficient Prediction of SO(3)-Equivariant Hamiltonian Matrices via SO(2) Local Frames11 Jun 2025
- 2025Self-Refining Training for Amortized Density Functional Theory2 Jun 2025
- 2025High-order Equivariant Flow Matching for Density Functional Theory Hamiltonian Prediction24 May 2025
- 2025Enhancing the Scalability and Applicability of Kohn-Sham Hamiltonians for Molecular Systems26 Feb 2025
- 2025Efficient and Scalable Density Functional Theory Hamiltonian Prediction through Adaptive Sparsity3 Feb 2025
- 2024Equivariant Graph Network Approximations of High-Degree Polynomials for Force Field Prediction6 Nov 2024
- 2024Learning local equivariant representations for quantum operators8 Jul 2024
- 2024NeuralSCF: Neural network self-consistent fields for density functional theory22 Jun 2024
- 2024nabla2DFT: A Universal Quantum Chemistry Dataset of Drug-Like Molecules and a Benchmark for Neural Network Potentials20 Jun 2024
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency11 Jun 2024
- 2024Infusing Self-Consistency into Density Functional Theory Hamiltonian Prediction via Deep Equilibrium Models6 Jun 2024
- 2024A Space Group Symmetry Informed Network for O(3) Equivariant Crystal Tensor Prediction3 Jun 2024
- 2024TraceGrad: a Framework Learning Expressive SO(3)-equivariant Non-linear Representations for Electronic-Structure Hamiltonian Prediction9 May 2024
- 2024Self-Consistency Training for Density-Functional-Theory Hamiltonian Prediction14 Mar 2024
- 2024Reducing the Cost of Quantum Chemical Data By Backpropagating Through Density Functional Theory6 Feb 2024
- 2024Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Products18 Jan 2024
- 2024Towards Harmonization of SO(3)-Equivariance and Expressiveness: a Hybrid Deep Learning Framework for Electronic-Structure Hamiltonian Prediction1 Jan 2024
- 2023Electronic excited states from physically-constrained machine learning1 Nov 2023
- 2023QH9: A Quantum Hamiltonian Prediction Benchmark for QM9 Molecules15 Jun 2023
- 2024Enabling Efficient Equivariant Operations in the Fourier Basis via Gaunt Tensor Productscited by 35
- 2023QH9: A Quantum Hamiltonian Prediction Benchmark for QM9 Moleculescited by 26
- 2024Equivariance via Minimal Frame Averaging for More Symmetries and Efficiencycited by 18
- 2025Enhancing the Scalability and Applicability of Kohn-Sham Hamiltonians for Molecular Systemscited by 13
- 2024Self-Consistency Training for Density-Functional-Theory Hamiltonian Predictioncited by 12
- 2025HIP: Hessian Interatomic Potentials without derivativescited by 11
- 2025Efficient and Scalable Density Functional Theory Hamiltonian Prediction through Adaptive Sparsitycited by 10
- 2025High-order Equivariant Flow Matching for Density Functional Theory Hamiltonian Predictioncited by 10
- 2024A Space Group Symmetry Informed Network for O(3) Equivariant Crystal Tensor Predictioncited by 7
- 2024Learning local equivariant representations for quantum operatorscited by 7
- 2025Learning from the electronic structure of molecules across the periodic tablecited by 7
- 2024Infusing Self-Consistency into Density Functional Theory Hamiltonian Prediction via Deep Equilibrium Modelscited by 6
12 most cited of 37 All 37 (JSON)
- 2026Physics-Aligned Electronic Ground-State Learning Improves Generalization7 Oct 2026
- 2026Continuous-Time Quantum Walks based Graph Neural Network21 Aug 2026
- 2026Coupled-cluster molecular properties across the main group that extrapolate beyond training size18 Aug 2026
- 2026A Density-Matrix Framework for Electronic-Structure Analysis of Electrolytes for Lithium Batteries28 Jul 2026
- 2026MALOQ: Massively Accelerated Learning of Operators for Quantum Transport27 Jun 2026
- 2026Equivariant Cellular Sheaves for Molecular Electronic Structure: Bridging Sheaf Cohomology and E(3)-Equivariant Hamiltonian Learning14 Jun 2026
- 2026A Fixed-Point Neural Operator for Size- and Functional-Transferable Hamiltonian Prediction12 Jun 2026
- 2026Derivative Informed Learning of Exchange-Correlation Functionals2 Jun 2026
- 2026M=oLe-Lambda: Learning the Coupled-Cluster Response State for Energies, Gradients, and Properties28 May 2026
- 2026Transferable SCF-Acceleration through Solver-Aligned Initialization Learning23 Apr 2026
- 2026Orbital Transformers for Predicting Wavefunctions in Time-Dependent Density Functional Theory3 Mar 2026
- 2026Coupled Cluster con M=oLe: Molecular Orbital Learning for Neural Wavefunctions23 Feb 2026
12 newest of 37 All 37 (JSON)