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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| ODE-Driven Control | ODE-Driven Control under Non-Linearity | Final Error Magnitude (|h|)0.001 | 6 | |
| Multi-Agent Nash Equilibrium | GridWorld | Final Magnitude |h|0.006 | 6 | |
| Image Classification | CIFAR-10 (test) | Top-1 Accuracy74.72 | 5 | |
| Bilevel optimization | Synthetic RF-BO | Average Time per Iteration (ms)0.02 | 4 | |
| Root-Finding Bilevel Optimization | Root-Finding Bilevel Optimization (RF-BO) (theoretical analysis) | Sample Complexity-2 | 4 | |
| SAC Temperature Tuning | Pendulum v1 | Return @ 5k Steps-996 | 3 |