Title of article
Modified Step Size for Enhanced Stochastic Gradient Descent: Convergence and Experiments
Author/Authors
Soheil shamaee ، Mahsa Department of Computer Science - Faculty of Mathematical Science - University of Kashan , Fathi Hafshejani ، Sajad Department of Applied Mathematics - Shiraz University of Technology
From page
237
To page
253
Abstract
This paper introduces a novel approach to enhance the performance of the stochastic gradient descent (SGD) algorithm by incorporating a modified decay step size based on \frac{1}{\sqrt{t}}. The proposed step size integrates a logarithmic term, leading to the selection of smaller values in the final iterations. Our analysis establishes a convergence rate of O(\frac{\ln T}{\sqrt{T}}) for smooth non-convex functions without the Polyak-Łojasiewicz condition. To evaluate the effectiveness of our approach, we conducted numerical experiments on image classification tasks using the Fashion-MNIST and CIFAR10 datasets, and the results demonstrate significant improvements in accuracy, with enhancements of $0.5\% and 1.4\% observed, respectively, compared to the traditional \frac{1}{\sqrt{t}} step size. The source code can be found at اttps://github.com/Shamaeem/LNSQRTStepSize.
Keywords
Stochastic gradient descent , Decay step size , Convergence rate
Journal title
Mathematics Interdisciplinary Research
Journal title
Mathematics Interdisciplinary Research
Record number
2765867
Link To Document