Multi-Agent Distributed Optimization via Inexact Consensus ADMM
Tsung-Hui Chang, Mingyi Hong, Xiangfeng Wang
I Introduction
We consider a network with multiple agents, for example a sensor network, a data cloud network or a communication network. The agents seek to collaborate to accomplish certain task. For example, distributed database servers may cooperate for data mining or for parameter learning in order to fully exploit the data collected from individual servers . Another example arises from large-scale machine learning applications , where a computation task may be executed by collaborative microprocessors with individual memories and storage spaces . Distributed optimization becomes favorable as it is not always efficient to pool all the local information for centralized computation, due to large size of problem dimension, a large amount of local data, energy constraints and/or privacy issues . Many of the distributed optimization tasks, such as those described above, can be cast as an optimization problem of the following form
In the setting of distributed optimization, it is commonly assumed that each agent only has knowledge about the local information , and . The challenge is to obtain, for each agent in the system, the optimal of (P1) using only local information and messages exchanged with neighbors .
In addition to (P1), another common problem formulation has the following form
Problem (7) thus has the same form as (P1). Given the optimal of (7) and assuming that (P2) has a zero duality gap , each agent can obtain the associated optimal variable by solving (8). Therefore, a distributed optimization method that can solve (P1) may also be used for (P2) through solving (7).
There is an extensive literature on distributed consensus optimization methods, such as the consensus subgradient methods; see and the recent developments in . The consensus subgradient methods are appealing owing to their simplicity and the ability to handle a wide range of problems. However, the convergence of the consensus subgradient methods are usually slow.
Recently, the alternating direction method of multipliers (ADMM) has become popular for solving problems with forms of (P1) and (P2) in a distributed fashion. In , distributed transmission designs for multi-cellular wireless communications were developed based on ADMM. In , several ADMM based distributed optimization algorithms were developed for solving the sparse LASSO problem . In , using a different consensus formulation from and assuming the availability of a certain coloring scheme for the graph, ADMM is applied to solving the BP problem for both row partitioned and column partitioned data models . In , the methodologies proposed in are extended to handling a more general class of problems with forms of (P1) and (P2). In , a distributed ADMM with a sequential update rule is proposed; while in , the method is extended and can be implemented asynchronously. The fast practical performance of ADMM is corroborated by its nice theoretical property. In particular, ADMM was found to converge linearly for a large class of problems , meaning a certain optimality measure can decrease by a constant fraction in each iteration of the algorithm. In , such fast convergence rate has also been built for distributed optimization.
It is important to note that existing ADMM based algorithms can be readily used to solve problems (P1) and (P2). For example, by applying the consensus formulation proposed in and ADMM to (P1), a fully parallelized distributed optimization algorithm can be obtained (where the agents update their variables in a fully parallel manner), which we refer to as the consensus ADMM (C-ADMM). To solve (P2), the same consensus formulation and ADMM can be used on its Lagrange dual problem in (7), referred to as the dual consensus ADMM (DC-ADMM). The main drawback of these algorithms lies in the fact that each agent needs to repeatedly solve certain subproblems to global optimality. This can be computationally demanding, especially when the cost functions ’s have complicated structures or when the problem size is large . If a low-accuracy suboptimal solution is used for these subproblems instead, the convergence is no longer guaranteed.
The main objective of this paper is to study algorithms that can significantly reduce the computational burden for the agents. In particular, we propose two algorithms, named the inexact consensus ADMM (IC-ADMM) and the inexact dual consensus ADMM (IDC-ADMM’), both of which allow the agents to perform a single proximal gradient (PG) step at each iteration. The benefit of the proposed approach lies in the fact that the PG step is usually simple, especially when ’s are structured functions . Notably, the cheap iterations of the proposed algorithms is made possible by inexactly solving the subproblems arising in C-ADMM and DC-ADMM, in a way that is not known in the ADMM or consensus literature. For example, the proposed IC-ADMM approximates the smooth functions ’s in C-ADMM, which is very different from the known inexact ADMM methods , where only the quadratic penalty is approximated (thus does not always result in cheap PG steps). We summarize our main contributions below.
For (P1), we propose an IC-ADMM method for reducing the computational complexity of C-ADMM. Conditions for global convergence of IC-ADMM are analyzed. Moreover, we show that IC-ADMM converges linearly, under similar conditions as in .
For (P2), we first propose a DC-ADMM method which can globally solve (P2) for any connected graph and convex ’s. We further propose an IDC-ADMM method for reducing the computational burden of DC-ADMM. Conditions for global (linear) convergence are presented.
Numerical examples for solving distributed sparse logistic regression problems will show that the proposed IC-ADMM and IDC-ADMM methods converge much faster than the consensus subgradient method . Further, compared with the original C-ADMM and DC-ADMM, the proposed method can reduce the overall computational cost by an order of magnitude.
The paper is organized as follows. Section II presents the applications and assumptions. The C-ADMM and IC-ADMM are presented in Section III; while DC-ADMM and IDC-ADMM are presented in Section IV. Numerical results are given in Section V and conclusions are drawn in Section VI.
Notations: () means that matrix is positive semidefinite (positive definite). is the identity matrix; is the -dimensional all-one vector. denotes the Euclidean norm of vector , and for some . Notation denotes the Kronecker product. is a diagonal matrix with the th diagonal element being ; while is a block diagonal matrix with the th diagonal block matrix being . and denote the maximum and minimum eigenvalues of matrix , respectively.
II Applications and Network Model
where contains training data vectors and are binary labels for the training data. It is clear that (9) has the same form as (P1). Here, the non-smooth function can be 1-norm for sparse regression, as well as mixture with an indicator functions specifying that is confined in certain constraint set.
By introducing a slack variable , the CPD LR problem can be reformulated as
which is an instance of (P2). In Section V, we will primarily test our algorithms on the RPD and CPD regression problems.
II-B Network Model and Assumptions
Let an undirected graph denote a multi-agent network, which contains a node set and an edge set . An edge if and only if agent and agent can communicate with each other (i.e., neighbors). The edge set defines an adjacency matrix , where if and otherwise. In addition, one can define an index subset for the neighbors of each agent , and a degree matrix (a diagonal matrix). With and , the Laplacian matrix of is given by which is a positive semidefinite matrix (i.e., ) and satisfies .
We make the following assumptions on and problems (P1) and (P2).
The undirected graph is connected.
Assumption 1 implies that any two agents in the network can always influence each other in the long run. We also have the following assumptions on problems (P1) and (P2).
For all , the smooth function in (2) is strongly convex, i.e., there exists some such that
Moreover, has Lipschitz continuous gradients, i.e., there exists some such that
Note that, even under Assumption 3, is not necessarily strongly convex in since the matrix can be fat and rank deficient. Both the LASSO problem and the LR function in (10) satisfy Assumption 3 The logistic regression function is strongly convex given that lies in a compact set..
III Distributed Consensus ADMM
In Section III-A, we briefly review the original C-ADMM for solving (P1). In Section III-B, we propose a computationally efficient inexact C-ADMM method.
Under Assumption 1, (P1) can be equivalently written as
where are slack variables. According to (15), each agent can optimize its local function with respect to a local copy of , i.e, , under the consensus constraints in (15b) and (15c). In , ADMM is employed to solve (15) in a distributed manner. Let and denote the Lagrange dual variables associated with constraints (15b) and (15c), respectively. According to , ADMM leads to the following iterative updates at each iteration :
where is a penalty parameter and . Note that variables are not shown in (16) as they can be expressed by variables ; see for the details.
The updates in (16) are useful for convergence analysis. For practical implementation, we define . Then, (16) boils down to Algorithm 1.
It is important to note from Step 4 and Step 5 of Algorithm 1 that, except for the parameter which has to be universally known, each agent updates the variables in a fully parallel manner, by only using the local function and messages , which come from its direct neighbors. It has been shown in that, under Assumptions 1 and 2, C-ADMM is guaranteed to converge for any In general, the parameter is chosen empirically. Only for some special instance, optimal may be analytically found; e.g., see . :
where and denote a pair of optimal primal and dual solutions to problem (15), and is optimal to (P1). It is also shown that C-ADMM can converge linearly when ’s are purely smooth (i.e., ) and strongly convex with respect to .
One key issue about C-ADMM is that the subproblem in (17) is not always easy to solve. For instance, for the LR function in (10), the associated subproblem (17) is given by
As seen, due to the complicated LR cost, problem (III-A) cannot yield simple solutions, and a numerical solver has to be employed. Clearly, obtaining a high-accuracy solution of (III-A) can be computationally expensive, especially when the problem dimension or the number of training data is large. While a low-accuracy solution to (III-A) can be adopted for complexity reduction, it may destroy the convergence behavior of C-ADMM, as will be shown in Section V.
III-B Proposed Inexact C-ADMM
To reduce the complexity of C-ADMM, instead of solving subproblem (17) directly, we consider the following update:
In (III-B) we have replaced the smooth cost function in (17) with a proximal first-order approximation around :
where . Clearly, using this definition, (III-B) can be expressed more compactly as
which is a proximal gradient (PG) update.
where . The IC-ADMM is presented in Algorithm 2.
Although the idea of “inexact ADMM” is not new, our approach is significantly different from the existing methods , where the inexact update is obtained by approximating the quadratic penalization term only. It can be seen that problem (III-A) is still difficult to solve even the inexact update in is applied. Two notable exceptions are the algorithms proposed in and where the cost function is also linearized. However, an additional back substitution step and two extragradient steps are required in and , respectively, which is not suited for distributed optimization.
The convergence properties of IC-ADMM is characterized by the following theorem.
Suppose that Assumptions 1, 2(a) and 3 hold. Let
and let and denote a pair of optimal primal and dual solutions to problem (15) (i.e., (P1)).
For Algorithm 2, converge to a common point .
If , where has full column rank, for all , then we have
The proof is presented in Appendix A. Theorem 1 implies that, given sufficiently large ’s, IC-ADMM not only achieves consensus and optimality, but also converges linearly provided that is purely smooth and strongly convex. Note that, to ensure (25), the global knowledge of is required by all agents. As a parallel work, we should mention that a concurrent result similar as Theorem 1(b) is presented in .
IV Distributed Dual Consensus ADMM
In this section, we turn the focus to (P2). In Section IV-A, we present a DC-ADMM method for solving (P2). In Section IV-B, an inexact DC-ADMM method is proposed.
The DC-ADMM is obtained by applying the C-ADMM (Algorithm 1) to problem (7) which is equivalent to the Lagrange dual of (P2). Firstly, similar to (15), we write problem (7) as
in which and are dual variables associated with the two constraints in (28b) and are updated in a similar fashion as in (16a) and (16b), i.e.,
In general, subproblem (29b) is not easy to handle because is implicit and (29b) is in fact a min-max optimization problem given by
Fortunately, since the objective function in (IV-A) is convex in for any and is concave in for any , the minimax theorem [38, Proposition 2.6.2] can be applied so that the min-max problem (IV-A) can be equivalently solved by considering its max-min counterpart and saddle point exists. Specifically, the max-min counterpart of (IV-A) is given by
where the equality is obtained by completing the quadratic term of . Let be an inner maximizer of (IV-A) so that is a saddle point of (IV-A). Then, is a pair of outer-inner solution to (IV-A) and (34) [38, Proposition 2.6.1]. From (34), the inner minimizer can be uniquely determined by
As a result, the min-max subproblem (29b) can actually be obtained by first solving the subproblem (IV-A) with respect to the primal variable followed by evaluating using the close-form in (IV-A). The proposed DC-ADMM is summarized in Algorithm 3.
Interestingly, while DC-ADMM handles the equivalent dual problem in (7), it directly yields primal optimal solution of (P2), as we state in the following theorem.
Suppose that Assumptions 1 and 2(b) hold. Then converges to a common point , which is optimal to the dual problem (7). Moreover, any limit point of is primal optimal to (P2).
Proof: Since DC-ADMM is a direct application of C-ADMM to the dual problem (7), it follows from that as ,
What remains is to show that any limit point of is asymptotically optimal to (P2), i.e., as ,
To show (40), consider the optimality condition of (IV-A), i.e.,
where the second equality is obtained by (IV-A). Since (IV-A) holds for all and by (39), (40) is true when .
where the last equality is obtained by (30) and (31). Upon summing (IV-A) for , and by the fact that
(by applying (A.13) and (A) in Appendix A), we can obtain
Note that as inferred from in (39). By applying this fact to (44), we obtain that (41) is true as .
Interestingly, from (44), one observes that the primal feasibility of to (P2) depends on the agents’ consensus on the dual variable .
We remark that Algorithm 3 is different from the D-ADMM algorithm in [12, Algorithm 3]. Firstly, Algorithm 3 can be implemented in a fully parallel manner; secondly, Algorithm 3 does not involve solving a min-max subproblem at each iteration; thirdly, convergence of Algorithm 3 can be achieved without the assumption that the graph is bipartite.
IV-B Proposed Inexact DC-ADMM
where, with a slight abuse of notation, is a penalty parameter. By (21), equation (IV-B) can be further written as the following PG update
We summarize the proposed IDC-ADMM in Algorithm 4.
The convergence property of IDC-ADMM is stated below.
Suppose that Assumptions 1, 2(b) and 3 hold and
Let denote an optimal solution to (P2), and let and denote a pair of optimal primal and dual solutions to problem (28) (i.e., (7)).
The sequence generated from Algorithm 4 converges to of (P2) while converge to a common point of problem (7).
If , where has full column rank, and has full row rank, for all , then for some and , we have
where and are defined similarly as in Theorem 1, is defined in (27), and .
The proof is presented in Appendix B. Note that, in addition to the smooth and strongly convex objective function, IDC-ADMM also requires matrices ’s to have full row rank in order to have a linear convergence rate.
V Numerical Results
In this section, we examine the numerical performance of Algorithm 1 to 4 presented so far.
To implement C-ADMM (Algorithm 1), we employed the fast iterative shrinkage thresholding algorithm (FISTA) to solve subproblem (17) for each agent . For (17), the associated FISTA steps can be shown as
For IC-ADMM (Algorithm 2), the corresponding step in (III-B) is given by
The stopping criterion of Algorithms 1 and 2 was based on measuring the solution accuracy and variable consensus error , where , denotes the objective value of (9) given , and is the optimal value of (9) which was obtained by FISTA with a high solution accuracy of . The two algorithms were set to stop whenever acc and cserr are both smaller than preset target values.
V-B Performance of DC-ADMM and IDC-ADMM
In Table II(b), we considered another example with increased to . We set for DC-ADMM, and set and for IDC-ADMM. From Table II(b) and Figs. 3(b) and 3(c), one can observe similar results.
VI Conclusions
In this paper, we have presented ADMM based distributed optimization methods for solving problems (P1) and (P2) in multi-agent networks. In particular, aiming at reducing the computational complexity of C-ADMM for solving large-scale instances of (P1) with complicated objective functions, we have proposed the IC-ADMM method (Algorithm 2) where agents perform one PG update only at each iteration. For (P2), we have proposed the DC-ADMM method (Algorithm 3) and its complexity reduced counterpart IDC-ADMM (Algorithm 4). Preliminary numerical results based on the distributed LR problems (9) and (II-A) have shown that the proposed methods converge faster than the consensus subgradient method. Moreover, both IC-ADMM and IDC-ADMM require more ADMM iterations than C-ADMM and DC-ADMM, but the traded computational complexity reduction is significant.
Appendix A Proof of Theorem 1
By recalling that , and by the optimality condition of (III-B) , we have that
By combining (A.4) with (A.1), one obtains
Adding and subtracting in the left hand side (LHS) of (A.5) followed by multiplying on both sides yields
Note that the first term on the LHS of (A) can be lower bounded as
for any , where the second inequality is due to (14) in Assumption 3. By the strong convexity of and convexity of , the third and fourth terms of (A) can respectively be lower bounded as
Moreover, it follows from (16a) and (16b) that the fifth term of (A) can be expressed as
By substituting (A) to (A) into (A) and summing over , we obtain
It can be observed from (A.3) and also (16a) and (16b) that
given the initial which is equivalent to setting (See Step 1 of Algorithm 2). Besides, due to the symmetric property of , for any , we have
By the above two properties, the fourth and fifth terms in the LHS of (A) can be written as
where the first equality is owing to (A), the second equality is by (A.12) and (A.13), and the third equality is due to (16a). In (A), () is a vector that stacks () for all , . The sixth term in the LHS of (A) can be rearranged as follows
Note that, by the graph theory , the normalized Laplacian matrix, i.e., , have . Thus, in (A),
By substituting (A) and (A) into (A), we obtain
for any sequence and matrix . By applying (A) to each of the terms in (A), one obtains that
Now, consider the condition on in (25). It can be easily checked that (25) implies that
for some , and therefore
Thirdly, by applying the result of and (A.23) to (A.4), we have
Proof of Theorem 1(b): Let be some positive number and rewrite (A) as
Then, in order to prove linear convergence rate, i.e., for some ,
By applying (A.12) and (A.13), (A) can be expressed as
After stacking (A) for , one obtains
According to Note that the matrix corresponds to matrix in ., both and lie in the range space of . Hence, one can show that
where is the minimum nonzero singular value of . From (A), we have that
where the first inequality is due to the fact that
for any and the second inequality is obtained by setting and (A.32), and the last inequality is by (14). Equation (A) implies that
which are respectively satisfied if the following three conditions can be satisfied for some
Appendix B Proof of Theorem 3
Firstly, by recalling that , it follows from (IV-A) and (A.39) that
By multiplying to the both sides of (A.43), we obtain
Secondly, from the optimality of (IV-B), we have that
where, in the first equality, we have added and subtracted and defined
the second equality is due to (IV-A); and the last equality is because is a maximizer to (8) with . Multiplying both (A.46) and (A.47) with , combining with (B), and summing for , yields
Similar to (A) and by (31), the fifth term in the LHS of (B) can be expressed as
Moreover, the sixth term in the LHS of (B) can be shown as
where . By applying (A), (A), (A.9), (B) and (B) to (B), one obtains
It is easy to show that, under (49), it holds true that
for some , which implies that
Thirdly, applying the fact of to (31a) yields
The result of and Assumption 1 implies that for any . Since the Laplacian matrix , one obtains
which, when combined with (A.56), further implies that
By applying (A.61) to (A.42), one obtains
where since (A.57) implies that is a maximizer to (8) with . Finally, by summing (A.62) for , followed by applying (A) and (A.58), one obtains
Therefore, it suffices to show that, for some ,
Firstly, from (A.45) and (A.47), we have that (without ’s)
By applying (A.34) to (B), we have, for some ,
where the second inequality is obtained by (14). Note that as . Hence, we have
where the second inequality is due to (B), \tau_{1}=\max_{i\in V}\big{\{}\frac{|{\mathcal{N}}_{i}|}{(1-\frac{1}{\mu_{1}})\lambda_{\min}({\bf E}_{i}{\bf E}_{i}^{T})}\big{\}}>0 and \tau_{2}=\max_{i\in V}\big{\{}\frac{(\mu_{1}-1)\lambda_{\max}^{2}({\bm{A}}_{i}^{T}{\bm{A}}_{i})|{\mathcal{N}}_{i}|L_{f,i}}{(1-\frac{1}{\mu_{1}})\lambda_{\min}({\bf E}_{i}{\bf E}_{i}^{T})}\big{\}}>0 are finite given that ’s have full row rank.
Secondly, upon stacking (A.43) for all and applying (A.3) and (A.12), one obtains
where and is given in (A.31). Analogously, by applying (A.34) to (B) and by (A.32), one can show that, for some ,
where . By (B) and (B), sufficient conditions for satisfying (B) are therefore given by:
Under (A.54) and full column rank ’s, we see that (A.71) is true for some . The proof is complete.