arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.08025math.OCcs.LGstat.ML

稳定边缘机制下参数化线性回归的判据与偏差

Criteria and Bias of Parameterized Linear Regression under Edge of Stability Regime

Yale University(耶鲁大学)

Peiyuan Zhang, Amin Karbasi

AI总结:

该研究针对稳定边缘(EoS)现象,通过参数化线性回归任务从理论和实验证明二次损失下也可出现EoS,揭示了大步长下对角线性网络的隐式偏差,拓展了EoS相关研究的适用模型范围。

摘要:

经典优化理论要求基于梯度的方法采用小步长才能收敛。然而,近期研究发现对传统观点提出了挑战:实验证明即使步长$η$超过$2/L$的阈值(其中$L$为全局光滑常数),梯度下降(GD)依然能够收敛,这一现象通常被称为稳定边缘(Edge of Stability, EoS)现象。学界普遍认为,具有次二次增长的目标函数在引发EoS现象中发挥着重要作用。\n本文针对回归任务中寻找线性插值器$β\in R^{d}$的问题给出了更全面的解答,其中损失函数为$l(\cdot)$,$β$可参数化为$β= w^2_{+} - w^2_{-}$。与此前认为次二次$l$是EoS必要条件的研究不同,本文的新发现表明:在适当条件下,即使$l$为二次函数,EoS现象依然会发生。本文通过实验与理论证据双重验证了这一结论的严谨性,证明梯度下降轨迹会以非渐近的方式收敛到线性插值器。此外,二次$l$下的模型(也被称为深度为2的对角线性网络)在EoS机制下的研究目前仍较为匮乏。本文的分析为采用大步长时对角线性网络的隐式偏差提供了新的见解,丰富了对更具实用性的模型中EoS现象的理解。

英文摘要:

Classical optimization theory requires a small step-size for gradient-based methods to converge. Nevertheless, recent findings challenge the traditional idea by empirically demonstrating Gradient Descent (GD) converges even when the step-size $η$ exceeds the threshold of $2/L$, where $L$ is the global smooth constant. This is usually known as the Edge of Stability (EoS) phenomenon. A widely held belief suggests that an objective function with subquadratic growth plays an important role in incurring EoS. In this paper, we provide a more comprehensive answer by considering the task of finding linear interpolator $β\in R^{d}$ for regression with loss function $l(\cdot)$, where $β$ admits parameterization as $β= w^2_{+} - w^2_{-}$. Contrary to the previous work that suggests a subquadratic $l$ is necessary for EoS, our novel finding reveals that EoS occurs even when $l$ is quadratic under proper conditions. This argument is made rigorous by both empirical and theoretical evidence, demonstrating the GD trajectory converges to a linear interpolator in a non-asymptotic way. Moreover, the model under quadratic $l$, also known as a depth-$2$ diagonal linear network, remains largely unexplored under the EoS regime. Our analysis then sheds some new light on the implicit bias of diagonal linear networks when a larger step-size is employed, enriching the understanding of EoS on more practical models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑