Hierarchical autoregressive neural networks for statistical systems

TL;DR

This paper introduces a hierarchical neural network approach for statistical systems that reduces computational costs and memory requirements, improving the accuracy of probability distribution approximation and efficiency in Monte Carlo simulations.

Contribution

It proposes a hierarchical association of degrees of freedom to neural networks, scaling with system size's linear dimension, enhancing training quality and reducing memory needs.

Findings

Improved neural network training quality and distribution approximation.

Closer variational free energy to theoretical values.

Reduced autocorrelation time in Monte Carlo simulations.

Abstract

It was recently proposed that neural networks could be used to approximate many-dimensional probability distributions that appear e.g. in lattice field theories or statistical mechanics. Subsequently they can be used as variational approximators to asses extensive properties of statistical systems, like free energy, and also as neural samplers used in Monte Carlo simulations. The practical application of this approach is unfortunately limited by its unfavorable scaling both of the numerical cost required for training, and the memory requirements with the system size. This is due to the fact that the original proposition involved a neural network of width which scaled with the total number of degrees of freedom, e.g. $L^2$ in case of a two dimensional $L\times L$ lattice. In this work we propose a hierarchical association of physical degrees of freedom, for instance spins, to neurons which replaces it with the scaling with the linear extent $L$ of the system. We demonstrate our approach on the two-dimensional Ising model by simulating lattices of various sizes up to $128 \times 128$ spins, with time benchmarks reaching lattices of size $512 \times 512$. We observe that our proposal improves the quality of neural network training, i.e. the approximated probability distribution is closer to the target that could be previously achieved. As a consequence, the variational free energy reaches a value closer to its theoretical expectation and, if applied in a Markov Chain Monte Carlo algorithm, the resulting autocorrelation time is smaller. Finally, the replacement of a single neural network by a hierarchy of smaller networks considerably reduces the memory requirements.