Stability of Primal-Dual Gradient Flow Dynamics for Multi-Block Convex Optimization Problems
Ibrahim K. Ozaslan, Panagiotis Patrinos, Mihailo R. Jovanović
TL;DR
The paper develops primal-dual gradient flow dynamics built on a proximal augmented Lagrangian to solve multi-block convex optimization with generalized consensus constraints, including nondifferentiable terms. It establishes GAS under feasibility/convexity, LES under PL and polyhedral/group-sparse nonsmooth terms, and GES under a range-space condition combined with strong primal convexity via the augmented Lagrangian, using novel Lyapunov constructions V1, V2, and their sum V3. The analysis leverages LaSalle's invariance principle, Hoffman-type error bounds, and a differentiable dual gradient to obtain global exponential rates without requiring strong convexity of the entire objective; it also demonstrates the necessity of the Lipschitz assumption for GES. The framework supports distributed implementations (via network Laplacians and neighbor communications) and yields exponential convergence in broad settings, with several illustrative examples including decentralized lasso, robust PCA, covariance completion, and sparse group lasso. Overall, the work provides a systematic, broadly applicable alternative to ADMM for large-scale multi-block problems, offering explicit exponential convergence guarantees under relatively mild structural conditions and enabling parallel/distributed computation.
Abstract
We examine stability properties of primal-dual gradient flow dynamics for composite convex optimization problems with multiple, possibly nonsmooth, terms in the objective function under the generalized consensus constraint. The proposed dynamics are based on the proximal augmented Lagrangian and they provide a viable alternative to ADMM which faces significant challenges from both analysis and implementation viewpoints in large-scale multi-block scenarios. In contrast to customized algorithms with individualized convergence guarantees, we develop a systematic approach for solving a broad class of challenging composite optimization problems. We leverage various structural properties to establish global (exponential) convergence guarantees for the proposed dynamics. Our assumptions are much weaker than those required to prove (exponential) stability of primal-dual dynamics as well as (linear) convergence of discrete-time methods such as standard two-block and multi-block ADMM and EXTRA algorithms. Finally, we show necessity of some of our structural assumptions for exponential stability and provide computational experiments to demonstrate the convenience of the proposed approach for parallel and distributed computing applications.
