最优 Hessian/Jacobian-free Nonconvex-PL 双层优化
最优化与控制
2024-07-26 v1 机器学习
摘要
双层优化在许多机器学习任务中得到广泛应用,如超参数学习、元学习和强化学习。尽管最近开发了许多算法来解决双层优化问题,但它们通常依赖于(强)凸的下层问题。更近期地,有一些方法被提出来解决非凸-PL 双层优化问题,其中上层问题可能是非凸的,下层问题也可能是非凸的,但满足 Polyak-{\L}ojasiewicz (PL) 条件。然而,这些方法仍然具有高收敛复杂度或高计算复杂度,例如需要计算昂贵的 Hessian/Jacobian 矩阵及其逆矩阵。在本文中,我们提出了一种高效的 Hessian/Jacobian-free 方法 (即 HJFBiO),用于解决非凸-PL 双层问题。理论上,在一些温和条件下,我们证明了我们的 HJFBiO 方法获得 的最优收敛率,其中 表示迭代次数,在寻找 -stationary 解方面具有 的最优梯度复杂度。我们在 bilevel PL game 和 hyper-representation learning 任务上进行数值实验,以演示我们所提出方法的有效性。
引用
@article{arxiv.2407.17823,
title = {Optimal Hessian/Jacobian-Free Nonconvex-PL Bilevel Optimization},
author = {Feihu Huang},
journal= {arXiv preprint arXiv:2407.17823},
year = {2024}
}
备注
ICML 2024 (Oral). arXiv admin note: text overlap with arXiv:2311.04520