Transfer learning for scalability of neural-network quantum states - CNRS - Centre national de la recherche scientifique Accéder directement au contenu
Article Dans Une Revue Physical Review E Année : 2020

Transfer learning for scalability of neural-network quantum states

Résumé

Neural-network quantum states have shown great potential for the study of many-body quantum systems. In statistical machine learning, transfer learning designates protocols reusing features of a machine learning model trained for a problem to solve a possibly related but different problem. We propose to evaluate the potential of transfer learning to improve the scalability of neural-network quantum states. We devise and present physics-inspired transfer learning protocols, reusing the features of neural-network quantum states learned for the computation of the ground state of a small system for systems of larger sizes. We implement different protocols for restricted Boltzmann machines on general-purpose graphics processing units. This implementation alone yields a speedup over existing implementations on multicore and distributed central processing units in comparable settings. We empirically and comparatively evaluate the efficiency (time) and effectiveness (accuracy) of different transfer learning protocols as we scale the system size in different models and different quantum phases. Namely, we consider both the transverse field Ising and Heisenberg XXZ models in one dimension, as well as in two dimensions for the latter, with system sizes up to 128 and 8×8 spins. We empirically demonstrate that some of the transfer learning protocols that we have devised can be far more effective and efficient than starting from neural-network quantum states with randomly initialized parameters.

Dates et versions

hal-02865828 , version 1 (10-11-2020)

Identifiants

Citer

Remmy Zen, Long My, Ryan Tan, Frédéric Hébert, Mario Gattobigio, et al.. Transfer learning for scalability of neural-network quantum states. Physical Review E , 2020, 101 (5), ⟨10.1103/PhysRevE.101.053301⟩. ⟨hal-02865828⟩
78 Consultations
0 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More