An Adaptive Algorithm for Bilevel Optimization on Riemannian Manifolds
Xu Shi, Rufeng Xiao, Rujun Jiang
Abstract
Existing methods for solving Riemannian bilevel optimization (RBO) problems require prior knowledge of the problem's first- and second-order information and curvature parameter of the Riemannian manifold to determine step sizes, which poses practical limitations when these parameters are unknown or computationally infeasible to obtain. In this paper, we introduce the Adaptive Riemannian Hypergradient Descent (AdaRHD) algorithm for solving RBO problems. To our knowledge, AdaRHD is the first method to incorporate a fully adaptive step size strategy that eliminates the need for problem-specific parameters in RBO. We prove that AdaRHD achieves an iteration complexity for finding an -stationary point, thus matching the complexity of existing non-adaptive methods. Furthermore, we demonstrate that substituting exponential mappings with retraction mappings maintains the same complexity bound. Experiments demonstrate that AdaRHD achieves comparable performance to existing non-adaptive approaches while exhibiting greater robustness.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext fe59e8f8-2333-47d4-b1fb-289da2dae7c0Cited by top-tier papers1
Ask how each one uses itBuilds on20
- Bilevel Optimization: Convergence Analysis and Enhanced DesignKaiyi Ji, Junjie Yang, Yingbin LiangICML 2021 · 343 citations
- On the Iteration Complexity of Hypergradient ComputationRiccardo Grazzi, Luca Franceschi, Massimiliano Pontil, Saverio SalzoICML 2020 · 241 citations
- Closing the Gap: Tighter Analysis of Alternating Stochastic Gradient Methods for Bilevel ProblemsTianyi Chen, Yuejiao Sun, Wotao YinNeurIPS 2021 · 176 citations
- Adaptive Gradient Descent without DescentYura Malitsky, Konstantin MishchenkoICML 2020 · 171 citations
- BOME! Bilevel Optimization Made Easy: A Simple First-Order ApproachBo Liu, Mao Ye, Stephen Wright, Peter Stone et al.NeurIPS 2022 · 170 citations
Related papers
- A Framework for Bilevel Optimization on Riemannian ManifoldsAndi Han, Bamdev Mishra, Pratik Kumar Jawanpuria, Akiko TakedaNeurIPS 2024 · 9 citations
- Riemannian Accelerated Zeroth-order Algorithm: Improved Robustness and Lower Query ComplexityChang He, Zhaoye Pan, Xiao Wang, Bo JiangICML 2024 · 8 citations
- Finite-Time Analysis of Stochastic Nonconvex Nonsmooth Optimization on the Riemannian ManifoldsEmre Sahinoglu, Youbang Sun, Shahin ShahrampourNeurIPS 2025 · 4 citations
- Learning-Rate-Free Stochastic Optimization over Riemannian ManifoldsDaniel Dodd, Louis Sharrock, Christopher NemethICML 2024 · 1 citation
- Double Momentum Method for Lower-Level Constrained Bilevel OptimizationWanli Shi, Yi Chang, Bin GuICML 2024 · 2 citations
