Search papers, labs, and topics across Lattice.
This paper investigates the robustness of Variational Monte Carlo (VMC) by analyzing the impact of nodal geometry on the integrability of local energy and gradient estimators. The authors establish that for many practical ansatz classes, these estimators exhibit heavy-tailed behavior, leading to failures in higher moment convergence. They introduce a new method, PS-Clip-VMC, which enhances robustness by clipping local energy and gradient estimators, demonstrating improved convergence properties in preliminary experiments with FermiNet on multi-electron systems.
Heavy-tailed estimators in Variational Monte Carlo can undermine convergence, but a new clipping method shows promise for robust optimization in electronic structure calculations.
Variational Monte Carlo (VMC) is a central algorithm in electronic structure theory and has gained renewed importance through modern neural-network ans盲tze such as FermiNet. At its core, VMC seeks ground states by minimizing the Rayleigh quotient by stochastic optimization. In this work, we show that the resulting stochastic optimization problem is intrinsically governed by the nodal geometry of the underlying wave function. More precisely, we establish that properties of the nodal set determine the integrability of the local energy and gradient estimators that drive VMC. For broad and practically relevant ansatz classes, including Slater-Jastrow wave functions with variable-exponent Slater-type orbitals, we prove that these estimators are generically heavy-tailed and fail to admit higher moments. At the same time, for general analytic ans盲tze, we prove weak moment bounds for the relevant estimators and identify precise low-moment regimes, showing how generic and degenerate nodal structures lead to different integrability thresholds. Building on this analysis, we introduce a new robust variant of VMC $\unicode{x2013}$ coined PS-Clip-VMC $\unicode{x2013}$ which is based on clipping both the local energy and the gradient random variable. We prove that PS-Clip-VMC converges both in expectation and with high probability in the weak moment regime of VMC. Preliminary experiments for training FermiNet on Atoms with up to 18 electrons suggest that PS-Clip-VMC is significantly more robust than standard methods.
