Lower Bounds on Inference Complexity for Polyak-Ruppert Averaged Stochastic Gradient Descent
Organizations: University of Duisburg-Essen, Germany · HSE University, Moscow, Russia
Abstract
In this paper, we establish sharp lower bounds for Gaussian approximation of Polyak-Ruppert averaged stochastic gradient descent with polynomially decreasing step sizes. For every fixed exponent , a single one-dimensional smooth, strongly convex example has approximation error with the limiting variance of order in Kolmogorov and convex distances. This matches the existing upper bound throughout the range of . The nonlinear bias contributes even when the Gaussian reference uses the finite linearized variance. We also derive a lower bound of order for the relative mean error of estimating the limiting variance from one SGD trajectory, uniformly over a scalar quadratic family. With the limiting variance known, achieving approximation error on this example takes order iterations for the best choice . Achieving both Gaussian approximation and relative mean variance-estimation errors at most uniformly over the model class requires at least order iterations for every fixed .
Stochastic Gradient Descent
Stochastic Gradient Descent (SGD) uses noisy, data-sampled gradients to train models and solve large-scale optimization problems.
See all 237 papers