JMLR

Decentralized Bilevel Optimization: A Perspective from Transient Iteration Complexity

Authors
Xinmeng Huang Kun Yuan Boao Kong Shuchen Zhu Songtao Lu
Research Topics
Computational Statistics
Paper Information
  • Journal:
    Journal of Machine Learning Research
  • Added to Tracker:
    Dec 30, 2025
Abstract

Stochastic bilevel optimization (SBO) is becoming increasingly essential in machine learning due to its versatility in handling nested structures. To address large-scale SBO, decentralized approaches have emerged as effective paradigms in which nodes communicate with immediate neighbors without a central server, thereby improving communication efficiency and enhancing algorithmic robustness. However, most decentralized SBO algorithms focus solely on asymptotic convergence rates, overlooking transient iteration complexity-the number of iterations required before asymptotic rates dominate, which results in limited understanding of the influence of network topology, data heterogeneity, and the nested bilevel algorithmic structures. To address this issue, this paper introduces D-SOBA, a Decentralized Stochastic One-loop Bilevel Algorithm framework. D-SOBA comprises two variants: D-SOBA-SO, which incorporates second-order Hessian and Jacobian matrices, and D-SOBA-FO, which relies entirely on first-order gradients. We provide a comprehensive non-asymptotic convergence analysis and establish the transient iteration complexity of D-SOBA. This provides the first theoretical understanding of how network topology, data heterogeneity, and nested bilevel structures influence decentralized SBO. Extensive experimental results demonstrate the efficiency and theoretical advantages of D-SOBA.

Author Details
Xinmeng Huang
Author
Kun Yuan
Author
Boao Kong
Author
Shuchen Zhu
Author
Songtao Lu
Author
Research Topics & Keywords
Computational Statistics
Research Area
Citation Information
APA Format
Xinmeng Huang , Kun Yuan , Boao Kong , Shuchen Zhu & Songtao Lu . Decentralized Bilevel Optimization: A Perspective from Transient Iteration Complexity. Journal of Machine Learning Research .
BibTeX Format
@article{paper690,
  title = { Decentralized Bilevel Optimization: A Perspective from Transient Iteration Complexity },
  author = { Xinmeng Huang and Kun Yuan and Boao Kong and Shuchen Zhu and Songtao Lu },
  journal = { Journal of Machine Learning Research },
  url = { https://www.jmlr.org/papers/v26/25-0677.html }
}