A central problem in quantum mechanics involves solving the Electronic Schrodinger Equation for a molecule or material. The Variational Monte Carlo approach to this problem approximates a particular variational objective via sampling, and then optimizes this approximated objective over a chosen parameterized family of wavefunctions, known as the ansatz. Recently neural networks have been used as the ansatz, with accompanying success. However, sampling from such wavefunctions has required the use of a Markov Chain Monte Carlo approach, which is inherently inefficient. In this work, we propose a solution to this problem via an ansatz which is cheap to sample from, yet satisfies the requisite quantum mechanical properties. We prove that a normalizing flow using the following two essential ingredients satisfies our requirements: (a) a base distribution which is constructed from Determinantal Point Processes; (b) flow layers which are equivariant to a particular subgroup of the permutation group. We then show how to construct both continuous and discrete normalizing flows which satisfy the requisite equivariance. We further demonstrate the manner in which the non-smooth nature ("cusps") of the wavefunction may be captured, and how the framework may be generalized to provide induction across multiple molecules. The resulting theoretical framework entails an efficient approach to solving the Electronic Schrodinger Equation.
翻译:量子力学中的一个核心问题涉及求解分子或材料的电子薛定谔方程。该问题的变分蒙特卡洛方法通过采样近似特定的变分目标函数,然后在选定的参数化波函数族(称为 ansatz)上优化该近似目标。近年来,神经网络被用作 ansatz 并取得了相应成功。然而,对此类波函数进行采样需要使用马尔可夫链蒙特卡洛方法,这种方法本质上是低效的。在本工作中,我们提出一种解决方案:采用一种采样成本低廉且满足必要量子力学性质的 ansatz。我们证明,使用以下两个关键成分的归一化流满足我们的要求:(a)基于行列式点过程构建的基分布;(b)对置换群特定子群具有等变性的流层。随后,我们展示了如何构建满足所需等变性条件的连续与离散归一化流。我们进一步论证了如何捕捉波函数的非光滑特性("尖点"),以及如何将该框架推广以实现跨多个分子的归纳。所得理论框架为求解电子薛定谔方程提供了一种高效途径。