arrow
Return

Stochastic Frank-Wolfe Algorithm for Constrained Bilevel Optimization With Improved Per-Iteration Complexity

delete2025-01-01
delete0
PRE
AI
J
Jie Hou
X
Xianlin Zeng
S
Shisheng Cui
江霞 cover
江霞 (Xia Jiang)
孙健 (Jian Sun)
DOI:10.1109/TSP.2025.3596231delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Bilevel optimization, where one optimization problem is inherently nested within another, has gained significant attention due to its extensive applications in machine learning, such as hyperparameter optimization and meta-learning. Most existing algorithms are designed to address unconstrained bilevel optimization problems, with few are capable of effectively tackling complex constrained settings. To address this gap, we propose a novel, fully single-loop stochastic Frank-Wolfe algorithm. This algorithm incorporates a Hessian-inverse-vector approximation technique, momentum-based gradient tracking, and a Frank-Wolfe update. Our proposed algorithm improves per-iteration complexity and achieves lower sample complexity compared to existing Frank-Wolfe algorithms for bilevel optimization. We also conduct numerical simulations to demonstrate the efficacy of our algorithm compared to state-of-the-art methods.
Keywords:
Stochastic Frank-Wolfe algorithm
bilevel optimization
hypergradient approximation
variance reduction technique

Journal

IEEE Transactions on Image Processing cover
IEEE Transactions on Image Processing
IF:
13.7
Papers:
1.0W
Citations:
8.4W

Organization

T
The Chinese University of Hong Kong
Scholars:
3.8K
Papers: 1.9K
Citations: 3
B
beijing institute of technology
Scholars:
5.4W
Papers: 3.9W
Citations: 63