提出新算法AccBO,将双层优化的收敛速度提升至1/ε³级,显著优于已有方法。
An Accelerated Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothness
- 上层用带递归动量的归一化随机梯度,下层用平均化的随机Nesterov加速梯度。
- 在下层梯度方差为O(ε)时,达到˜O(1/ε³)的最优器调用复杂度。
- 适用于需要快速双层优化的序列学习任务,如RNN文本分类。
本文研究一类上层目标函数非凸且平滑性无界、下层问题强凸的随机双层优化问题,这类问题在序列数据学习(如基于循环神经网络的文本分类)中有重要应用。上层函数的平滑性常数随梯度范数线性增长,缺乏统一上界。现有最优算法需˜O(1/ε⁴)次随机梯度或海森/雅可比-向量积调用才能找到ε-驻点。本文提出新算法AccBO:上层变量采用归一化随机梯度下降加递归动量更新,下层变量使用平均化的随机Nesterov加速梯度法。理论证明,在下层随机梯度方差为O(ε)条件下,算法实现˜O(1/ε³)的最优器复杂度。核心贡献在于提出一个刻画分布漂移下随机Nesterov加速梯度动态的新引理,该引理对分析超梯度估计误差至关重要,具有独立研究价值。实验验证了理论加速效果,算法显著优于基线。
原文摘要 · Abstract (English)
This paper investigates a class of stochastic bilevel optimization problems where the upper-level function is nonconvex with potentially unbounded smoothness and the lower-level problem is strongly convex. These problems have significant applications in sequential data learning, such as text classification using recurrent neural networks. The unbounded smoothness is characterized by the smoothness constant of the upper-level function scaling linearly with the gradient norm, lacking a uniform upper bound. Existing state-of-the-art algorithms require $\widetilde{O}(1/ε^4)$ oracle calls of stochastic gradient or Hessian/Jacobian-vector product to find an $ε$-stationary point. However, it remains unclear if we can further improve the convergence rate when the assumptions for the function in the population level also hold for each random realization almost surely. To address this issue, we propose a new Accelerated Bilevel Optimization algorithm named AccBO. The algorithm updates the upper-level variable by normalized stochastic gradient descent with recursive momentum and the lower-level variable by the stochastic Nesterov accelerated gradient descent algorithm with averaging. We prove that our algorithm achieves an oracle complexity of $\widetilde{O}(1/ε^3)$ to find an $ε$-stationary point, when the lower-level stochastic gradient's variance is $O(ε)$. Our proof relies on a novel lemma characterizing the dynamics of stochastic Nesterov accelerated gradient descent algorithm under distribution drift with high probability for the lower-level variable, which is of independent interest and also plays a crucial role in analyzing the hypergradient estimation error over time. Experimental results on various tasks confirm that our proposed algorithm achieves the predicted theoretical acceleration and significantly outperforms baselines in bilevel optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。