提出单循环算法SLIP,高效解决非凸双层优化问题。
A Nearly Optimal Single Loop Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothness
- 采用单循环设计,同时更新上下层变量,避免复杂调参。
- 理论证明在期望和高概率下仅需约 $\widetilde{O}(1/ε^4)$ 次梯度或海森向量计算。
- 适用于序列数据元学习等场景,实测性能显著优于现有方法。
本文研究上层目标函数非凸且光滑性可能无界的随机双层优化问题,下层为强凸函数。该问题源于递归神经网络在文本分类等序列数据上的元学习应用,其中上层损失的光滑性常随梯度范数线性增长,可能无界。现有算法依赖嵌套循环结构,需大量调参且不实用。本文提出单循环双层优化器(SLIP):先用几步随机梯度下降更新下层变量,再同步使用带动量的归一化随机梯度下降更新上层变量,同时继续更新下层变量。在标准假设下,算法以期望和高概率找到 $ε$-平稳点,所需随机梯度或海森向量乘积查询次数为 $\widetilde{O}(1/ε^4)$,近似最优(忽略对数因子),无需梯度均方光滑性假设。证明关键在于对下层变量的精细化控制,以及建立双层优化与分布漂移下的随机优化之间的新联系。实验表明,该算法在多种任务中显著优于强基线。
原文摘要 · Abstract (English)
This paper studies the problem of stochastic bilevel optimization where the upper-level function is nonconvex with potentially unbounded smoothness and the lower-level function is strongly convex. This problem is motivated by meta-learning applied to sequential data, such as text classification using recurrent neural networks, where the smoothness constant of the upper-level loss function scales linearly with the gradient norm and can be potentially unbounded. Existing algorithm crucially relies on the nested loop design, which requires significant tuning efforts and is not practical. In this paper, we address this issue by proposing a Single Loop bIlevel oPtimizer (SLIP). The proposed algorithm first updates the lower-level variable by a few steps of stochastic gradient descent, and then simultaneously updates the upper-level variable by normalized stochastic gradient descent with momentum and the lower-level variable by stochastic gradient descent. Under standard assumptions, we show that our algorithm finds an $ε$-stationary point within $\widetilde{O}(1/ε^4)$\footnote{Here $\widetilde{O}(\cdot)$ compresses logarithmic factors of $1/ε$ and $1/δ$, where $δ\in(0,1)$ denotes the failure probability.} oracle calls of stochastic gradient or Hessian-vector product, both in expectation and with high probability. This complexity result is nearly optimal up to logarithmic factors without mean-square smoothness of the stochastic gradient oracle. Our proof relies on (i) a refined characterization and control of the lower-level variable and (ii) establishing a novel connection between bilevel optimization and stochastic optimization under distributional drift. Our experiments on various tasks show that our algorithm significantly outperforms strong baselines in bilevel optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。