Send email Copy Email Address
2024-07-21

On Convergence of Incremental Gradient for Non-convex Smooth Functions

Summary

In machine learning and neural network optimization, algorithms like incremental gradient, single shuffle SGD, and random reshuffle SGD are popular due to their cache-mismatch efficiency and good practical convergence behavior. However, their optimization properties in theory, especially for non-convex smooth functions, remain incompletely explored. This paper delves into the convergence properties of SGD algorithms with arbitrary data ordering, within a broad framework for non-convex smooth functions. Our findings show enhanced convergence guarantees for incremental gradient and single shuffle SGD. Particularly if n is the training set size, we improve n times the optimization term of convergence guarantee to reach accuracy ϵ from O(nϵ) to O(1ϵ).

Conference Paper

International Conference on Machine Learning (ICML)

Date published

2024-07-21

Date last modified

2024-10-10