arXiv:2505.08306cs.LGmath.OC2025-05被引 2

多轮随机梯度下降易过拟合,几轮后性能反而下降。

Rapid Overfitting of Multi-Pass Stochastic Gradient Descent in Stochastic Convex Optimization

  • 分析多轮SGD在非光滑凸优化中的泛化性能,发现仅需少量轮次即出现过拟合。
  • 使用最优步长时,第二轮后损失可达Ω(1),远超单轮的Θ(1/√n)。
  • 揭示了非光滑情形下过拟合的相变现象,适合关注实际训练稳定性的研究者。

我们研究了多轮随机梯度下降(SGD)在基本随机凸优化(SCO)模型中的泛化性能。尽管单轮SGD在样本量为n时可达到最优的Θ(1/√n)超出期望损失,但对实践中广泛使用的多轮版本了解甚少。令人意外的是,在一般非光滑情况下,仅需几轮迭代便可能显著损害泛化性能并导致过拟合。具体而言,当使用步长η=Θ(1/√n)(单轮最优),第二轮后总体损失可达Ω(1)。更一般地,从第二轮开始,总体损失为Θ(1/(ηT)+η√T),其中T为总步数。这揭示了第一轮后的泛化行为存在相变,且光滑与非光滑情形下的过拟合速率有明显差异。此外,我们将结果推广至有放回的SGD,证明在O(n log n)步后仍保持相同的渐近界。最后,我们证明了在维度d=˜O(n)时,单轮SGD的泛化间隙下界为Ω(η√n),优于Koren等(2022)和Schliserman等(2024)的结果。

原文摘要 · Abstract (English)

We study the out-of-sample performance of multi-pass stochastic gradient descent (SGD) in the fundamental stochastic convex optimization (SCO) model. While one-pass SGD is known to achieve an optimal $Θ(1/\sqrt{n})$ excess population loss given a sample of size $n$, much less is understood about the multi-pass version of the algorithm which is widely used in practice. Somewhat surprisingly, we show that in the general non-smooth case of SCO, just a few epochs of SGD can already hurt its out-of-sample performance significantly and lead to overfitting. In particular, using a step size $η= Θ(1/\sqrt{n})$, which gives the optimal rate after one pass, can lead to population loss as large as $Ω(1)$ after just one additional pass. More generally, we show that the population loss from the second pass onward is of the order $Θ(1/(ηT) + η\sqrt{T})$, where $T$ is the total number of steps. These results reveal a certain phase-transition in the out-of-sample behavior of SGD after the first epoch, as well as a sharp separation between the rates of overfitting in the smooth and non-smooth cases of SCO. Additionally, we extend our results to with-replacement SGD, proving that the same asymptotic bounds hold after $O(n \log n)$ steps. Finally, we also prove a lower bound of $Ω(η\sqrt{n})$ on the generalization gap of one-pass SGD in dimension $d = \smash{\widetilde O}(n)$, improving on recent results of Koren et al.(2022) and Schliserman et al.(2024).

优化理论过拟合分析随机梯度下降

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。