paper-with-me

홈 › Papers

Tuning-Free Bilevel Optimization: New Algorithms and Convergence Analysis

2024-10-07 · Yifan Yang, Hao Ban, Minhui Huang, Shiqian Ma, Kaiyi Ji

Bilevel optimization has recently attracted considerable attention due to its abundant applications in machine learning problems. However, existing methods rely on prior knowledge of problem parameters to determine stepsizes, resulting in significant effort in tuning stepsizes when these parameters are unknown. In this paper, we propose two novel tuning-free algorithms, D-TFBO and S-TFBO. D-TFBO employs a double-loop structure with stepsizes adaptively adjusted by the "inverse of cumulative gradient norms" strategy. S-TFBO features a simpler fully single-loop structure that updates three variables simultaneously with a theory-motivated joint design of adaptive stepsizes for all variables. We provide a comprehensive convergence analysis for both algorithms and show that D-TFBO and S-TFBO respectively require $O(\frac{1}{\epsilon})$ and $O(\frac{1}{\epsilon}\log^4(\frac{1}{\epsilon}))$ iterations to find an $\epsilon$-accurate stationary point, (nearly) matching their well-tuned counterparts using the information of problem parameters. Experiments on various problems show that our methods achieve performance comparable to existing well-tuned approaches, while being more robust to the selection of initial stepsizes. To the best of our knowledge, our methods are the first to completely eliminate the need for stepsize tuning, while achieving theoretical guarantees.

📄 PDF Abstract BibTeX arXiv:2410.05140

Code (1)

optmn-lab/tfbo 공식 구현 jax

Tasks

Bilevel Optimization

Methods 이 논문이 사용한 방법론

Softmax The Softmax output function transforms a previous layer's output into a vector of probabilities. It is commonly used for multiclass classification. Given an input vector $x$…
Attention 설명 없음

Similar Papers 제목 키워드 기반

On the Convergence Theory for Hessian-Free Bilevel Algorithms

2021-10-13 · Daouda Sow, Kaiyi Ji, Yingbin Liang

Bilevel optimization has arisen as a powerful tool in modern machine learning. However, due to the nested structure of bilevel optimization, even gradient-based methods require second-order derivative approximations via …

Bilevel OptimizationMeta-Learning

Optimal Hessian/Jacobian-Free Nonconvex-PL Bilevel Optimization

2024-07-25 · Feihu Huang

Bilevel optimization is widely applied in many machine learning tasks such as hyper-parameter learning, meta learning and reinforcement learning. Although many algorithms recently have been developed to solve the bilevel…

Bilevel OptimizationMeta-LearningRepresentation Learning

Bilevel Optimization for Machine Learning: Algorithm Design and Convergence Analysis

2021-07-31 · Kaiyi Ji

Bilevel optimization has become a powerful framework in various machine learning applications including meta-learning, hyperparameter optimization, and network architecture search. There are generally two classes of bile…

BIG-bench Machine LearningBilevel OptimizationHyperparameter OptimizationMeta-Learning

Escaping the Variance Trap: Jacobian-Free Dynamics for Root-Finding Bilevel Optimization

2026-06-21 · Zhiyu Li, Xi Xuan, Davide Carbone arxiv

Many central machine learning tasks, from entropy tuning in reinforcement learning to equilibrating generative adversarial networks, are fundamentally stochastic root-finding problems rather than loss minimization. Yet, …

Reinforcement LearningBilevel Optimization

Bilevel ZOFO: Bridging Parameter-Efficient and Zeroth-Order Techniques for Efficient LLM Fine-Tuning and Meta-Training

2025-02-05 · Reza Shirkavand, Qi He, Peiran Yu, Heng Huang

Fine-tuning pre-trained Large Language Models (LLMs) for downstream tasks using First-Order (FO) optimizers presents significant computational challenges. Parameter-Efficient Fine-Tuning(PEFT) methods have been proposed …

Bilevel Optimizationparameter-efficient fine-tuning