Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

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, they are frequently forced into a minimization framework via squared residuals, introducing a critical flaw we identify as the Variance Trap. Standard bilevel minimization algorithms require estimating hypergradients involving implicit Jacobians; in stochastic settings, these terms act as noise amplifiers, destabilizing convergence. We formalize Root-Finding Bilevel Optimization (RF-BO) as a distinct problem class that bypasses this pathology. We propose a Jacobian-free solution using Two-Time-Scale Stochastic Approximation (TTSA) that updates directly along the root error, structurally avoiding variance amplification. We provide the first non-asymptotic convergence guarantees for TTSA in this setting under Markovian noise. Extensive experiments demonstrate the decisive advantage of this paradigm: compared to squared-residual and implicit-gradient baselines, our framework achieves a 2.6\% top-1 accuracy gain in SimCLR, 17$\times$ faster convergence in non-linear ODE control where baselines fail, significantly improved entropy stability in reinforcement learning, and an 11.1\% quality improvement in generative modeling.

Zhiyu Li, Xi Xuan, Davide Carbone• 2026

Related benchmarks

TaskDatasetResultRank
ODE-Driven ControlODE-Driven Control under Non-Linearity
Final Error Magnitude (|h|)0.001
6
Multi-Agent Nash EquilibriumGridWorld
Final Magnitude |h|0.006
6
Image ClassificationCIFAR-10 (test)
Top-1 Accuracy74.72
5
Bilevel optimizationSynthetic RF-BO
Average Time per Iteration (ms)0.02
4
Root-Finding Bilevel OptimizationRoot-Finding Bilevel Optimization (RF-BO) (theoretical analysis)
Sample Complexity-2
4
SAC Temperature TuningPendulum v1
Return @ 5k Steps-996
3
Showing 6 of 6 rows

Other info

Follow for update