中文

具有最优复杂度的交替随机方差减少算法用于双层优化

最优化与控制 2025-03-18 v2

摘要

本文研究了无约束的非凸-强凸双层优化问题。解决该问题的常用方法是交替更新上层和下层变量,采用(有偏)随机梯度或其变体,其中下层变量的更新方式为单步或多步。在此背景下,我们提出了两种交替随机方差减少算法,即 ALS-SPIDER 和 ALS-STORM,通过引入辅助变量来估计超梯度,以更新上层变量。ALS-SPIDER 采用 SPIDER 估计器进行变量更新,而 ALS-STORM 是一种修改版 ALS-SPIDER,旨在避免每迭代中使用大批量样本。理论上,这两种算法可在任意固定次数的下层变量更新下,找到双层问题的 ϵ\epsilon - stationary 点,样本复杂度为 O(ϵ1.5)O(\epsilon^{-1.5})。迄今为止,我们是已知首个在进行多次下层变量更新时实现 O(ϵ1.5)O(\epsilon^{-1.5}) 最优复杂度的算法。通过数值实验说明了本文算法的有效性。

关键词

引用

@article{arxiv.2404.11377,
  title  = {Alternating Stochastic Variance-Reduced Algorithms with Optimal Complexity for Bilevel Optimization},
  author = {Haimei Huo and Zhixun Su},
  journal= {arXiv preprint arXiv:2404.11377},
  year   = {2025}
}