Distributed Optimization Based on Gradient Tracking Revisited: Enhancing Convergence Rate via Surrogation 1: SONATA Converges R-Linearly over Undirected Graphs for Small Step SizesResearch Paper
Motivation
In distributed optimization, agents connected by a communication network jointly minimize the average of their private losses, exchanging vectors only with their neighbours. This is the computational model of decentralized machine learning (data held by many machines, no central server), of sensor networks and of multi-agent control. A central family of methods is gradient tracking: each agent keeps, besides its copy of the decision variable, an estimate of the network-wide average gradient, refreshed by mixing with neighbours. Gradient tracking removes the steady-state bias of plain decentralized gradient descent and, for smooth strongly convex problems, converges linearly (Nedić, Olshevsky & Shi 2017; Qu & Li 2018).
Sun, Daneshmand and Scutari (arXiv:1905.02637v2, SIAM J. Optim. 2022) study SONATA, a gradient-tracking scheme in which each agent replaces its loss by a strongly convex surrogate and solves a small local problem per round. The setting is broader than earlier linear-rate results in three ways: the problem may be composite (a convex, possibly nonsmooth regularizer ), constrained (a closed convex set ), and only the average loss needs to be strongly convex, not each local loss. Linearizing surrogates recover proximal gradient tracking; surrogates that keep the full local loss recover DANE- and CEASE-type methods, whose rates improve when the local losses are similar.
Setting
Agents hold functions . With , Problem (P) is
Assumption A: is nonempty, closed and convex; each is twice differentiable on an open set and convex; on with ; is convex on . Then on for some constants (display (1)), and .
The network is a connected undirected graph on the agents (Assumption B) and a doubly stochastic weight matrix with , on edges and off edges (Assumption D). Its mixing rate is , the largest singular value of (tensored with ).
Each agent has a surrogate , on , whose gradient in matches at , is -Lipschitz, and which is -strongly convex in (Assumption C). Display (15) supplies constants with on ; , , , .
SONATA (Algorithm 1) with step size starts from , and repeats
With the optimal solution, the analysis tracks the optimality gap , the consensus errors , (deviations from the network averages), the tracking error and the step .
Formalization targets
Goal: Theorem 3.9
If , there is such that for every , every SONATA run and every agent ,
The goal fixes no rate: it asserts R-linear convergence for all small step sizes, which is the shape the paper proves. The explicit rate of Theorem 3.10 and the iteration counts of Corollaries 3.11–3.14 are not posed.
Milestones
In the order of the proof: the tracking invariant (19); Lemma 3.1 (per-agent descent); Lemma 3.2 (a lower bound on ); Lemma 3.3 (); Proposition 3.4 (); Proposition 3.5 (consensus errors contract at rate up to ); Proposition 3.6 ( bounded by and ); Proposition 3.8 (the four inequalities (50a)–(50d) for the transformed sequences ); and (54), .
Significance
Theorem 3.9 gives a linear rate for distributed composite, constrained optimization with gradient tracking, under strong convexity of the average loss only. It covers, through the choice of surrogate, proximal gradient tracking and the distributed versions of DANE and CEASE, and the constants quantify how much a surrogate gains when local losses are similar. The chain of Propositions 3.4–3.8 is a reusable template: an optimality-gap recursion, consensus recursions and a step bound, closed by a small-gain argument.
The result is proved in the paper; to our knowledge no part of it has been formalized. The mission produces a machine-checked statement of the model and of the full chain. Useful contributions include proofs of the individual milestones, a proof of the goal from the milestones, and the general facts they rest on.
Difficulty
The obvious argument fails in two places. First, the local step does not descend on by itself: the tracker is not , and the surrogate's curvature differs from 's, so Lemma 3.1 must compare Hessians of and along a segment through (15), and the descent holds only up to the tracking error. Second, no single quantity contracts: contracts up to consensus errors, consensus errors contract up to , and is bounded back by . No Lyapunov function is given, and the loop gain of (55) involves every constant of the problem, so showing it can be pushed below 1 for some is where the step-size threshold comes from. Also, is not an assumption: it must be derived from connectivity and Assumption D.
Formalization scope
Agents' vectors live in EuclideanSpace ℝ (Fin d), stacked vectors are Fin m → EuclideanSpace ℝ (Fin d), and their norm is always the Euclidean norm written out (), never Lean's sup norm. Hessians are derivatives of gradients; Loewner bounds are quadratic-form inequalities. The surrogate's gradient and Hessian are in its first argument. is the operator norm of on Euclidean . Time is 0-based.
A SONATA run is a relation: is any minimizer of the local problem, and any minimizer of on . Assumption A gives no lower semicontinuity of , so minimizers need not exist in general; the paper assumes they do, and so does the formalization, without adding a hypothesis on . A sanity file exhibits instances of all assumptions and of runs (including a nonconstant one), so the run predicate is not empty. " convex" is read as convexity on plus a positive semidefinite Hessian on ; is convex on only. The constants of (1) and (15) are parameters: each statement holds for every valid choice. All of Assumptions A–D are hypotheses of every statement (the paper's standing assumption of §3.3).
Trivializing formalizations are ruled out: is not assumed in place of Assumptions B and D; is chosen before the step size, the starting point and the run; is required; norms of stacked vectors are Euclidean. Two printed slips are handled openly: in Proposition 3.4 is stated with positivity only when or (at , (41) gives ), and is kept as printed in (52d) although (47) has 3. In Lemma 3.2, Lean's makes the bound trivial when .
A full development needs: the mean-value and Taylor estimates for functions with bounded Hessian on a convex set, first-order optimality for constrained convex minimization, the spectral bound for graph-compliant doubly stochastic matrices (posed elsewhere on the platform as StochGradTrack.Const.lemma_1 for a closely related model), and elementary estimates on finite maxima. These are reusable beyond this mission.
Selected references
- Y. Sun, A. Daneshmand, G. Scutari, Distributed Optimization Based on Gradient-tracking Revisited: Enhancing Convergence Rate via Surrogation, arXiv:1905.02637v2, 2020; SIAM J. Optim. 32 (2022). https://arxiv.org/abs/1905.02637v2
- A. Nedić, A. Olshevsky, W. Shi, Achieving Geometric Convergence for Distributed Optimization over Time-Varying Graphs, SIAM J. Optim. 27 (2017). https://doi.org/10.1137/16M1084316
- G. Qu, N. Li, Harnessing Smoothness to Accelerate Distributed Optimization, IEEE Trans. Control Netw. Syst. 5 (2018). https://doi.org/10.1109/TCNS.2017.2698261
- P. Di Lorenzo, G. Scutari, NEXT: In-Network Nonconvex Optimization, IEEE Trans. Signal Inf. Process. Netw. 2 (2016). https://doi.org/10.1109/TSIPN.2016.2524588