International Conference on Advanced Computational Intelligence (ICACI)
A General RNN Model for Reconstructing Arbitrary Wave Functions
Shuang Cong, Li‐Min Lin
Abstract
A general recurrent neural network (RNN) model is proposed for reconstructing wave functions with varying parameters. This model is suitable for both real and complex ground-state wave functions in condensed matter physics models. We propose three straightforward techniques for integrating variable parameters, enabling the reconstruction of an entire quantum state family within a unified RNN model through numerical augmentation and concatenation. This innovation significantly broadens the model's generalization capabilities. The core unit for wave function reconstruction in the RNN is the gated recurrent unit (GRU), and the RNN's autoregressive property is utilized to generate independent training samples. The training of the network employs a variational Monte Carlo (VMC) optimization strategy. The proposed RNN is applied to two classical condensed matter physics models: the one-dimensional (1D) transverse field Ising model (TFIM) and the 1D J1-J2 model, and it successfully reconstructs arbitrary wave functions. In numerical experiments conducted on a 4-qubit system, using exact diagonalization (ED) as a reference method and relative energy error as a performance indicator, our real- and complex-valued general RNNs demonstrate remarkable reconstruction and generalization performance of wave functions under different physical parameters. To further refine accuracy and reduce training time, we fine-tune the designed network using target samples from a few specific tasks. We present an efficient and accurate general generation model for reconstructing arbitrary wave functions to better accommodate the diversity and variability encountered in practical applications.