随机双层优化中一阶方法的复杂度研究
最优化与控制
2024-02-13 v1 机器学习
摘要
我们考虑在下层问题无约束且强凸的情况下,在双层优化中寻找稳定点的问题。该问题近年来得到了广泛研究;主要的技术挑战在于跟踪下层解 以响应上层变量 的变化。随后,所有现有方法都将其分析与一个知晓下层解的“精灵”算法绑定,因此无需查询远离这些解的任何点。我们考虑与这些方法对偶的一个问题:假设我们有一个称为 -aware 的预言机,它除了在 周围的 -球内提供局部无偏的一阶梯度估计外,还能返回下层解的 估计。我们研究了使用这种 -aware 预言机寻找稳定点的复杂度:我们提出了一种简单的一阶方法,该方法使用 、 次对一阶 -aware 预言机的访问即可收敛到一个 稳定点。我们的上界也适用于标准的无偏一阶预言机,在最小假设下将一阶方法的最佳已知复杂度改进了 。然后,我们分别在未对 -aware 预言机施加额外平滑性假设和施加该假设的情况下,提供了匹配的 、 下界。我们的结果表明,任何模拟使用 -aware 预言机的算法的方法都必须承受相同的下界。
引用
@article{arxiv.2402.07101,
title = {On the Complexity of First-Order Methods in Stochastic Bilevel Optimization},
author = {Jeongyeol Kwon and Dohyun Kwon and Hanbaek Lyu},
journal= {arXiv preprint arXiv:2402.07101},
year = {2024}
}