SE(3)-Transformers for predicting the electronic structure of hydrogen molecules


SE(3)-Transformers for predicting the electronic structure of hydrogen molecules

Brzoza, B.; Cangi, A.

In this work, we demonstrate the efficacy of a neural network model
implemented as the Materials Learning Algorithms (MALA) package
in predicting the electronic structure of a system of hydrogen molecules
under various pressure and temperature conditions across the molecular liquid-solid phase boundary, demonstrating the potential of our
methods for molecular systems. Additionally, we investigate the use
of SE(3)-Transformer Graph Neural Networks to improve the generalizability and extrapolation capabilities of our models. Our results
indicate that the MALA framework provides a powerful and efficient
tool for accelerating Kohn-Sham density functional theory calculations
in molecular systems. This work paves the way for future research in
developing advanced machine-learning algorithms for accelerating electronic structure calculations both accurately and efficiently.

Keywords: MALA; DFT; GNN; SE(3); Equivariant; Neural Networks; Electronic Structure

  • Lecture (Conference)
    DPG Spring Meeting, 18.-21.03.2024, Berlin, Deutschland

Permalink: https://www.hzdr.de/publications/Publ-38901