Learning local equivariant representations for quantum operators

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhouyin, Zhanghao, Gan, Zixi, Liu, MingKang, Pandey, Shishir Kumar, Zhang, Linfeng, Gu, Qiangqiang
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929752021925888
author Zhouyin, Zhanghao
Gan, Zixi
Liu, MingKang
Pandey, Shishir Kumar
Zhang, Linfeng
Gu, Qiangqiang
author_facet Zhouyin, Zhanghao
Gan, Zixi
Liu, MingKang
Pandey, Shishir Kumar
Zhang, Linfeng
Gu, Qiangqiang
contents Predicting quantum operator matrices such as Hamiltonian, overlap, and density matrices in the density functional theory (DFT) framework is crucial for material science. Current methods often focus on individual operators and struggle with efficiency and scalability for large systems. Here we introduce a novel deep learning model, SLEM (strictly localized equivariant message-passing) for predicting multiple quantum operators, that achieves state-of-the-art accuracy while dramatically improving computational efficiency. SLEM's key innovation is its strict locality-based design for equivariant representations of quantum tensors while preserving physical symmetries. This enables complex many-body dependency without expanding the effective receptive field, leading to superior data efficiency and transferability. Using an innovative SO(2) convolution and invariant overlap parameterization, SLEM reduces the computational complexity of high-order tensor products and is therefore capable of handling systems requiring the $f$ and $g$ orbitals in their basis sets. We demonstrate SLEM's capabilities across diverse 2D and 3D materials, achieving high accuracy even with limited training data. SLEM's design facilitates efficient parallelization, potentially extending DFT simulations to systems with device-level sizes, opening new possibilities for large-scale quantum simulations and high-throughput materials discovery.
format Preprint
id arxiv_https___arxiv_org_abs_2407_06053
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Learning local equivariant representations for quantum operators
Zhouyin, Zhanghao
Gan, Zixi
Liu, MingKang
Pandey, Shishir Kumar
Zhang, Linfeng
Gu, Qiangqiang
Materials Science
Machine Learning
Quantum Physics
Predicting quantum operator matrices such as Hamiltonian, overlap, and density matrices in the density functional theory (DFT) framework is crucial for material science. Current methods often focus on individual operators and struggle with efficiency and scalability for large systems. Here we introduce a novel deep learning model, SLEM (strictly localized equivariant message-passing) for predicting multiple quantum operators, that achieves state-of-the-art accuracy while dramatically improving computational efficiency. SLEM's key innovation is its strict locality-based design for equivariant representations of quantum tensors while preserving physical symmetries. This enables complex many-body dependency without expanding the effective receptive field, leading to superior data efficiency and transferability. Using an innovative SO(2) convolution and invariant overlap parameterization, SLEM reduces the computational complexity of high-order tensor products and is therefore capable of handling systems requiring the $f$ and $g$ orbitals in their basis sets. We demonstrate SLEM's capabilities across diverse 2D and 3D materials, achieving high accuracy even with limited training data. SLEM's design facilitates efficient parallelization, potentially extending DFT simulations to systems with device-level sizes, opening new possibilities for large-scale quantum simulations and high-throughput materials discovery.
title Learning local equivariant representations for quantum operators
topic Materials Science
Machine Learning
Quantum Physics
url https://arxiv.org/abs/2407.06053