Convergence Rate of Distributed ADMM over Networks
Ali Makhdoumi, Asuman Ozdaglar
I Introduction
Many of today’s optimization problems in data science (including statistics, machine learning, and data mining) include an abundance of data, which cannot be handled by a single processor alone. This necessitates distributing data among multiple processors and processing it in a decentralized manner based on the available local information. The applications in machine learning along with other applications in distributed data processing where information is inherently distributed among many processors (see e.g. distributed sensor networks , coordination and flow control problems ) have spearheaded a large literature on distributed multiagent optimization.
In this paper, we focus on the following optimization problem:
Least-Absolute Shrinkage and Selection Operator (LASSO):
Suppose our distributed computing system consists of machines each with data points (without loss of generality suppose is divisible by , otherwise one of the machines have the remainder of data points). For all , we define a function based on the available data to machine as
I-B Related Works and Contributions
Much of this literature builds on the seminal works , which proposed gradient methods that can parallelize computations across multiple processors. A number of recent papers proposed subgradient type methods or a dual averaging method to design distributed optimization algorithms.
An alternative approach is to use Alternating Direction Method of Multipliers (ADMM) type methods which for separable problems leads to decoupled computations (see e.g. and for comprehensive tutorials on ADMM). ADMM has been studied extensively in the 80’s . More recently, it has found applications in a variety of distributed settings in machine learning such as model fitting, resource allocation, and classification (see e.g. ).
In this paper we present a new distributed ADMM algorithm for solving problem (1) over a network. Our algorithm relies on a novel node-based reformulation of (1) and leads to an ADMM algorithm that uses dual variables with dimension given by the number of nodes in the network. This results in a significant reduction in the number of variables stored and communicated with respect to edge-based ADMM algorithm presented in the literature (see ). Our main contribution is a unified convergence rate analysis for this algorithm that applies to both the case when the local objective functions are convex and also the case when the local objective functions are strongly convex with Lipschitz continuous gradients. In particular, our analysis shows that when the local objective functions are convex (with no further assumptions), then the objective function at the ergodic average of the estimates generated by our algorithm converges with rate . Moreover, when the local objective functions are strongly convex with Lipschitz continuous gradients we show that the iterates converges linearly, i.e., the iterates converge to an -neighborhood of the optimal solution after steps, where is the condition number defined as , where is the maximum Lipschitz gradient parameter and is the minimum strong convexity constant of the local objective functions. This matches the best known iteration complexity and condition number dependence for the centralized ADMM (see e.g. ). Our convergence rate estimates also highlight a novel dependence on the network structure as well as the communication weights. In particular, for communication weights that are governed by the Laplacian of the graph, we establish a novel iteration complexity , where is the minimum degree, is the maximum degree, and is the algebraic connectivity of the network. Finally, we illustrate the performance of our algorithm with numerical examples.
Our paper is most closely related to , which studied edge-based ADMM algorithms for solving (1). In , the authors consider convex local objective functions and provide an convergence rate. The more recent paper assumes strongly convex local objective functions with Lipschitz gradients and show a linear convergence rate through a completely different analysis. This analysis does not extend to the node-based ADMM algorithm under these assumptions. In contrast, our paper provides a unified convergence rate analysis for both cases for the node-based distributed ADMM algorithm. Our paper is also related to that study the basic centralized ADMM where the goal is to miminize sum of two functions with a linearly coupled constraint. Our work is also related to the literature on the converge of operator splitting schemes, such as Douglas-Rachford splitting and relaxed Peaceman-Rachford .
I-C Outline
The organization of paper is as follows. In Section II we give the problem formulation and propose a novel distributed ADMM algorithm. In Section III, we show some preliminary results that helps us to show the main results. In Section IV we show the sub-linear convergence rate. In Section V we show the linear convergence rate of our algorithm. Finally, in Section VI we show the effect of network on the convergence rate and provide numerical results that illustrate the performance of our algorithm, which leads to concluding remarks in Section VII. All the omitted proofs are presented in the appendix.
II Framework
Consider a network represented by a connected graph where is the set of agents and is the set of edges. For any , let denote its set of neighbors including agent itself, i.e., , and let denote the degree of agent , i.e., . We let and .
The goal of the agents is to collectively solve optimization problem (1), where is a function known only to agent . In order to solve optimization problem (1), we introduce a variable for each and write the objective function of problem (1) as , so that the objective function is decoupled across the agents. The constraint that all the ’s are equal can be imposed using the following matrix.
Let be a matrix whose entries satisfy the following property: For any , for . We refer to as the communication matrix.
The communication matrix satisfies , where is a vector with all entries equal to one and denotes the null-space of the matrix .
If for all , summation of each row of is zero, and the graph is connected, then Assumption 1 holds. As a particular case, the Laplacian matrix of the graph given by when and zero otherwise, and is a communication matrix that satisfies Assumption 1.
We next show that the constraint that all ’s are equal can be enforced by the linear constraint , where where each is a sub-vector of dimension and is a matrix defined as the Kronecker product between communication matrix and , i.e., .
Under Assumption 1, the constraint guarantees that for all .
Using Lemma 1, under Assumption 1, we can reformulate optimization problem (1) as
where .
The optimal solution set of problem (3) is non-empty. We let denote an optimal solution of the problem (3).
II-B Multiagent Distributed ADMM
We now use ADMM algorithm (see e.g. ). ADMM algorithm generates primal-dual sequences , and which at iteration are updated as follows:
For any , we update as
For any , we update the vector as
where .
For and we update as
One can implement this algorithm in a distributed manner, where node maintains variables and for all (). However, using the inherent symmetries in the problem, we can significantly reduce the number of variables that each node requires to maintain from to .
We first show that for all , , and , we have . This reduction shows that the algorithm need not maintain dual variables for each and its neighbors , but instead can operate with the lower dimensional node-based dual variable . The dual variable can be updated using primal variables for all . The second observation is that , where and . This reduction shows that the algorithm need not maintain primal variables for each and its neighbors , but instead can operate with the lower dimensional node-based primal variables , where is node ’s estimate of the primal variable (obtained as the average of primal variables of his own neighbors). The aforementioned reductions are shown in the following proposition.
The sequence for generated by implementing the steps presented in Algorithm (1) is the same as the sequence generated by the ADMM algorithm.
The steps of the algorithm can be implemented in a distributed way, meaning that each node first updates her estimates based on the information received from her neighboring nodes and then broadcasts her updated estimates to her neighboring nodes. Each node maintains local variables , , and and updates these variables using communication with its neighbors as follows:
At the end of iteration , each node sends out and to all of its neighbors and then each node such as uses and of all to update as in step 1.
Each node sends out to all of its neighbors and then each node such as computes as in step 2.
Each node updates as in step 3.
Using this algorithm agent need to store only three variables, , , and and update them at each iteration. Also, each agent need to communicate only with (broadcast her estimates to) its neighbors. Therefore, the overall storage requirement is and the overall communication requirement at each iteration is .
III Preliminary Results
In this section, we present the preliminary results that we will use to establish our convergence rate. we define
The update of Algorithm 1 can be written as
for some .
Lemma 2 shows can be written as a perturbed linear combination of with the perturbation being the term . The intuition behind the convergence rate analysis is that the linear term that relates to guarantees that the sequence converges to a consensus point where for all ; and the perturbation term guarantees that the converging point minimizes the objective function .
IV Sub-linear Rate of Convergence
In this section, we show the sublinear rate of convergence. We define two auxiliary sequences that we will use in proving the convergence rates. Since is positive semidefinite (see Lemma 6 in the appendix), we can define . In other words, we let , where is the singular value decomposition of the symmetric matrix . We define the auxiliary sequences
Next, we show a proposition that bounds the function value at each iteration.
where .
In order to obtain convergence rate, we consider the performance of the algorithm at the ergodic vector defined as , where
for any . Note that each agent can construct this vector by simple recursive time-averaging of its estimate . Let be a primal-dual optimal solution of
Since , under Assumption (1), the optimal primal solution of this problem is the same as of the original problem (3) Next, we show both objective function and feasibility violation converges with rate to the optimal value.
This theorem shows that the objective function at the ergodic average of the sequence of estimates generated by Algorithm 1 converges with rate to the optimal solution. We next characterize the network effect on the performance guarantee.
For any , starting form , we have
V Linear Rate of Convergence
In order to show the linear rate of convergence, we adopt the following standard assumptions.
For any , the function is differentiable and has Lipschitz continuous gradient, i.e.,
for some . The function is also strongly convex with parameter , i.e., is convex.
We let and , and define the condition number of (or the condition number of problem (3)) as . Note that when the functions are differentiable, we have
Assumption 3 results in the following standard inequalities for the aggregate function .
Under Assumption (3) we show that the sequence generated by Algorithm (1) converges linearly to the optimal solution (which is unique under these assumptions). The idea is to use strong convexity and Lipschitz gradient property of in order to show that the -norm of sequence contracts at each iteration, providing a linear rate.
Suppose Assumptions 1, 2, and 3 hold. For any value of the penalty parameter and , the sequence generated by Algorithm 1 satisfies
The rate of convergence in Theorem 3 holds for any choice of penalty parameter . In other words, for any choice of , the convergence rate is linear. We now optimize the rate of convergence over all choices of and provide an explicit convergence rate estimate that highlights dependence on the condition number of the problem.
Suppose Assumptions 1, 2, and 3 hold. Let be the sequence generated by Algorithm 1. There exist for which we have
VI Network Effects
We can choose communication matrix (and the corresponding matrix ) in the Algorithm 1 to be any matrix that satisfies Assumption (1). One natural choice for the matrix is the Laplacian of the graph which leads to having for all and . Using Laplacian as the communication matrix we can now capture the effect of network structure in the convergence rate.
The following proposition explicitly show the networks dependence in the bounds provided in Theorem 2.
For any , starting form and using standard Laplacian as the communication matrix, we have
where is a bound on the subgradients of the function at , i.e., for all and is the algebraic connectivity of the graph.
Therefore, highly connected graphs with larger algebraic connectivity has a faster convergence rate (see e.g. , for an overview of the results on algebraic connectivity).
VI-B Network Effect in Linear Rate
The following proposition explicitly show the networks dependence in the bound provided in Theorem 4.
Suppose Assumptions 1, 2, and 3 hold. Using standard Laplacian as the communication matrix, in order to reach an -optimal solution iterations suffice.
Both of our guaranteed rates for sub-linear and linear rates depends on three parameters , and . The convergence rate is faster for larger and smaller . Finally, the convergence rate is faster for larger algebraic connectivity .
VII conclusion
We proposed a novel distributed algorithm based on Alternating Direction Method of Multipliers (ADMM) to minimize the sum of locally known convex functions. We first showed that ADMM can be implemented by only keeping track of some node-based variables. We then showed that our algorithm reaches -optimal solution in number of iterations for convex functions and in iterations for strongly convex and Lipschitz functions. Our analysis shows that the performance of our algorithm depends on the algebraic connectivity of the graph, the minimum degree of the nodes, and the maximum degree of the nodes. Finally, we illustrated the performance of our algorithm with numerical examples.
Appendix
Consider the -th coordinate of all ’s and form a vector . From and the fact that , we obtain that . This shows that . Using Assumption 1, we have , which guarantees that all entries of are equal. Similarly, for any , the -th entries of all ’s are equal. This leads to for all .
VII-B Proof of Lemma 2
Using the first step of Algorithm (1) for , we can write as
where for differentiable functions and in general. We next use second and third steps of Algorithm (1) to write and in terms of . Using the update for , we have
Moreover, we can write the term based on the sequence as follows
Substituting (VII-B) and (VII-B) in (VII-B), we can write the update of in terms of the sequence , which then can compactly be written as
where if the functions are differentiable and in general. Left multiplying by , completes the proof.
VII-C Proof of Proposition 2
We first show a lemma that we will use in the proof of this proposition. The following lemma shows the relation between the auxiliary sequence and the primal sequence .
Suppose Assumptions 1 and 2 hold. The sequence satisfies
for some .
We subtract from both sides and rearrange the terms to obtain
Back to the proof of Proposition 2: Using Lemma 7 and Lemma 3 part (c), we have that
VII-D Proof of Theorem 1
Taking summation of the relation given in Proposition 2 from up to , we obtain that
Using convexity of the functions and Jensen’s inequality, we obtain
Next, we will bound the term . We add the term to both sides of (12) to obtain
Again, using Proposition 2 to bound the right-hand side of (14), we obtain
Using (15) to bound the right-hand side of (13), and then combining the result with (12), we obtain
We next bound the feasibility violation. Using Proposition 2 with , we have
Since is a primal-dual optimal solution, using saddle point inequality, we have that
Combining the two previous relations, we obtain
we can further bound as
VII-E Proof of Theorem 2
We first show a lemma that bounds the norm of dual optimal solution of (16).
Next, we use this lemma to analyze the network effect. Using Theorem 1 with zero initial condition, we have
Using completes the proof.
VII-F Proof of Lemma 3
For any , since is convex, is monotone and we have
Let . Note that has Lipschitz gradient with parameter . Moreover, we have that , since
is zero for . Therefore, using the previous relation, we have that
We add the two preceding relations to obtain
for any . Combining this relation for all ’s completes the proof. Finally, we prove part (c). Let . By definition of subgradient, for any we have
Taking summation of this inequality for all shows that
VII-G Proof of Theorem 3
We first show two lemmas that we use in the proof of this theorem. The first lemma shows that both matrices and are positive semidefinite and the second lemma shows a relation between the sequences , , and same as the one shown in Lemma 7.
The matrices and are positive semidefinite.
Both matrices are clearly symmetric. We first show is positive semidefinite. By definition, we have
where we used the fact that , for any . Therefore, by Greshgorin Circle Theorem, for any eigen value of , for some we have
where we used the fact that that evidently holds. We next show that is positive semidefinite. We have
Since , similarly, by Greshgorin Circle Theorem, the matrix is positive semidefinite. ∎
Suppose Assumptions 1 and 2 hold. For differentiable functions, the sequence satisfies
for some that satisfies . Moreover, belongs to the column span of .
Using Lemma 2 for differentiable functions we have
We subtract from both sides and rearrange the terms to obtain
Back to the proof of Theorem 3: Note that since is positive semidefinite,
is a semi-inner product.This means it satisfies conjugate symmetry, linearity and semipositive-definiteness (instead of positive-definiteness). We first show that for a given by the statement of theorem, we have
Again, using Lemma (3) and Lemma (7), we have
Using (VII-G) and (VII-G), for any , we have
Comparing this relation with (18), it remains to show
Using Lemma (6), in order to show this inequality it suffices to show
Since both and are orthogonal to and , using Lemma (7), we obtain
Comparing (VII-G) and (VII-G), it suffices to have
This shows that (18) holds. Using (18) along with (VII-G) completes the proof.
VII-H Proof of Theorem 4
The largest possible that satisfies the constraint given in Theorem 1 by maximizing over is the solution of
VII-I Proof of Proposition 3
where is the algebraic connectivity of the graph which is the smallest non-zero eigenvalue of the Laplacian matrix. Moreover, we have that
Plugging these two bounds in Theorem 2 an using the fact that completes the proof.
VII-J Proof of Proposition 4
Using Theorem 4, for large enough (small enough ) we have , in order to have , we need to have
Using these two bounds along with , we obtain
Plugging this bound into (VII-J) completes the proof.