Learning Chern Numbers of Topological Insulators with Gauge Equivariant Neural Networks

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Huang, Longde, Balabanov, Oleksandr, Linander, Hampus, Granath, Mats, Persson, Daniel, Gerken, Jan E.
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917932156583936
author Huang, Longde
Balabanov, Oleksandr
Linander, Hampus
Granath, Mats
Persson, Daniel
Gerken, Jan E.
author_facet Huang, Longde
Balabanov, Oleksandr
Linander, Hampus
Granath, Mats
Persson, Daniel
Gerken, Jan E.
contents Equivariant network architectures are a well-established tool for predicting invariant or equivariant quantities. However, almost all learning problems considered in this context feature a global symmetry, i.e. each point of the underlying space is transformed with the same group element, as opposed to a local ``gauge'' symmetry, where each point is transformed with a different group element, exponentially enlarging the size of the symmetry group. Gauge equivariant networks have so far mainly been applied to problems in quantum chromodynamics. Here, we introduce a novel application domain for gauge-equivariant networks in the theory of topological condensed matter physics. We use gauge equivariant networks to predict topological invariants (Chern numbers) of multiband topological insulators. The gauge symmetry of the network guarantees that the predicted quantity is a topological invariant. We introduce a novel gauge equivariant normalization layer to stabilize the training and prove a universal approximation theorem for our setup. We train on samples with trivial Chern number only but show that our models generalize to samples with non-trivial Chern number. We provide various ablations of our setup. Our code is available at https://github.com/sitronsea/GENet/tree/main.
format Preprint
id arxiv_https___arxiv_org_abs_2502_15376
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning Chern Numbers of Topological Insulators with Gauge Equivariant Neural Networks
Huang, Longde
Balabanov, Oleksandr
Linander, Hampus
Granath, Mats
Persson, Daniel
Gerken, Jan E.
Machine Learning
Mesoscale and Nanoscale Physics
Equivariant network architectures are a well-established tool for predicting invariant or equivariant quantities. However, almost all learning problems considered in this context feature a global symmetry, i.e. each point of the underlying space is transformed with the same group element, as opposed to a local ``gauge'' symmetry, where each point is transformed with a different group element, exponentially enlarging the size of the symmetry group. Gauge equivariant networks have so far mainly been applied to problems in quantum chromodynamics. Here, we introduce a novel application domain for gauge-equivariant networks in the theory of topological condensed matter physics. We use gauge equivariant networks to predict topological invariants (Chern numbers) of multiband topological insulators. The gauge symmetry of the network guarantees that the predicted quantity is a topological invariant. We introduce a novel gauge equivariant normalization layer to stabilize the training and prove a universal approximation theorem for our setup. We train on samples with trivial Chern number only but show that our models generalize to samples with non-trivial Chern number. We provide various ablations of our setup. Our code is available at https://github.com/sitronsea/GENet/tree/main.
title Learning Chern Numbers of Topological Insulators with Gauge Equivariant Neural Networks
topic Machine Learning
Mesoscale and Nanoscale Physics
url https://arxiv.org/abs/2502.15376