56
5

On the equivalence of different adaptive batch size selection strategies for stochastic gradient descent methods

Abstract

In this study, we demonstrate that the norm test and inner product/orthogonality test presented in \cite{Bol18} are equivalent in terms of the convergence rates associated with Stochastic Gradient Descent (SGD) methods if ϵ2=θ2+ν2\epsilon^2=\theta^2+\nu^2 with specific choices of θ\theta and ν\nu. Here, ϵ\epsilon controls the relative statistical error of the norm of the gradient while θ\theta and ν\nu control the relative statistical error of the gradient in the direction of the gradient and in the direction orthogonal to the gradient, respectively. Furthermore, we demonstrate that the inner product/orthogonality test can be as inexpensive as the norm test in the best case scenario if θ\theta and ν\nu are optimally selected, but the inner product/orthogonality test will never be more computationally affordable than the norm test if ϵ2=θ2+ν2\epsilon^2=\theta^2+\nu^2. Finally, we present two stochastic optimization problems to illustrate our results.

View on arXiv
Comments on this paper