Federated Incremental Subgradient Method for Convex Bilevel Optimization Problems
Sudkobfa Boontawee, Mootta Prangprakhon, Nimit Nimana
TL;DR
This work tackles convex bilevel optimization in a federated setting, where the outer objective $H(x)$ is strongly convex and the inner objective is a finite-sum across clients. It introduces the Federated Incremental Subgradient Method (FISM), which blends iterative regularization with a federated update protocol, sharing only a subgradient $\mathcal{H}_k$ of $H$ to preserve privacy and reduce local computation. The authors prove convergence of the iterates to the unique outer-optimal $x_H^*$ and establish a near $1/K^{0.5}$ convergence rate for the inner-level objective value, under standard assumptions and suitably chosen step sizes. Empirical results on binary MNIST classification and a location problem demonstrate that FISM, especially with more groups $S$, achieves faster convergence and better accuracy than the IR-IG baseline, highlighting the method’s scalability and privacy benefits in distributed optimization settings. The work also discusses practical considerations and future directions, including asynchronous communication and extensions to nonconvex problems.
Abstract
In this letter, we consider a bilevel optimization problem in which the outer-level objective function is strongly convex, whereas the inner-level problem consists of a finite sum of convex functions. Bilevel optimization problems arise in situations where the inner-level problem does not have a unique solution. This has led to the idea of introducing an outer-level objective function to select a solution with the specific desired properties. We propose an iterative method that combines an incremental algorithm with a broadcast algorithm, both based on the principles of federated learning. Under appropriate assumptions, we establish the convergence results of the proposed algorithm. To demonstrate its performance, we present two numerical examples related to binary classification and a location problem.
