Jan, 2024

一种基于 Kaczmarz 的方法加速神经网络波函数优化

TL;DR通过使用变分蒙特卡罗方法优化的神经网络波函数,虽然高昂的优化成本限制了其在更大系统中的应用,但是我们提出了子采样投影增量自然梯度下降(SPRING)优化器来减少这个瓶颈。我们证明了 SPRING 优于 MinSR 和 KFAC 在一些小原子和分子上的效果,前提是所有方法的学习率都经过最优调整。例如,对于氧原子,在四万次训练迭代后,SPRING 达到了化学精度,而 MinSR 和 KFAC 即使在十万次迭代后仍然未能达到。