Learning Theory for Kernel Bilevel Optimization
Fares El Khoury, Edouard Pauwels, Samuel Vaiter, Michael Arbel
Abstract
Bilevel optimization has emerged as a technique for addressing a wide range of machine learning problems that involve an outer objective implicitly determined by the minimizer of an inner problem. While prior works have primarily focused on the parametric setting, a learning-theoretic foundation for bilevel optimization in the nonparametric case remains relatively unexplored. In this paper, we take a first step toward bridging this gap by studying Kernel Bilevel Optimization (KBO), where the inner objective is optimized over a reproducing kernel Hilbert space. This setting enables rich function approximation while providing a foundation for rigorous theoretical analysis. In this context, we derive novel finite-sample generalization bounds for KBO, leveraging tools from empirical process theory. These bounds further allow us to assess the statistical accuracy of gradient-based methods applied to the empirical discretization of KBO. We numerically illustrate our theoretical findings on a synthetic instrumental variable regression task.
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.
Builds on17
- Efficient and Modular Implicit DifferentiationMathieu Blondel, Quentin Berthet, Marco Cuturi, Roy Frostig et al.NeurIPS 2022 · 386 citations
- Bilevel Optimization: Convergence Analysis and Enhanced DesignKaiyi Ji, Junjie Yang, Yingbin LiangICML 2021 · 343 citations
- Closing the Gap: Tighter Analysis of Alternating Stochastic Gradient Methods for Bilevel ProblemsTianyi Chen, Yuejiao Sun, Wotao YinNeurIPS 2021 · 176 citations
- A framework for bilevel optimization that enables stochastic and global variance reduction algorithmsMathieu Dagréou, Pierre Ablin, Samuel Vaiter, Thomas MoreauNeurIPS 2022 · 149 citations
- Towards Gradient-based Bilevel Optimization with Non-convex Followers and BeyondRisheng Liu, Yaohua Liu, Shangzhi Zeng, Jin ZhangNeurIPS 2021 · 111 citations
Related papers
- Functional Bilevel Optimization for Machine LearningIeva Petrulionyte, Julien Mairal, Michael ArbelNeurIPS 2024 · 27 citations
- Enhanced Bilevel Optimization via Bregman DistanceFeihu Huang, Junyi Li, Shangqian Gao, Heng HuangNeurIPS 2022 · 41 citations
- Differentially Private Bilevel Optimization: Efficient Algorithms with Near-Optimal RatesAndrew Lowy, Daogao LiuNeurIPS 2025 · 3 citations
- Non-Convex Bilevel Optimization with Time-Varying Objective FunctionsSen Lin, Daouda Sow, Kaiyi Ji, Yingbin Liang et al.NeurIPS 2023 · 11 citations
- Distributed Learning of Conditional Quantiles in the Reproducing Kernel Hilbert SpaceHeng LianNeurIPS 2022 · 11 citations
