In this work we design and train deep neural networks to predict topological invariants for onedimensional 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.Learning topological invariants of these two models is significantly harder than that in Ref. [15], as the mathematical formula of topological invariants in these models are intrinsically more complicated (see Eq.(2) and Eq. (7)) and the sizes of the input data are much larger. Consequently, to guarantee a good performance, neural networks used in this work are much deeper than the one used in Ref. [15]. As shown in Fig. 1, there are more than nine hidden layers in each neural network. Because the neural network becomes more complicated, it becomes more difficult to analyze how the neural network works. Nevertheless, we show that the intermediate output of certain hidden layer is, for case (i) the local winding angle, and for case (ii) the local Berry curvature -both are the integrands in the mathematical formula of the corresponding topological invariant. In this way, we demonstrate that the complicated function fitted by the neural network is essentially the same as the mathematical formula for the topological invariant.The paper is organized as follows. In Section II we train a neural network to learn the winding number of one-dimensional four-band models in AIII class. After introducing the model Hamiltonian and the mathematical formula of the winding number, we present our neural network in detail and report its performance. We then analyze the mechanism of why the neural network works. arXiv:1805.10503v2 [cond-mat.str-el] 9 Jun 2018 * They contribute equally to this work.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.