Deep Learning Topological Invariants of Band Insulators

Abstract

In this work we design and train deep neural networks to predict topological invariants for one-dimensional four-band insulators in AIII class whose topological invariant is the winding number, and two-dimensional two-band insulators in A class whose topological invariant is the Chern number. Given Hamiltonians in the momentum space as the input, neural networks can predict topological invariants for both classes with accuracy close to or higher than 90%, even for Hamiltonians whose invariants are beyond the training data set. Despite the complexity of the neural network, we find that the output of certain intermediate hidden layers resembles either the winding angle for models in AIII class or the solid angle (Berry curvature) for models in A class, indicating that neural networks essentially capture the mathematical formula of topological invariants. Our work demonstrates the ability of neural networks to predict topological invariants for complicated models with local Hamiltonians as the only input, and offers an example that even a deep neural network is understandable.

8 pages, 5 figures

References in corpus (43)

• Classification of topological quantum matter with symmetries
• Machine learning phases of matter
• Chern Numbers in Discretized Brillouin Zone: Efficient Method of Computing (Spin) Hall Conductances
• Searching for Exotic Particles in High-Energy Physics with Deep Learning
• Rotation-invariant convolutional neural networks for galaxy morphology prediction
• Learning phase transitions by confusion
• Discovering Phase Transitions with Unsupervised Learning
• Unsupervised learning of phase transitions: from principal component analysis to variational autoencoders
• Quantum Entanglement in Neural Network States
• Data Mining and Machine Learning in Astronomy
• Machine learning quantum phases of matter beyond the fermion sign problem
• Discovering Phases, Phase Transitions and Crossovers through Unsupervised Machine Learning: A critical examination
• Restricted-Boltzmann-Machine Learning for Solving Strongly Correlated Quantum Systems
• Quantum Loop Topography for Machine Learning
• Machine Learning Topological Invariants with Neural Networks
• Machine Learning Phases of Strongly Correlated Fermions
• Self-Learning Monte Carlo Method
• Accelerate Monte Carlo Simulations with Restricted Boltzmann Machines
• Detection of phase transition via convolutional neural network
• Unsupervised machine learning account of magnetic transitions in the Hubbard model
• Probing many-body localization with neural networks
• Machine learning out-of-equilibrium phases of matter
• Machine learning for many-body physics: The case of the Anderson impurity model
• Kernel methods for interpretable machine learning of order parameters
• Unsupervised Learning of Frustrated Classical Spin Models I: Principle Component Analysis
• Deep Learning the Quantum Phase Transitions in Random Two-Dimensional Electron Systems
• Machine Learning ${\mathbb{Z}}_2$ Quantum Spin Liquids with Quasi-particle Statistics
• A Separability-Entanglement Classifier via Machine Learning
• Discriminative Cooperative Networks for Detecting Phase Transitions
• Self-Learning Monte Carlo Method in Fermion Systems
• Parameter diagnostics of phases and phase transition learning by neural networks
• Machine Learning Spatial Geometry from Entanglement Features
• Self-learning Monte Carlo with Deep Neural Networks
• Self-Learning Determinantal Quantum Monte Carlo Method
• Deep Learning the Quantum Phase Transitions in Random Electron Systems: Applications to Three Dimensions
• Self-Learning Monte Carlo Method: Continuous-Time Algorithm
• Applications of neural networks to the studies of phase transitions of two-dimensional Potts models
• Principal Component Analysis for Fermionic Critical Points
• Deep Neural Network Detects Quantum Phase Transition
• Recommender Engine for Continuous Time Quantum Monte Carlo Methods
• Phase Diagrams of Three-Dimensional Anderson and Quantum Percolation Models using Deep Three-Dimensional Convolutional Neural Network
• Identifying Product Order with Restricted Boltzmann Machines
• The discrepancy between Einstein mass and dynamical mass for SIS and power-law mass models

Cited by in corpus (43)

• Machine learning and the physical sciences
• Provably efficient machine learning for quantum many-body problems
• Identifying Quantum Phase Transitions using Artificial Neural Networks on Experimental Data
• Unsupervised machine learning and band topology
• Quantum Adversarial Machine Learning
• AI Poincaré: Machine Learning Conservation Laws from Trajectories
• Topological quantum phase transitions retrieved through unsupervised machine learning
• Phase diagram of disordered higher-order topological insulator: A machine learning study
• Smallest Neural Network to Learn the Ising Criticality
• Drawing Phase Diagrams of Random Quantum Systems by Deep Learning the Wave Functions
• Machine learning topological phases in real space
• Interpreting machine learning of topological quantum phase transitions
• Hybrid convolutional neural network and PEPS wave functions for quantum many-particle states
• Machine learning topological invariants of non-Hermitian systems
• Unsupervised and supervised learning of interacting topological phases from single-particle correlation functions
• Unsupervised learning using topological data augmentation
• Phase transition encoded in neural network
• Artificial Neural Network Based Computation for Out-of-Time-Ordered Correlators
• Rapid Exploration of Topological Band Structures using Deep Learning
• Machine prediction of topological transitions in photonic crystals
• Machine learning non-Hermitian topological phases
• Applications of deep learning to relativistic hydrodynamics
• A supervised learning algorithm for interacting topological insulators based on local curvature
• Can a CNN trained on the Ising model detect the phase transition of the $q$-state Potts model?
• Multi-faceted machine learning of competing orders in disordered interacting systems
• Engineering Topological Phases Guided by Statistical and Machine Learning Methods
• Deep learning of deformation-dependent conductance in thin films: nanobubbles in graphene
• Unsupervised identification of Floquet topological phase boundaries
• Exponentially improved efficient machine learning for quantum many-body states with provable guarantees
• Unsupervised interpretable learning of topological indices invariant under permutations of atomic bands
• Deep Learning of Phase Transitions for Quantum Spin Chains from Correlation Aspects
• Machine learning of mirror skin effects in the presence of disorder
• Deep learning of topological phase transitions from entanglement aspects for two-dimensional chiral p-wave superconductors
• Discovering conservation laws from trajectories via machine learning
• Quaternion-based machine learning on topological quantum systems
• Renormalization-group-inspired neural networks for computing topological invariants
• Preparing Quantum States by Measurement-feedback Control with Bayesian Optimization
• Supervised learning of an interacting 2D hard-core boson model of a weak topological insulator using correlation functions
• Quantum circuit complexity and unsupervised machine learning of topological order
• Deep learning of topological phase transitions from entanglement aspects: An unsupervised way
• Topological Learning in Multi-Class Data Sets
• Machine learning that predicts well may not learn the correct physical descriptions of glassy systems
• Machine Learning Study on the Flat-Band States Constructed by Molecular-Orbital Representation with Randomness