A structural optimization algorithm with stochastic forces and stresses
arXiv:2204.12074 · doi:10.1038/s43588-022-00350-w
Abstract
We propose an algorithm for optimizations in which the gradients contain stochastic noise. This arises, for example, in structural optimizations when computations of forces and stresses rely on methods involving Monte Carlo sampling, such as quantum Monte Carlo or neural network states, or are performed on quantum devices which have intrinsic noise. Our proposed algorithm is based on the combination of two key ingredients: an update rule derived from the steepest descent method, and a staged scheduling of the targeted statistical error and step-size, with position averaging. We compare it with commonly applied algorithms, including some of the latest machine learning optimization methods, and show that the algorithm consistently performs efficiently and robustly under realistic conditions. Applying this algorithm, we achieve full-degree optimizations in solids using ab initio many-body computations, by auxiliary-field quantum Monte Carlo with planewaves and pseudopotentials. A new metastable structure in Si was discovered in a mixed geometry and lattice relaxing simulation. In addition to structural optimization in materials, our algorithm can potentially be useful in other problems in various fields where optimization with noisy gradients is needed.
18 pages, 14 figures
References in corpus (6)
- ADADELTA: An Adaptive Learning Rate Method
- Unbiasing Fermionic Quantum Monte Carlo with a Quantum Computer
- Phaseless auxiliary-field quantum Monte Carlo calculations with planewaves and pseudopotentials--applications to atoms and molecules
- Accurate, efficient and simple forces with Quantum Monte Carlo methods
- Coupled Electron-Ion Monte Carlo simulation of hydrogen molecular crystals
- High pressure hydrogen by machine learning and quantum Monte Carlo