1: \begin{abstract}

2: Stochastic gradient descent (SGD) is a widely used algorithm in machine learning, particularly for neural network training. Recent  studies on SGD  for  canonical quadratic optimization or linear regression show it attains well generalization under suitable high-dimensional settings.  %benign overfitting or attains well generalization with no explicit regularization

3: However, a fundamental question\textbf{---}\emph{for what kinds of high-dimensional learning problems SGD and its accelerated variants can achieve optimality}\textbf{---} has yet to be well studied. This paper investigates SGD with two essential components in practice: exponentially decaying step size schedule and momentum. We establish the convergence upper bound for momentum accelerated SGD (ASGD) and propose concrete classes of learning problems under which SGD or ASGD achieves  min-max \emph{optimal} convergence rates. The characterization of the target function is based on standard power-law decays in (functional) linear regression. 

4: Our results unveil new insights for understanding the learning bias of SGD: 

5: (i)  SGD  is efficient in learning ``dense'' features where the corresponding weights are subject to an $\ell_{\infty}$ constraint; (ii) SGD is efficient for easy problem without suffering from the saturation effect; (iii)  momentum  can  accelerate the convergence rate by order when the learning problem is relatively hard.  

6: To our knowledge, this is the \emph{first} work to clearly identify the optimal boundary of SGD versus ASGD for the problem under mild settings.

7: 

8: \end{abstract}

9: