Superquantile-Gibbs Relaxation for Minima-selection in Bilevel Optimization
arXiv:2505.05991
Abstract
Bilevel optimization (BLO) becomes more challenging when the lower-level objective admits multiple minimizers. Compared with the commonly studied unique-minimizer setting, this introduces two difficulties: (1) evaluating the hyper-objective requires minima selection over the lower-level solution set; and (2) may be discontinuous without additional structure. We address both issues under a parameter-uniform local Polyak-Lojasiewicz (PL) condition on the lower-level objective, denoted by . Unlike the global PL condition often assumed in BLO, the PL inequality in is imposed only near the local minima. This formulation accommodates bounded, non-singleton minimizer sets and is motivated by hyperparameter tuning in over-parameterized learning. We show that is Lipschitz continuous and that the lower-level minimizer sets are connected, compact, embedded submanifolds with a common intrinsic dimension . This intrinsic dimension determines the explicit accuracy exponents in our complexity bound. We propose a randomized method whose output satisfies an -Goldstein stationarity condition for in expectation. In the small-accuracy regime, suppressing logarithmic and fixed-problem factors, it uses at most queries to the gradient oracle of the lower-level objective, where is the upper-level dimension. Its key ingredient is a Superquantile-Gibbs relaxation that provides an approximate function-evaluation oracle for and turns minima selection into a sampling problem addressed by standard Langevin dynamics. To our knowledge, this is the first rigorous treatment of minima selection in BLO that explicitly relates overall complexity to the lower-level intrinsic dimension.