Distributed Optimization Based on Gradient Tracking Revisited: Enhancing Convergence Rate via Surrogation 2: SONATA with Push-Sum Converges R-Linearly over Time-Varying Directed GraphsResearch Paper
Motivation
In distributed optimization a network of agents minimizes a sum of private functions while each agent talks only to its neighbours. Applications include federated and decentralized learning, sensor networks and multi-robot systems, where the data cannot be pooled at one site. Many of these networks are directed (communication links are one-way, for instance because transmission powers differ) and time-varying (links come and go). On such networks no doubly stochastic mixing matrix is available in general, and plain averaging drifts toward a weighted average instead of the true one. The push-sum protocol corrects this drift by carrying an auxiliary weight at every agent (Kempe, Dobra & Gehrke 2003; Nedić & Olshevsky 2015).
Y. Sun, A. Daneshmand and G. Scutari (arXiv:1905.02637v2) analyse SONATA, a gradient-tracking method in which each agent solves a strongly convex surrogate of its local problem instead of taking a gradient step, for composite problems (a convex, possibly nonsmooth regulariser and a convex constraint set). They prove linear convergence on undirected graphs (their Theorem 3.9, the companion mission of this series) and, in Section 4, for a push-sum version over time-varying digraphs (Theorem 4.5). This mission formalizes the second result.
Timeline. Gradient tracking with push-sum over time-varying digraphs appears in NEXT/SONATA (Di Lorenzo & Scutari 2016; Scutari & Sun 2019), where convergence was proved without rates for nonconvex problems. Linear rates for smooth unconstrained strongly convex problems over time-varying digraphs were obtained for Push-DIGing (Nedić, Olshevsky & Shi 2017). Sun, Daneshmand and Scutari (2019, revised 2020) extend the linear rate to the composite, constrained setting with surrogates, and require only the average , not every , to be strongly convex.
Setting
Problem (P). Agent holds ; the agents solve
Assumption A: is nonempty, closed and convex; each is twice differentiable on an open set and convex; on with ; is convex on . is the solution and .
Surrogates (Assumption C). Each agent has , on , with , -Lipschitz and -strongly convex on . Constants satisfy on (15); , .
Network (Assumption B′). At time the digraph has edge set ; is a link from to . There is an integer such that the union of the edge sets over every window is strongly connected.
Weights (Assumption E). is column stochastic, , if and for other .
Algorithm 3. From , , , each agent computes
then , and .
The analysis uses the weighted gap , the tracking error , and the consensus errors , measured against the -weighted averages.
Formalization targets
Goal: Theorem 4.5
If , there is such that for every , every run and every agent ,
The goal is qualitative: neither nor the rate is fixed, so it survives any improvement of the constants.
Milestones
In attack order: the weighted tracking identity (74) and its consequence ; the bounds (p. 29); the per-agent descent inequality (Lemma I.1); the lower bound on the weighted step length (Lemma I.2); the tracking-error bound (Lemma I.3); the contraction (Proposition 4.1); the bound of by and (Proposition 4.3).
Significance
Theorem 4.5 gives a linear rate for distributed composite and constrained optimization over time-varying directed networks. Two features matter: the individual need not be strongly convex, and the surrogates cover proximal-gradient steps, DANE-type local solves and mirror-descent steps under one analysis. The rate depends on the surrogate-mismatch constants , which is the mechanism by which the paper shows that local computation helps when the are similar.
The result is proved in the paper; the proof of Theorem 4.5 refers to the undirected argument, and several of its supporting statements (Propositions 4.1 and 4.3) have their proofs omitted or relegated to the supplement, with consensus estimates quoted from Scutari & Sun (2019). To our knowledge none of it is machine-checked. A formalization produces a checked proof of the directed result together with reusable statements about push-sum weights over -strongly connected digraphs.
Difficulty
The obvious argument is to repeat the undirected proof with weighted averages. Two steps fail. First, the -weighted average is not the minimiser of , so consensus errors measured against it lose a constant (the 8 in Lemma I.3 instead of 4), and every passage between weighted and unweighted sums needs the bounds , which rest on a product-of-matrices argument over windows of length . Second, the matrices change with , so there is no single contraction factor; geometric decay of the consensus errors holds only over windows, and the coupling between the optimality gap, the consensus errors and the step lengths must be closed by a small-gain argument whose constants are uniform over the run. The step-size threshold must not depend on the starting point.
Formalization scope
Points are EuclideanSpace ℝ (Fin d); stacked vectors are Fin m → ℝ^d with the Euclidean norm , never Lean's sup norm. The Hessian is the derivative of the gradient, and Loewner bounds are quadratic-form inequalities. Time is 0-based. A run is a relational predicate: is some minimizer of the subproblem, whose existence the paper takes for granted; is assumed to exist. The divisions by appear as in the paper and are genuine, since Assumption E keeps the weights positive. is convex on and locally convex on (positive semidefinite Hessian; need not be convex), , and are parameters: the statements hold for every valid choice. The graph sequence, , and the matrices come before ; the starting point and the run come after . The bounds on are stated for (the printed is at ), and positivity of is claimed only when or .
The following do not count as a formalization of the goal: an unsatisfiable run predicate (a sorry-free verification file exhibits a run satisfying every hypothesis), a chosen after the run, the bounds or a contraction factor assumed in place of Assumptions B′ and E, and dropping the normalisation.
Not posed: the explicit rate (Theorem III.1), the complexity corollaries, and the consensus estimates Lemma 4.2 and Proposition 4.4, whose constants come from Scutari & Sun (2019) (as printed, is negative). A complete development needs Taylor expansions with integral remainder along segments, first-order optimality for composite problems on convex sets, and products of column-stochastic matrices over strongly connected windows; the last is reusable for any push-sum method. Contributions to any milestone, and to the missing consensus estimates as auxiliary lemmas, are welcome.
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(2), 2022. https://arxiv.org/abs/1905.02637v2
- G. Scutari, Y. Sun, Distributed nonconvex constrained optimization over time-varying digraphs, Math. Program. 176, 2019. https://doi.org/10.1007/s10107-018-01357-w
- P. Di Lorenzo, G. Scutari, NEXT: In-Network Nonconvex Optimization, IEEE Trans. Signal Inf. Process. Netw. 2(2), 2016. https://doi.org/10.1109/TSIPN.2016.2524588
- A. Nedić, A. Olshevsky, W. Shi, Achieving Geometric Convergence for Distributed Optimization over Time-Varying Graphs, SIAM J. Optim. 27(4), 2017. https://doi.org/10.1137/16M1084316
- A. Nedić, A. Olshevsky, Distributed optimization over time-varying directed graphs, IEEE Trans. Autom. Control 60(3), 2015. https://doi.org/10.1109/TAC.2014.2364096
- D. Kempe, A. Dobra, J. Gehrke, Gossip-based computation of aggregate information, FOCS 2003. https://doi.org/10.1109/SFCS.2003.1238221