提出双空间预条件梯度下降法,加速过参数化模型收敛并揭示隐式偏差机制。
Dual Space Preconditioning for Gradient Descent in the Overparameterized Regime
- 设计基于凸函数梯度的双空间预条件算子,统一多种优化器框架。
- 证明全批量下算法指数收敛至完美拟合解,且在特定条件下逼近标准GD极限。
- 适用于研究过参数化模型的收敛性与隐式正则化,适合优化理论研究者。
本文研究双空间预条件梯度下降(Dual Space Preconditioned Gradient Descent)在过参数化情形下的收敛性,涵盖归一化梯度下降、梯度裁剪等优化器。考虑形式为∇K的预条件器,其中K: ℝ^{d×k}→ℝ为凸函数,用于训练过参数化线性模型,损失函数为ℓ(XW−Y),权重W∈ℝ^{d×k},标签Y∈ℝ^{n×k},数据X∈ℝ^{n×d}。在上述假设下,证明全批量预条件梯度下降的迭代序列以指数速率收敛至满足XW∞=Y的点W∞∈ℝ^{d×k}。进一步分析其隐式偏差:对一般K(·),W∞依赖于固定步长,难以精确刻画;给出近似隐式偏差性质:‖W₀−W∞‖_F ≤ c‖W₀−W_GD,∞‖_F(c>0),其中W_GD,∞为同初始化下标准梯度下降的极限点。对于形式为K(G)=h(‖G‖_F)的各向同性预条件器,以及任意批量大小的随机变体,证明线性收敛至W_GD,∞。实验表明,使用平滑矩阵弹性网作为预条件器,在非线性模型上实现更快收敛。
原文摘要 · Abstract (English)
In this work, we study the convergence properties of the Dual Space Preconditioned Gradient Descent, encompassing optimizers such as Normalized Gradient Descent and Gradient Clipping. We consider preconditioners of the form $\nabla K$, where $K: \mathbb{R}^{d \times k} \to \mathbb{R}$ is convex and apply $\nabla K(\cdot)$ to train an over-parameterized linear model with a convex loss of the form $\ell(X W - Y)$, for weights $W \in \mathbb{R}^{d \times k}$, labels $Y \in \mathbb{R}^{n \times k}$ and data $X \in \mathbb{R}^{n \times d}$. Under the aforementioned assumptions, we prove that the iterates of the full-batch preconditioned gradient descent converge at an exponential rate to a point $W_{\infty} \in \mathbb{R}^{d \times k}$ satisfying $XW_{\infty} = Y$. We also study the implicit bias of Dual Space Preconditioned Gradient Descent. First, we demonstrate analytically and empirically that, for general $K(\cdot)$, $W_\infty$ depends on the chosen constant step size, hindering a precise characterization of the implicit bias. We also provide an approximate implicit bias property for general preconditioners, namely, $\|W_0 - W_{\infty}\|_F \le c \|W_0 - W_{\text{GD}, \infty}\|_F$ for a constant $c>0$ and $W_{\text{GD}, \infty}$ denoting the convergence point of GD initialized at $W_0$. Furthermore, for preconditioners of the form $K(G) = h(\|G\|_F)$, known as {\it isotropic preconditioners}, and for the stochastic variation of the algorithm with arbitrary batch-size, we prove linear convergence to $W_{\text{GD}, \infty}$. Finally, in the experiments, we demonstrate faster convergence on a nonlinear model obtained using the smoothed matrix elastic-net as a preconditioner.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。