Trajectory surface hopping combined with ab initio electronic structure calculations is a popular and powerful approach for on-the-fly nonadiabatic dynamics simulations. For large systems, however, this remains a significant challenge because of the unaffordable computational cost of large-scale electronic structure calculations. Here, we present an efficient divide-and-conquer approach to construct the system Hamiltonian based on Wannier analysis and machine learning. In detail, the large system under investigation is first decomposed into small building blocks, and then all possible segments formed by building blocks within a cutoff distance are found out. Ab initio molecular dynamics is carried out to generate a sequence of geometries for each equivalent segment with periodicity. The Hamiltonian matrices in the maximum localized Wannier function (MLWF) basis are obtained for all geometries and utilized to train artificial neural networks (ANNs) for the structure-dependent Hamiltonian elements. Taking advantage of the orthogonality and spatial locality of MLWFs, the one-electron Hamiltonian of a large system at arbitrary geometry can be directly constructed by the trained ANNs. As demonstrations, we study charge transport in a zigzag graphene nanoribbon (GNR), a coved GNR, and a series of hybrid GNRs with a state-of-the-art surface hopping method. The interplay between delocalized and localized states is found to determine the electron dynamics in hybrid GNRs. Our approach has successfully studied GNRs with >10 000 atoms, paving the way for efficient and reliable all-atom nonadiabatic dynamics simulation of general systems.
Keywords: Wannier analysis; charge transport; graphene nanoribbons; machine learning; nonadiabatic dynamics.