Search papers, labs, and topics across Lattice.
This paper introduces Excited Pfaffians, a novel neural network architecture for representing multiple electronic states within a single model, enabling efficient multi-state calculations in Variational Monte Carlo (VMC). They propose Multi-State Importance Sampling (MSIS) to reduce the computational cost of estimating state overlaps by leveraging samples from all states. Experiments on carbon dimer and beryllium atom demonstrate significant speedups and the ability to model more states compared to traditional approaches, while also showing the capacity to represent excited states across different molecules.
Neural networks can now find all distinct energy levels of the beryllium atom, thanks to a novel wave function architecture that models multiple electronic states simultaneously and trains 200x faster.
Neural-network wave functions in Variational Monte Carlo (VMC) have achieved great success in accurately representing both ground and excited states. However, achieving sufficient numerical accuracy in state overlaps requires increasing the number of Monte Carlo samples, and consequently the computational cost, with the number of states. We present a nearly constant sample-size approach, Multi-State Importance Sampling (MSIS), that leverages samples from all states to estimate pairwise overlap. To efficiently evaluate all states for all samples, we introduce Excited Pfaffians. Inspired by Hartree-Fock, this architecture represents many states within a single neural network. Excited Pfaffians also serve as generalized wave functions, allowing a single model to represent multi-state potential energy surfaces. On the carbon dimer, we match the $O(N_s^4)$-scaling natural excited states while training $>200\times$ faster and modeling 50\% more states. Our favorable scaling enables us to be the first to use neural networks to find all distinct energy levels of the beryllium atom. Finally, we demonstrate that a single wave function can represent excited states across various molecules.
