Efficient Numerical Algorithm for Large-Scale Damped Natural Gradient Descent
Voice is AI-generated
Connected to paperThis paper is a preprint and has not been certified by peer review
Efficient Numerical Algorithm for Large-Scale Damped Natural Gradient Descent
Yixiao Chen, Hao Xie, Han Wang
AbstractWe propose a new algorithm for efficiently solving the damped Fisher matrix in large-scale scenarios where the number of parameters significantly exceeds the number of available samples. This problem is fundamental for natural gradient descent and stochastic reconfiguration. Our algorithm is based on Cholesky decomposition and is generally applicable. Benchmark results show that the algorithm is significantly faster than existing methods.