Adaptive Bilevel Search (ABS)
- Adaptive Bilevel Search (ABS) is a collection of adaptive bilevel optimization frameworks that coordinate an outer variable with inner problem solutions.
- It integrates multiple strategies such as curvature-aware updates, first-order methods under PŁ conditions, and fragmentation-aware guidance across diverse applications.
- ABS optimizes complex tasks in neural architecture search, hyperparameter learning, and computing networks by adaptively managing descent directions, step sizes, and solution accuracy.
Adaptive Bilevel Search (ABS) designates a set of adaptive bilevel optimization frameworks rather than a single standardized algorithm. In recent arXiv literature, the label is used for an auxiliary-program approach to differentiable neural architecture search, for adaptive fully first-order algorithms for nonconvex bilevel optimization under Polyak–Łojasiewicz conditions, for an adaptively inexact first-order method for hyperparameter learning, and for a fragmentation-aware bilevel service-mapping framework in Computing Power Networks. In each usage, an outer search variable is coordinated with inner decisions or inner solves, and the adaptation operates through coupled descent directions, step-size control, accuracy control, search-support reduction, or fragmentation-aware global guidance (Shukla et al., 28 Jun 2026, Shi et al., 30 Dec 2025, Salehi et al., 2023, Xie et al., 10 Jul 2025).
1. Terminological scope and research usage
The term ABS is used across several technical contexts, and the label does not identify a universal operator set. One ABS family is curvature-aware and second-order; one is fully first-order and tuning-free under PŁ assumptions; one uses backtracking line search with inexact oracles; and one is a distributed particle-swarm framework for graph-structured service mapping. The shared feature is a bilevel architecture in which upper-level progress is made only in conjunction with lower-level structure or lower-level feasibility.
| ABS usage | Outer / inner structure | Adaptive components |
|---|---|---|
| AP-NAS | for architecture, for weights | curvature-aware coupling, trust-region or line-search, damping |
| AFBA / AFBA | for outer variable, for lower-level and penalty subproblems | AdaGrad-Norm or AC-GM, fully first-order stepsizes |
| MAID | or for hyperparameters, for inner state | backtracking line search, dynamic 0 control |
| CPN ABS | 1 for co-location proportions, local PW-kGPP and IMCF decisions | elite-guided PSO, dynamic top-2 masking, fragmentation-aware fitness |
This multiplicity matters methodologically. In neural architecture search and hyperparameter learning, ABS is a bilevel optimization method for continuous outer variables. In Computing Power Networks, ABS is a bilevel search framework over proportion vectors whose induced lower-level solutions are graph partitions and routing plans. The common terminology therefore indicates adaptive outer–inner coordination, not a single convergence theory or implementation template.
2. Mathematical structure of bilevel search
The canonical bilevel problem in the cited literature is
3
with the corresponding hyper-objective
4
The stationarity notion used in the PŁ-based work is standard: 5 is an 6-stationary point of a differentiable function 7 if 8 (Shi et al., 30 Dec 2025). In differentiable NAS, the same hierarchy appears as
9
where 0 are continuous architecture parameters and 1 are network weights (Shukla et al., 28 Jun 2026).
When the lower level is sufficiently regular, implicit differentiation yields a hypergradient. For NAS, near a lower-level optimum,
2
The PŁ-based theory gives parallel formulas for 3, including a penalty-regularized hyper-objective 4 for non-strongly convex lower levels, and an explicit expression with the Moore–Penrose inverse of 5 (Shi et al., 30 Dec 2025). In the hyperparameter-learning setting, strong convexity of the inner problem in 6 permits the classical implicit-differentiation formula
7
together with computable inexact hypergradient oracles (Salehi et al., 2023).
These formulations separate ABS variants by how they handle the outer–inner coupling. Some preserve lower-level optimality explicitly through coupled directions; some approximate the hypergradient with only first-order information; and some accept inexact inner solves but regulate their errors so that outer descent remains certified.
3. ABS in neural architecture search
In "Bilevel Optimization for Neural Architecture Search" (Shukla et al., 28 Jun 2026), ABS denotes the auxiliary-program approach within Auxiliary Program-based NAS (AP-NAS), with LP-DARTS and SOCP-DARTS as concrete instantiations. The central idea is to replace a hypergradient-only update of 8 followed by retraining of 9 with a joint descent direction 0 that decreases validation loss to first order while preserving lower-level optimality to first order. At a reference point 1 with 2, the lower-level stationarity linearization yields
3
where 4 and 5. ABS then solves the auxiliary program
6
or an LP relaxation with 7. Eliminating 8 gives the best-response coupling
9
The search is adaptive in three explicit senses. First, it is curvature-aware: 0 follows the locally correct first-order manifold of inner optima through 1. Second, step sizes are chosen by line-search along 2 or via trust-region acceptance tests. Third, reduced Hessians, L-BFGS or rank-limited approximations, and damping 3 are used to manage nonconvex curvature and memory. PyTorch Hessian–vector products, conjugate gradient solves, and warm-started restoration steps are the practical mechanism by which these updates are realized.
The AP-NAS paper positions ABS against both sampling-based NAS and gradient-based DARTS-style methods. Sampling-based methods are described as less sample-efficient because they evaluate more architectures, whereas ABS uses gradients and second-order inner curvature. Relative to DARTS, ABS enforces the linearized KKT stationarity of the inner problem and updates 4 simultaneously rather than updating only 5 and then retraining 6. The consolidated comparisons report that differentiable NAS drastically reduces search cost relative to RL and evolutionary methods, with DARTS at approximately 4 GPU days versus NASNet and AmoebaNet in the thousands, and that under comparable CPU-hour budgets LP-DARTS and SOCP-DARTS outperform standard DARTS on CIFAR-10 and transfer better to ImageNet. The reported figures are DARTS: CIFAR-10 82.57%, ImageNet Top-1 39.09% (5-top1 64.10%), Params 3.38M; LP-DARTS: CIFAR-10 92.56%, ImageNet Top-1 42.50% (67.55%), Params 1.66M; SOCP-DARTS: CIFAR-10 90.56%, ImageNet Top-1 44.02% (68.76%), Params 2.23M. The same auxiliary program is also extended to simultaneous hyperparameter tuning and fine-tuning, including the LiFT transformer fine-tuning setting.
4. Fully first-order adaptive ABS under PŁ conditions
"Adaptive Algorithms for Nonconvex Bilevel Optimization under PŁ Conditions" (Shi et al., 30 Dec 2025) treats ABS as a fully first-order adaptive pipeline instantiated by AF7BA and its accelerated variant A8F9BA. The lower-level function 0 and the penalty function 1 are assumed to satisfy a PŁ condition with respect to 2, and the analysis also assumes Lipschitz gradients and Lipschitz Hessians for 3 and 4, a lower bound on 5, and a sufficiently small choice of 6, set in the paper to 7. The algorithms solve two inner subproblems per outer iteration: a lower-level search for 8 approximating 9, and a penalty search for 0 approximating 1. They then form the fully first-order hypergradient estimator
2
which avoids Jacobian-vector and Hessian-vector products entirely.
Adaptivity enters at both inner and outer levels. AF3BA uses AdaGrad-Norm in the inner searches and in the outer update
4
A5F6BA accelerates the inner loops with AC-GM, an auto-conditioned gradient method that is linesearch-free and uses evolving curvature surrogates 7 and a scale parameter 8. The paper proves linear convergence for AC-GM under the PŁ regime, with the optimal choice stated as 9, and uses this to reduce the inner iteration counts to 0.
The complexity guarantees are explicit. Both algorithms achieve 1 outer iteration complexity for finding an 2-stationary point of the hyper-objective. AF3BA has 4 first-order oracle complexity because its inner iteration counts scale as 5, whereas A6F7BA achieves 8 first-order complexity, matching the best-known tuned first-order methods up to logarithmic factors. The work emphasizes that no prior knowledge of Lipschitz or PŁ constants is required, but it also states several scope conditions: the analysis is deterministic, depends on PŁ and smoothness assumptions, and does not include empirical experiments.
5. Adaptively inexact ABS for hyperparameter learning
"An adaptively inexact first-order method for bilevel optimization with application to hyperparameter learning" (Salehi et al., 2023) presents MAID, identified there as an instantiation of ABS. The setting is bilevel learning with inexact function values and inexact hypergradients, motivated by imaging, inverse problems, and multinomial logistic regression. The reduced outer objective is
9
and, in the paper’s notation, 0 is the hyperparameter vector and 1 is the inner minimizer of 2. The algorithm solves the inner problem only approximately, solves the associated linear system only approximately, and forms an inexact hypergradient 3. A computable a posteriori bound on the error 4 links inner tolerances 5 to outer hypergradient accuracy.
The adaptive mechanism has two parts. First, MAID enforces a descent-direction condition
6
so that 7 is guaranteed to be a descent direction for the true outer objective. If the condition fails, the algorithm tightens the inner tolerances by 8 and 9 and recomputes the inexact hypergradient. Second, MAID performs backtracking line search with an inexact Armijo test based on computable upper and lower bounds for the exact objective value. The sufficient-decrease rule is checked through a quantity 0 built from 1, gradient norms, the inner accuracy 2, and the term 3. If no trial step is accepted within the current backtracking budget, the algorithm again tightens 4 and retries. This realizes the principle stated in the paper: inner work is spent only when needed to certify descent or to accept a step.
The convergence theorem establishes finite termination of the line search, a sufficient-decrease inequality,
5
a uniform positive lower bound on accepted step sizes, summability of 6, and therefore 7. The empirical section reports three classes of experiments. On a synthetic quadratic bilevel problem, MAID adaptively drove 8 to approximately 9 regardless of starting accuracy and reached lower loss with less total inner work than any fixed high-accuracy run. On multinomial logistic regression for MNIST with 1440 hyperparameters, MAID achieved lower validation loss than HOAG under the same total inner budget, and the a posteriori 00 sequence could be fit by an approximately summable schedule 01 that MAID discovered automatically. In variational image denoising, MAID reached training loss approximately 02 for smoothed total variation versus approximately 03–04 for DFO-LS with dynamic accuracy, with test PSNR approximately 05 dB versus 06 dB, and for convex field-of-experts it achieved approximately 07 dB PSNR versus approximately 08 dB for smoothed TV learned by MAID on the same data.
6. Fragmentation-aware ABS for service mapping in Computing Power Networks
"A Fragmentation-Aware Adaptive Bilevel Search Framework for Service Mapping in Computing Power Networks" (Xie et al., 10 Jul 2025) uses ABS in a different application domain. Here the substrate is a Computing Power Network graph 09 with compute capacities on nodes and bandwidth capacities on links, and each service request is a graph 10 whose service functions have CPU demands and whose logical links have bandwidth demands. The joint decision problem includes binary placement variables 11 and routing selectors 12 under exact-one-placement, compute-capacity, endpoint–tunnel coherence, and network-bandwidth constraints. For a fixed arriving request, the operational objective reduces to minimizing external bandwidth cost on the current residual network. The paper proves NP-hardness of this per-request problem through a polynomial-time reduction from the Graph Bisection Problem.
ABS reformulates the placement component through continuous co-location proportions 13 over substrate nodes. For fixed 14, service-function mapping becomes a Proportional Weight-constrained 15-way Graph Partitioning Problem, and routing of the resulting Cut-LLs becomes an Integer Multi-Commodity Flow instance over precomputed path sets. The framework is therefore bilevel: the upper level searches over 16, while the lower level solves PW-kGPP, then IMCF, for each candidate. The paper states that if the upper level could exhaustively enumerate all feasible 17 and the lower level solved PW-kGPP and IMCF to optimality, the nested approach would recover the optimum of the per-request subproblem.
The search guidance is fragmentation-aware rather than gradient-based. For a candidate mapping 18, ABS computes Node Resource Exhaustion Degree (NRED), Computing-to-Bandwidth Utilization Gap (CBUG), and Path Node Valuelessness Level (PNVL), then minimizes
19
The reported cumulative-distribution analyses indicate that NRED correlates strongest with global performance, then CBUG, then PNVL. The optimizer is a distributed elite-guided PSO. Particles represent continuous 20 vectors, negative entries are clipped to zero, dynamic top-21 masking retains the most promising support and normalizes it to sum to one, and successful feasibility can shrink the active support for subsequent iterations. Workers run local swarms asynchronously, maintain elite and common subsets and local archives, and communicate best particles to a controller holding a bounded global archive.
The experimental setting includes a Waxman topology with 100 nodes and 500 links, Rocketfuel AS6461 with 129 nodes and 363 links, capacities uniformly in 22, and 2000 service entities per run with 50–100 service functions and edge density approximately 23. Against heuristics, metaheuristics, and learning-based baselines adapted to allow co-location, ABS and ABS initialized with RW-BFS dominate across both topologies and metrics. The headline results include acceptance ratio 24 versus 25 on the random topology and 26 versus 27 on Rocketfuel; revenue 28M versus 29M on the random topology and 30M versus 31M on Rocketfuel; long-term average revenue 32 versus 33 on the random topology and 34 versus 35 on Rocketfuel; and profit 36M versus 37M on the random topology and 38 versus 39 on Rocketfuel. The paper emphasizes that ABS improves compute utilization by at least 40 on the random topology and 41 on Rocketfuel over the best baseline, and that its service acceptance ratio is 42 higher than the best baseline in the complex Rocketfuel scenario. The reported limitations are equally explicit: the implementation optimizes CPU and bandwidth only, assumes accurate instantaneous capacities and stationary request statistics within a run, relies on heuristic lower-level solvers, and adds control-plane and computation overhead relative to simple heuristics.