Stochastic Gradient Methods with Preconditioned Updates

Abdurakhmon Sadiev,Aleksandr Beznosikov,Abdulla Jasem Almansoori,Dmitry Kamzolov,Rachael Tappenden,Martin Takáč

DOI: https://doi.org/10.1007/s10957-023-02365-3

2024-01-15

Abstract:This work considers the non-convex finite sum minimization problem. There are several algorithms for such problems, but existing methods often work poorly when the problem is badly scaled and/or ill-conditioned, and a primary goal of this work is to introduce methods that alleviate this issue. Thus, here we include a preconditioner based on Hutchinson's approach to approximating the diagonal of the Hessian, and couple it with several gradient-based methods to give new scaled algorithms: Scaled SARAH and Scaled L-SVRG. Theoretical complexity guarantees under smoothness assumptions are presented. We prove linear convergence when both smoothness and the PL condition are assumed. Our adaptively scaled methods use approximate partial second-order curvature information and, therefore, can better mitigate the impact of badly scaled problems. This improved practical performance is demonstrated in the numerical experiments also presented in this work.

Optimization and Control,Machine Learning

What problem does this paper attempt to address?

### Problems Addressed by the Paper

The paper aims to address the issues of poor scaling and ill-conditioning in non-convex finite-sum minimization problems. Specifically, the paper proposes two new algorithms: **Scaled SARAH** and **Scaled L-SVRG**, which incorporate preconditioners to improve performance during the optimization process.

#### Main Contributions
1. **Scaled SARAH**: This is a new algorithm that combines SARAH and PAGE algorithms and introduces a diagonal preconditioner based on the Hutchinson approximation method. This approach can adaptively adjust each element in the search direction, resulting in better performance under poor scaling and ill-conditioned scenarios.
2. **Scaled L-SVRG**: This is another new algorithm based on L-SVRG, incorporating the same preconditioner. This method also leverages adaptive local curvature information to improve practical performance.
3. **Convergence Guarantees**: Theoretical analysis shows that both Scaled SARAH and Scaled L-SVRG have linear convergence rates under smoothness and PL conditions.
4. **Numerical Experiments**: Extensive numerical experiments were conducted to validate the practical performance of the new algorithms under different parameter settings. The experimental results indicate that the new algorithms with preconditioners often outperform methods without preconditioners and frequently surpass the Adam algorithm.

### Summary
The main goal of the paper is to improve the performance of optimization algorithms in non-convex optimization problems by introducing preconditioners, especially under poor scaling and ill-conditioned scenarios. Through theoretical analysis and experimental validation, the effectiveness and superiority of the new algorithms are demonstrated.