提出无海森矩阵的多目标双层优化算法,解决非凸下层问题。
Efficient Hessian-Free Methods for Multi-Objective Bilevel Optimization with Nonconvex Lower Level

- 用Moreau包络将双层问题转为单层带约束问题。
- 在确定与随机设置下均证明算法收敛性。
- 适用于少样本元学习与神经架构搜索场景。
多目标双层优化在自动化学习和多任务元学习等人工智能领域有广泛应用。尽管已有研究开始关注该问题,但现有方法依赖于(强)凸的下层问题。事实上,多数多目标双层学习问题的下层是非凸的。为此,本文提出一类基于Moreau包络的无海森矩阵多目标算法(MOMEHA),用于求解具有非凸下层的多目标双层学习问题。具体地,该方法利用Moreau包络将原问题转化为带包络约束的多目标单层优化问题。通过引入平滑加权Tchebycheff标量化策略,保持了单循环、无海森矩阵的计算优势。此外,我们提出了MOMEHA的动量变体(MB-MOMEHA)以处理随机情形。理论上,我们在确定性和随机设置下均给出了算法的收敛性分析。在少样本元学习和神经架构搜索任务上的实验表明,本方法在帕累托前沿表现优于现有方法,验证了其有效性和鲁棒性。
原文摘要 · Abstract (English)
Multi-objective bilevel optimization has wide applications in the AI area such as automated learning and multi-task meta-learning. Although recently some works have been begun to study the multi-objective bilevel optimization, the proposed methods rely on the (strongly) convex lower level problems. In fact, these multi-objective bilevel learning problems are generally nonconvex, and particularly their lower level problems are nonconvex. To fill this gap, we propose a class of Multi-Objective Moreau Envelope based Hessian-free Algorithms (MOMEHA) to solve the multi-objective bilevel learning problems with nonconvex lower level. Specifically, our method uses the Moreau envelope to convert the original problem into a multi-objective single-level optimization with an envelope constraint. In particular, our method retains computational advantages of being single-loop and Hessian-free in the multi-objective setting by incorporating a smooth weighted Tchebycheff scalarization. Furthermore, we propose a momentum-based variant of MOMEHA (i.e., MB-MOMEHA) method to solve the stochastic multi-objective bilevel learning problems. In theory, we provide the convergence properties of our algorithms under both deterministic and stochastic setting. Some experiments on few-shot meta-learning and neural architecture search demonstrate that our methods outperform the existing approaches in Pareto front, validating its effectiveness and robustness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。