Neural-network Quantum States for Spin-1 systems: spin-basis and parameterization effects on compactness of representations

Michael Y. Pei,Stephen R. Clark
DOI: https://doi.org/10.3390/e23070879
2021-07-19
Abstract:Neural network quantum states (NQS) have been widely applied to spin-1/2 systems where they have proven to be highly effective. The application to systems with larger on-site dimension, such as spin-1 or bosonic systems, has been explored less and predominantly using spin-1/2 Restricted Boltzmann Machines (RBMs) with a one-hot/unary encoding. Here we propose a more direct generalisation of RBMs for spin-1 that retains the key properties of the standard spin-1/2 RBM, specifically trivial product states representations, labelling freedom for the visible variables and gauge equivalence to the tensor network formulation. To test this new approach we present variational Monte Carlo (VMC) calculations for the spin-1 antiferromagnetic Heisenberg (AFH) model and benchmark it against the one-hot/unary encoded RBM demonstrating that it achieves the same accuracy with substantially fewer variational parameters. Further to this we investigate how the hidden unit complexity of NQS depend on the local single-spin basis used. Exploiting the tensor network version of our RBM we construct an analytic NQS representation of the Affleck-Kennedy-Lieb-Tasaki (AKLT) state in the $xyz$ spin-1 basis using only $M = 2N$ hidden units, compared to $M \sim O(N^2)$ required in the $S^z$ basis. Additional VMC calculations provide strong evidence that the AKLT state in fact possesses an exact compact NQS representation in the $xyz$ basis with only $M=N$ hidden units. These insights help to further unravel how to most effectively adapt the NQS framework for more complex quantum systems.
Quantum Physics,Strongly Correlated Electrons
What problem does this paper attempt to address?
The problem that this paper attempts to solve is: How to construct Neural - network Quantum States (NQS) that are more efficient and have a smaller number of parameters for spin - 1 systems, in order to improve their compactness and accuracy in representing complex quantum states. Specifically, the paper mainly focuses on the following points: 1. **Limitations of existing methods**: - For spin - 1/2 systems, NQS has achieved very good results. However, for systems with larger local dimensions (such as spin - 1 or boson systems), there are relatively few studies on the application of NQS. - Traditional methods usually use "one - hot" or "unary" encoding to handle multi - valued variables, which will significantly increase the number of visible units, resulting in a substantial increase in the number of parameters and optimization complexity, thus limiting the scale of the system. 2. **Proposed new method**: - The paper proposes a method of directly generalizing Restricted Boltzmann Machines (RBMs) to spin - 1 systems, retaining the key characteristics of spin - 1/2 RBMs, such as the ability to describe any direct - product state, the freedom of assigning values to visible variables (labelling freedom), and the equivalence with tensor network representations. - New quadratic biases and interaction weights are introduced to adapt to the characteristics of spin - 1 systems. 3. **Verify the effectiveness of the new method**: - Through Variational Monte Carlo (VMC) calculations, the spin - 1 antiferromagnetic Heisenberg model (AFH model) is studied and compared with the traditional "one - hot" encoding method. The results show that the new method can significantly reduce the number of variational parameters while maintaining the same accuracy. - The influence of different local single - spin bases on the complexity of hidden units is studied, especially the performance in the xyz basis and the Sz basis. 4. **Specific case analysis**: - Through the analysis of the Affleck - Kennedy - Lieb - Tasaki (AKLT) model, it is shown that the new method can accurately represent the AKLT state with only \( M = 2N \) or even \( M = N \) hidden units in the xyz basis, while in the Sz basis, it requires \( M \sim O(N^2) \) hidden units. In summary, this paper aims to improve the structure of NQS to make it more suitable for spin - 1 systems and improve its efficiency and accuracy in representing complex quantum states.