First-Order Bi-Level Optimization
- FOBLO is a family of bilevel optimization methods that use first-order and proximal information to replace explicit Hessian or KKT-system computations.
- It leverages fixed-point iterations, penalty reformulations, and smoothing techniques as surrogates for exact lower-level sensitivity, balancing efficiency and accuracy.
- Designed for applications like hyperparameter tuning, meta-learning, and constrained optimization, FOBLO methods also address challenges such as approximation bias and theoretical complexity.
First-Order Bi-Level Optimization (FOBLO) denotes a family of methods for bilevel programs that rely on first-order or proximal information, rather than explicit Hessian inverses, full KKT-system solves, or exact implicit differentiation through the lower-level solution map. In the literature represented here, the term covers several distinct but related regimes: classical hierarchical convex selection problems, nonconvex–strongly-convex hyperparameter-style bilevel programs, constrained and coupled lower-level problems, online bilevel optimization, stochastic bilevel approximation, and domain-specific extensions in meta-learning, pruning, speech adaptation, and reinforcement learning (Sabach et al., 2017, Kwon et al., 2023, Jiang et al., 2024, Jia et al., 12 Feb 2026, Zeng et al., 23 Jan 2026). A central unifying theme is that FOBLO replaces exact lower-level sensitivity computation by structural surrogates—fixed-point iterations, penalty formulations, primal-dual tracking, barrier smoothing, stochastic sampling, or first-order approximation—while preserving a meaningful link to the original bilevel objective.
1. Conceptual scope and problem classes
The canonical bilevel template in this literature is
but FOBLO is not restricted to this single form. One important branch studies hierarchical selection problems of the form
where the outer level selects a preferred point from a possibly nonunique lower-level solution set. The method in "A First Order Method for Solving Convex Bi-Level Optimization Problems" specializes Xu’s Sequential Averaging Method to this setting and emphasizes fixed-point selection over a convex composite lower problem rather than parameterized hypergradient computation (Sabach et al., 2017).
A second branch studies optimistic bilevel optimization without the lower-level singleton assumption. "A Generic First-Order Algorithmic Framework for Bi-Level Programming Beyond Lower-Level Singleton" reformulates the objective as
and argues that lower-level descent alone can converge to a lower-level minimizer that is poor for the upper-level objective when is set-valued (Liu et al., 2020). This perspective makes FOBLO not merely a matter of cheap hypergradients, but also of choosing the correct lower-level selector.
A third branch centers on modern hyperparameter-optimization-style formulations with a distinct upper variable and lower variable, often under strong convexity of the lower problem. "A Fully First-Order Method for Stochastic Bilevel Optimization" works in the unconstrained stochastic setting with unique induced by lower-level strong convexity (Kwon et al., 2023), while "Accelerated Fully First-Order Methods for Bilevel and Minimax Optimization" studies nonconvex–strongly-convex bilevel problems and the minimax special case obtained by setting (Li, 2024).
Constraint structure further enlarges the taxonomy. "A Primal-Dual-Assisted Penalty Approach to Bilevel Optimization with Coupled Constraints" treats lower-level feasible sets of the form
so the follower’s feasible region moves with the leader’s decision (Jiang et al., 2024). "A Single-Loop First-Order Algorithm for Linearly Constrained Bilevel Optimization" and "A Barrier-Metric First-Order Method for Linearly Constrained Bilevel Optimization" focus on linear inequality structure, respectively with coupled linear constraints and a fixed polyhedral lower set (Shen et al., 28 Oct 2025, Hong et al., 12 May 2026). "Fully First-Order Algorithms for Online Bilevel Optimization" adds temporal drift, replacing static optimality by local regret against time-varying bilevel objectives (Jia et al., 12 Feb 2026).
This variety matters because FOBLO is not a single algorithmic recipe. It is a design space organized by lower-level geometry, constraint structure, stochasticity, and the meaning assigned to “first-order.”
2. What “first-order” means in FOBLO
In the strongest oracle sense, a fully first-order method uses only gradients of the upper and lower objectives, projected or proximal steps, and simple averaging or primal-dual updates. This is the meaning adopted by F0SA, which uses only stochastic gradients of 1 and 2 and tracks two lower-level points 3 and 4 to approximate the gradient of a penalized surrogate 5, entirely avoiding Hessians, Jacobians, and Hessian-vector products (Kwon et al., 2023). The same oracle-level interpretation appears in RAF6BA/PRAF7BA, which optimize the surrogate
8
using only gradients of 9 and 0, plus accelerated gradient descent on lower subproblems (Li, 2024).
In constrained settings, “first-order” often means that the method avoids differentiating through the lower-level KKT system, even when extra structure such as multipliers or barrier geometry enters the updates. BLOCC, for example, is “fully first-order” because it uses gradients of 1, 2, and 3, projected primal-dual steps, and no Hessian inverses or Hessian-vector products, even though the correct value-function sensitivity must include the multiplier term
4
in the coupled-constraint case (Jiang et al., 2024). The barrier-metric method is similarly first-order in the sense that it uses only 5, together with the explicit logarithmic barrier Hessian induced by known linear constraints rather than second derivatives of 6 or 7 (Hong et al., 12 May 2026).
A looser meaning appears in approximate hypergradient methods. "A Stochastic Approach to Bi-Level Optimization for Hyperparameter Optimization and Meta Learning" introduces a “first-order approximation” inside a forward-mode-like recurrence through SGLD, but the method still uses Hessian-vector products such as
8
The paper therefore distinguishes computational first-order character from strict derivative-order purity (Kim et al., 2024). A related warning appears in "Debiasing a First-order Heuristic for Approximate Bi-level Optimization": the popular FOM heuristic has constant memory because it drops the second-derivative terms in the exact ABLO hypergradient, but that simplification induces bias and can prevent convergence to a stationary point of the ABLO objective (Likhosherstov et al., 2021).
This distinction suggests that FOBLO is best understood as a spectrum. At one end are methods that are first-order in oracle usage; at the other are scalable first-order approximations that still retain some second-order structure internally.
3. Principal algorithmic paradigms
FOBLO methods differ mainly in how they replace exact lower-level sensitivity.
| Paradigm | Representative methods | Core mechanism |
|---|---|---|
| Fixed-point and averaging | BiG-SAM, BDA | Select or aggregate lower-level solutions using nonexpansive maps or combined upper/lower descent |
| Penalty and primal-dual reformulation | F9SA, BLOCC, SFLCB, penalty methods | Replace the bilevel constraint by lower-level optimality-gap penalties and optimize a surrogate |
| Smoothing and geometry control | Moreau smoothing, barrier smoothing, stochastic Gibbs smoothing | Regularize the lower or outer problem to restore differentiability or robustness |
BiG-SAM is the cleanest fixed-point example. It rewrites the lower solution set as 0, where
1
and averages a lower proximal-gradient step with a contraction generated by the upper objective, producing a single-loop first-order scheme for convex hierarchical selection (Sabach et al., 2017). BDA generalizes the idea of biased lower-level motion: for fixed 2, it updates
3
so the lower iterate is simultaneously driven toward lower-level optimality and upper-level preference. This is the paper’s mechanism for handling non-singleton lower solutions without assuming 4 is single-valued (Liu et al., 2020).
Penalty reformulation is the dominant modern fully first-order strategy. In F5SA, the surrogate 6 approximates the true bilevel objective and yields an 7-gradient that depends only on 8, 9, and 0, where 1 tracks the minimizer of the penalized lower problem and 2 tracks the true lower optimum (Kwon et al., 2023). "On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximation" extends this idea to smooth constrained, generally nonconvex bilevel problems by defining
3
and then optimizing a smoothed min-max representation with proximal-envelope subproblems (Kwon et al., 2023). BLOCC uses a primal-dual-assisted penalty formulation for coupled constraints rather than joint projection onto 4, while SFLCB introduces slack variables and an augmented Lagrangian to obtain a single-loop primal-dual gradient method for linearly constrained lower levels (Jiang et al., 2024, Shen et al., 28 Oct 2025).
Smoothing plays several distinct roles. Moreau smoothing allows BiG-SAM to handle nonsmooth strongly convex outer objectives through 5 and the Moreau envelope 6 (Sabach et al., 2017). Barrier smoothing replaces a polyhedrally constrained lower problem by the interior objective 7, after which a proxy-gradient method operates in local Dikin geometry rather than Euclidean geometry (Hong et al., 12 May 2026). HPO-SGLD changes the underlying target even more radically by replacing the deterministic inner minimizer with a Gibbs distribution
8
so the outer problem becomes optimization of an expected validation loss under the induced inner distribution (Kim et al., 2024).
A final strand is approximation and debiasing. FOM in ABLO drops second-derivative terms entirely; UFOM restores unbiasedness by occasionally computing an exact correction with Bernoulli randomization, preserving constant memory in the inner-loop length 9 (Likhosherstov et al., 2021). In speech adaptation, MAdaPT-FOBLO adopts a Reptile-style parameter-difference approximation for a heterogeneous-loss bilevel objective with self-supervised inner adaptation and supervised outer evaluation (Luthra et al., 24 Dec 2025).
4. Theory: stationarity, approximation, and complexity
Theoretical guarantees in FOBLO are highly setting-dependent, and the performance criteria vary accordingly. In convex hierarchical selection, BiG-SAM proves convergence to the unique bilevel solution and an explicit 0 bound in the lower-level objective value,
1
but the paper is explicit that this rate is not stated for the outer objective gap or a full bilevel optimality gap (Sabach et al., 2017). In non-singleton optimistic bilevel optimization, BDA proves convergence of minimizers of the surrogate value functions 2 to minimizers of the optimistic value function 3, rather than a standard hypergradient stationarity statement (Liu et al., 2020).
For stochastic unconstrained bilevel optimization, F4SA gives finite-time convergence to an 5-stationary point of the true bilevel objective after 6 iterations when both levels are noisy, 7 when only the upper level is noisy, and 8 in the deterministic setting. Its momentum-assisted variant improves the stochastic bounds to 9 and 0, while retaining 1 deterministically (Kwon et al., 2023). RAF2BA/PRAF3BA further show that accelerated fully first-order complexity is possible in nonconvex–strongly-convex bilevel optimization, achieving 4 gradient-query complexity for approximate first-order and second-order stationarity, while the minimax specialization recovers 5 (Li, 2024).
Penalty and smoothing methods hinge on approximation theorems. The nonconvex penalty paper proves that 6 is 7-close to the original optimistic hyper-objective 8 in both value and derivative under proximal error-bound and local regularity assumptions, enabling the choice 9 when solving the surrogate (Kwon et al., 2023). The barrier-smoothed constrained method proves
0
then establishes stationarity rates for the smoothed objective 1: 2 deterministically and 3 with upper-level-only stochastic noise (Hong et al., 12 May 2026). SFLCB likewise quantifies closeness between 4 and the original constrained hyper-objective 5, including 6 under additional regularity, and improves prior double-loop complexity from 7 to 8 in its linearly constrained setting (Shen et al., 28 Oct 2025).
Coupled constraints introduce additional structural conditions. BLOCC proves convergence to approximate stationary points of its penalized reformulation using the projected gradient mapping 9, with total complexity 0 in the general nonlinear-constraint case and 1 when the lower-level constraints are affine in 2 and the dual function becomes globally strongly concave (Jiang et al., 2024). Online FOBLO replaces stationarity of a single objective by local regret:
3
F4OBO achieves 5 regret, and AF6OBO removes the dependence on lower-solution drift from the regret bound, attaining 7 (Jia et al., 12 Feb 2026).
A different theoretical caution comes from the ABLO debiasing paper. It proves an explicit FOM bias bound and constructs a family of examples where FOM-based SGD does not converge to a stationary point of the ABLO objective; UFOM restores unbiasedness while preserving constant memory in 8 (Likhosherstov et al., 2021). This is one of the clearest demonstrations that not all “first-order” bilevel heuristics optimize the intended objective.
5. Applications and empirical domains
FOBLO has been used across a strikingly broad range of applications. Hyperparameter optimization and meta-learning are recurrent motifs: BDA evaluates on data hyper-cleaning and few-shot learning benchmarks (Liu et al., 2020); F9SA studies MNIST data hyper-cleaning and reports competitive or superior practical behavior relative to second-order baselines (Kwon et al., 2023); HPO-SGLD targets hyperparameter optimization, loss-function learning, MAML-style few-shot learning, invariance learning, and scales to 87M hyperparameters in the ViT-B-16 L1-regularizer setting (Kim et al., 2024). Online bilevel optimization extends these concerns to drifting environments (Jia et al., 12 Feb 2026).
Constrained real-world planning problems are another major domain. BLOCC is motivated by lower-level coupled inequalities and is evaluated on support vector machine hyperparameter selection and transportation network design with real data from the city of Seville, avoiding joint projection onto coupled feasible sets that render prior approaches impractical at scale (Jiang et al., 2024). SFLCB studies both SVM hyperparameter optimization and transportation network design, reporting faster convergence or better upper-level utility than its baselines in those linearly constrained settings (Shen et al., 28 Oct 2025). The barrier-metric method targets fixed polyhedral lower constraints, which suggests applications involving allocations, flows, simplex or box constraints, and equilibrium-like lower problems, although its guarantees are stated for the barrier-smoothed surrogate rather than directly for the nonsmoothed bilevel objective (Hong et al., 12 May 2026).
Several papers show how FOBLO ideas migrate into domain-specific learning systems. "A Regularized Actor-Critic Algorithm for Bi-Level Reinforcement Learning" formulates reward-parameterized reinforcement learning as a bilevel problem and proposes a single-loop, first-order, penalty-based actor-critic method with attenuating entropy regularization, proving convergence to a stationary point of the original unregularized bilevel RL objective and illustrating the framework on a GridWorld goal-position problem and happy tweet generation through RLHF (Zeng et al., 23 Jan 2026). "SpidR-Adapt" casts low-resource speech adaptation as a meta-learning bilevel problem and uses a Reptile-style FOBLO approximation inside MAdaPT; empirically it reports rapid gains in phonemic discriminability and spoken language modeling, including improvement over in-domain LLMs after training on less than 1h of target-language audio and over 0 more data-efficient than standard training (Luthra et al., 24 Dec 2025). Model pruning is also cast as a bilevel problem in BiP, where the paper argues that a bi-linear pruning structure makes the resulting bilevel problem as easy as first-order optimization and reports 2–7 times speedup over IMP at matched accuracy and sparsity (Zhang et al., 2022).
Empirical results also illuminate the cost-structure claims that motivate FOBLO. BiG-SAM generally reaches stopping criteria much faster than the MNG baseline on inverse problems such as phillips, baart, and foxgood, which the paper attributes to MNG’s extra half-space optimization (Sabach et al., 2017). HPO-SGLD is reported to be more stable and less sensitive than IFT-based baselines in noisy or multimodal inner problems (Kim et al., 2024). These findings do not establish a uniform empirical ordering across all settings, but they do show that first-order surrogate design can offset weaker worst-case rates by reducing per-iteration cost and approximation fragility.
6. Limitations, misconceptions, and open directions
A common misconception is that FOBLO is synonymous with differentiating through a unique lower-level argmin map. The literature here shows otherwise. Some of the earliest and clearest first-order bilevel methods solve hierarchical selection problems by fixed-point averaging rather than implicit differentiation (Sabach et al., 2017). Other works treat set-valued lower solutions directly and demonstrate that LL-only descent can be fundamentally wrong without a lower-level singleton assumption (Liu et al., 2020). Constraint-sensitive methods show that once the feasible set depends on the upper variable, the correct first-order sensitivity may include dual multiplier terms, and naive unconstrained formulas can point in the wrong direction (Jiang et al., 2024).
A second misconception is that any first-order approximation is harmless. The ABLO debiasing paper proves that the FOM heuristic can remain arbitrarily far from stationarity of the ABLO objective, which makes bias analysis indispensable for approximate FOBLO (Likhosherstov et al., 2021). HPO-SGLD reinforces the taxonomic ambiguity: it is first-order-like in memory and oracle scaling, but not purely first-order in the strict derivative-order sense because it still uses Hessian-vector products (Kim et al., 2024). The term “first-order” therefore must always be read against the paper’s oracle model.
The dominant limitations remain structural. Many strongest results still require lower-level strong convexity, PL-type or proximal error-bound conditions, LICQ, strict complementarity, bounded multipliers, fixed or linearly parameterized constraints, or full-row-rank conditions on lower-level constraint matrices (Kwon et al., 2023, Kwon et al., 2023, Jiang et al., 2024, Shen et al., 28 Oct 2025). The barrier-metric method does not handle 1-dependent lower constraints, and its guarantees are for the smoothed surrogate 2 rather than directly for the original nonsmoothed objective (Hong et al., 12 May 2026). The bilevel RL method assumes finite state-action spaces, tabular softmax policies in theory, a special regularization-dependent PL condition, and yields a slow 3 sample complexity for the original unregularized problem (Zeng et al., 23 Jan 2026).
Open directions are correspondingly clear in the cited work. BLOCC identifies relaxing lower-level strong convexity, removing local dual curvature assumptions, and designing more practical single-loop variants for general coupled constraints as natural next steps (Jiang et al., 2024). The RL paper notes that extending the analysis to 4-dependent transition kernels is nontrivial because 5 becomes harder to derive and estimate (Zeng et al., 23 Jan 2026). The broader literature represented here suggests a continuing tension between three goals: wider modeling scope, oracle purity, and sharp complexity. FOBLO advances whenever a method pushes one of these frontiers without collapsing the others.