TL;DR
This paper investigates structure learning of sparse Gaussian Graphical Models over distributed data in a wireless network, proposing two communication strategies and analyzing their theoretical and empirical performance.
Contribution
It introduces two novel methods, Signs and Uncoded, for reliable GGM structure learning over noisy wireless channels with limited power and bandwidth.
Findings
Both methods can recover the structure with high probability given sufficient samples.
Signs method outperforms Uncoded method in various scenarios.
Theoretical analysis confirms the effectiveness of the proposed strategies.
Abstract
A central machine is interested in estimating the underlying structure of a sparse Gaussian Graphical Model (GGM) from datasets distributed across multiple local machines. The local machines can communicate with the central machine through a wireless multiple access channel. In this paper, we are interested in designing effective strategies where reliable learning is feasible under power and bandwidth limitations. Two approaches are proposed: Signs and Uncoded methods. In Signs method, the local machines quantize their data into binary vectors and an optimal channel coding scheme is used to reliably send the vectors to the central machine where the structure is learned from the received data. In Uncoded method, data symbols are scaled and transmitted through the channel. The central machine uses the received noisy symbols to recover the structure. Theoretical results show that both…
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Code & Models
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Structure Learning of Sparse GGMs over Multiple Access Networks
Mostafa Tavassolipour, Armin Karamzade, Reza Mirzaeifard, Seyed Abolfazl Motahari, and Mohammad-Taghi Manzuri Shalmani
Abstract
A central machine is interested in estimating the underlying structure of a sparse Gaussian Graphical Model (GGM) from datasets distributed across multiple local machines. The local machines can communicate with the central machine through a wireless multiple access channel. In this paper, we are interested in designing effective strategies where reliable learning is feasible under power and bandwidth limitations. Two approaches are proposed: Signs and Uncoded methods. In Signs method, the local machines quantize their data into binary vectors and an optimal channel coding scheme is used to reliably send the vectors to the central machine where the structure is learned from the received data. In Uncoded method, data symbols are scaled and transmitted through the channel. The central machine uses the received noisy symbols to recover the structure. Theoretical results show that both methods can recover the structure with high probability for large enough sample size. Experimental results indicate the superiority of Signs method over Uncoded method under several circumstances.
Index Terms:
Structure learning, Gaussian graphical model, distributed learning
I Introduction
In recent years, by the explosion of the volume of training data, distributed machine learning has become more important than ever. Many modern big datasets are distributed over several hosting machines which are connected to each other via some communication links. In such systems, designing distributed learning algorithms which efficiently exploit the available resource demands a careful design where intensive computational workloads and the amount of data communications are taken into account.
This paper is focused on the problem of structure learning in Gaussian Graphical Models (GGMs) in distributed environments. A GGM for a -dimensional random vector is specified by a graph where is the set of vertices and is the set of edges. The model comprises all -dimensional normal distributions where is the precision matrix with iff . It worths mentioning that GGM is indeed a Markov Random Field (MRF).
In our problem setting, we assume that the data are distributed over multiple local machines so that each one contains a single dimension of the whole dataset. The local machines are connected to a central machine via a wireless medium. The central machine is responsible for inferring the conditional dependencies between the gathered data by the local machines. The system’s block diagram is depicted in Figure 1. We also assume that the communication links between the local and the central machine are bandwidth limited implying that transmission of the whole local datasets to the central machine is impossible. Due to this constraint, each local machine transmits some information from its local dataset to the central machine. Then, the central machine estimates the underlying graph structure using received information from the local machines.
In this paper, we consider a wireless multiple access channel between the local machines and the central node. Each local machine is equipped with a single antenna while the central node is equipped with multiple antennas. Hence, the overall channel between the local and central machines is modeled as a single-input multiple-output multiple access channel (SIMO-MAC).
We have proposed two communication schemes for transmitting information from the local machines to the central machine. In the first scheme, we have separate the source coding from the channel coding. In this scheme, we quantize the source samples into single bits and assume that there exists a channel coding such that the bits can be sent through the channel reliably. We refer to this scheme as Signs method in the paper.
In the second scheme, we do not use any source and channel coding. In this scheme, we put the source samples into the channel without any encoding. At the central machine, we directly estimate the underlying graph structure using received data from the channel. We refer to this scheme as Uncoded method.
We have shown through theoretical analysis and experiments that by transmitting only 1 bit per sample; the central machine can reliably recover the underlying graph structure. More precisely, we have shown theoretically that by consuming only 1 bit per sample, under some mild conditions, the central machine can perfectly recover the graph structure with high probability. Moreover, the true signs of the edges weights of the graph are obtained.
The paper is organized as follows. In Section II, we provide a brief review on structure learning of GGMs. Section III describes the detail of our modeling for the source and the communication channel. In sections IV and V, we describe Signs and Uncoded methods, respectively. Section VI provides experimental results to compare and evaluate the proposed methods. Finally, Section VII concludes the paper.
II Related Work
The problem of structure learning of GGMs from data samples has applications in many fields including biology and social networks. For example, it has been used for gene regulatory networks reconstruction in [1, 2] and analysis of users relationships in social networks in [3]. There are many studies addressing this problem from various perspectives [4, 5, 6].
The Chow-Liu algorithm obtains the maximum likelihood estimate of the structure if the underlying graph is a tree [7]. Although this algorithm is applicable for discrete random variables, it can be used for tree structured GGMs in a similar manner [8]. Tavassolipour et al. [9] proposed a distributed version of the Chow-Liu algorithm and proved it can recover the underlying tree structure with high probability. Tan et al. in [10] and [11] provided an analysis of the error exponent of the Chow-Liu algorithm on tree-structured GGMs.
GGMs have the property that the neighbors of each variable can be obtained by solving a linear regression problem for the corresponding variable on other variables. This approach is referred to as neighborhood selection in the literature. For the sparse structures, there are some methods which penalize the linear regression problem with of the coefficient vector [12, 13].
Among the proposed methods for sparse structure estimation of GGMs, the -regularized maximum likelihood approaches are more popular [6, 12, 14]. This class of model is analyzed in several articles (see [4] and refs therein). For instance, Ravikumar et al. [4] analyzed the performance of the -regularized maximum likelihood estimator (MLE) under high dimensional scaling. They showed that with probability converging to one, the estimated structure correctly specifies the zero pattern of the true precision matrix. A similar study is conducted by [15] which analyzed the consistency of the -regularized MLE in the Frobenius norm.
Besides the lasso typed estimators, thresholding based estimators are proposed for sparse recovering of the precision matrix [5, 8]. For example, Sojoudi [5] proposed a simple thresholding method and showed, under certain conditions, the resulting structure is identical to the structure obtained by the lasso.
In the distributed setting, there are several studies which address the problem of covariance/precision matrix estimation [16, 17, 18]. Arroyo and Hou [18] studied the problem of sparse precision matrix estimation in the situation where the samples are distributed among several machines. Their work differs from our setting in the sense that we assume the data are split across dimensions whereas they split the data across samples. Meng et al. [16] addressed estimation of the precision matrix in a distributed manner where the zero pattern of the precision matrix is known in advanced.
III Problem Formulation
We are given i.i.d. random vectors drawn from a -dimensional zero mean normal distribution with . The focus of this paper is the problem of estimating the zero pattern of the sparse precision matrix in a situation where the data is stored in separate local machines such that each machine possesses one dimension of the sample vectors.
Denoting the whole gathered data by , the th machine captures the -th dimension of the sample vectors. We denote the -th dimension of the -th sample by , i.e. . Hence, the local data at the -th local machine is .
The probability density function of the normal distribution is given by
[TABLE]
The negative log-likelihood of i.i.d. samples is given by
[TABLE]
where is reffered to as sample covariance matrix. One of the well known methods for estimating the sparse precision matrix is to solve an -regularized maximum likelihood function stated as
[TABLE]
where , and is a user defined regularization parameter. There is an efficient algorithm known as glasso [19] for solving the above log-determinant program.
In [4], it is stated that the Hessian matrix of (2) is given by
[TABLE]
where is the Kronecker matrix product. The entry corresponds to the second derivative , evaluated at and .
We define the set of non-zero entries in the true precision matrix as
[TABLE]
Recall that . Note that includes the diagonal entries. Let be the complement set of which includes all pairs where . For any two subsets and of , the notation denotes a sub-matrix of with rows and columns indexed by and , respectively.
We adopt the incoherence condition used by Ravikumar et al. [4] to obtain error bounds on consistency of the solutions. We define max-row-sum norm of a by matrix as
[TABLE]
Assumption 1**.**
*(Incoherence Condition [4])
There exists some such that*
[TABLE]
An implication of the above condition is that the non-edge pairs cannot have strong influence on the edges.
Assumption 2**.**
*(Covariance Control [4])
There exist constants such that*
[TABLE]
The above assumptions imply that the covariance elements along any row of and have bounded norms.
Having no access to the original data, the central machine finds the solution of the optimization problem (3) using received data from the local machines. The likelihood function in (2) depends on the samples via the sample covariance matrix . Thus, obtaining an appropriate approximate of the sample covariance matrix at the central machine would result in a good estimate of the underlying structure.
III-A Channel Model
There is a wireless channel between the local machines and the central node. In this paper, we assume that the channel can be modeled as a single-input multiple-output Gaussian multiple access channel (SIMO-GMAC) with additive white noise. The number of antennas at the receiver is assumed to be . All transmitters have equal transmit power which is denoted by . This implies, for all , the following constraint on the transmit symbols should be satisfied:
[TABLE]
where ’s are the channel inputs at the local machine . Denoting the transmit symbols by vector , the channel output is modeled by
[TABLE]
where is assumed to be an invertible complex matrix. In the fading environments, the channel gains are drawn from independent circularly symmetric complex Gaussian distribution. The additive noise is an independent circularly symmetric Gaussian vector with covaraience matrix .
IV Signs Method
We assume that each local machine applies a sign function on its local dataset to obtain binary data. More precisely, given samples at local machine , it obtains the signs dataset where . Then, it transmits the binary data to the central machine with the rate of bit per sample using a channel encoder and decoder.
Denoting the bit rate of the channel at local machine by , the achievable bit rates for all machines are characterized by [20]
[TABLE]
where is the sub-matrix of that includes the rows and columns indexed by , is the Hermitian of , is the identity matrix, and is the logarithm function in base 2. Throughout this section, we assume that , for all . Therefore, there exists a channel encoding to transmit 1 bit per sample at each local machine.
At the central machine, our goal is to solve the optimization problem (3) on the received binary data from all local machines. In order to obtain a solution that is close to the solution obtained by the original data, we seek a suitable approximation for the sample covariance matrix in (3). Thus, our goal is to estimate the sample covariance matrix as accurate as possible using the received signs data. In this section, we propose an estimator for the sample covariance matrix and theoretically show its error decreases exponentially by the sample size .
Let and be jointly normal with zero means, unit variances and correlation coefficient . If and be the corresponding sign variables, then the joint probability mass function (pmf) of and is given by [21]
[TABLE]
where and given by
[TABLE]
The equation (14) can be rewritten as
[TABLE]
Thus, by proposing an estimator for , using (15) we can obtain an estimator for . The following estimator for is optimal in the sense that it is unbiased and has minimum variance (UMVE) [22],
[TABLE]
where is the indicator function. is indeed a binomial random variable with success probability of . By substituting in (15), we use
[TABLE]
as an estimator for . Although is indeed biased, it is a consistent estimator for . Following lemma gives an error bound on this estimator.
Lemma 1**.**
Let and be two jointly normal variables with zero means, unit variances and correlation coefficient . Then, for the estimator (17), we have
[TABLE]
where .
Proof.
Since the function is a 1-Lipschitz function, i.e. , we have
[TABLE]
Since is sum of independent Bernoulli random variables, applying the Hoeffding inequality yields
[TABLE]
which completes the proof. ∎
Lemma 1 shows that the error of proposed estimator is controlled by the number of samples exponentially. Using this estimator, we can obtain an estimator for the sample covariance matrix as follows
[TABLE]
where , and function is applied on the input matrix element wise, i.e. . By substituting the above sample covariance matrix into (3), we can solve the regularized maximum likelihood problem.
Note that the sample covariance matrix defined in (19) is not necessarily positive semi-definite. But, it does not affect the convexity and uniqueness solution of (3). Ravikumar et al. [4] proved that the problem (3) is convex and has a unique solution for any sample covariance matrix with strictly positive diagonal elements which holds for in (19).
By incorporating Lemma 1 and the theorems 1 and 2 in [4], we can conclude that the precision matrix obtained by solving (3) with the sample covariance matrix in (28), recovers the true structure with high probability. Moreover, the proposed method correctly recovers the signs of the edges with high probability.
More precisely, the event indicates that and do agree on the zero entries and for the nonzero entries they have the same sign. Theorems 1 and 2 state that the event occurs with high probability. Before stating the theorems we should define some properties of the underlying GGM. We denote the maximum degree of the underlying graph structure by . The minimum absolute value of the edges weighs in the precision matrix is denoted by which is
[TABLE]
Theorem 1**.**
Consider a normal distribution satisfying the incoherence Assumption 1 and 2 with parameter . Let be the solution of the log-determinant program (3) with sample covariance in (19) and regularization parameter for some . Then,
- (a)
If the sample size is lower bounded as
[TABLE]
where
[TABLE]
then with probability at least , the edge set specified by is a subset of the true edge set. 2. (b)
If the sample size satisfies the lower bound
[TABLE]
where
[TABLE]
then,
[TABLE]
Remark 1**.**
Note that the proposed sign method is applicable for any channel with capacity greater than or equal to 1 bit.
V Uncoded Method
In this section, we assume that each local machine puts its local data into the channel without any source or channel coding. The central machine estimates the underlying graph structure using received data from the channel. At the central machine, no source or channel decoding is used. It infers the structure directly from the output samples of the channel.
As described in Section III-A, we consider a SIMO-GMAC. Each local machine can transmit two consequent samples by each channel use. More precisely, denoting the channel input symbol by , each local machine can put two consequent samples as real and imaginary parts of the input symbol. Therefore, at the central machine, vectors are received that each one is -dimensional.
In this way, the equation (11) can be decomposed as
[TABLE]
where . Hence, it can be rewritten in a block-matrix form as
[TABLE]
where are real matrices, and all the real and imaginary part vectors are -dimensional. In this way, two source samples can be transmitted per channel use: a sample is put into the real part and the other is put into the imaginary part. In particular, , where and are two independent samples from the source. In this way, the transmit power constraints are satisfied.
The central machine estimates the conditional dependencies of the vectors using the received vectors .
V-A Approximating the Sample Covariance
Defining matrix , , and . When transmitting samples through the channel, if we put two independent samples as real and imaginary parts of the vector , then the covariance matrix of is
[TABLE]
On the other hand, according to equation (25), we have
[TABLE]
where and is the covariance matrix of and , respectively. By substituting the sample covariance matrix of into the above expression, we obtain an approximation for the sample covariance matrix of , as
[TABLE]
Lemma 2**.**
Given i.i.d. samples of the vector . Then, for the sample covariance matrix in (28) we have
[TABLE]
where
[TABLE]
and is the minimum eigenvalue of .
Proof.
We define the random variable as follows
[TABLE]
It is clear that , where . Denoting , is a standard normal variable which is sub-Gaussian with parameter . Thus, according to Lemma 1 in [4], we have
[TABLE]
where is the sample covariance over samples. On the other hand, we have
[TABLE]
By finding an upper bound on , we can obtain an upper bound on the above probability.
[TABLE]
By decomposing as , we have
[TABLE]
By combining the above bound and (32), and substituting into (31) we obtain the claimed bound in the lemma. ∎
Next, we define the approximate sample covariance matrix of as
[TABLE]
where is a zero matrix.
Lemma 3**.**
For defined in (33), we have
[TABLE]
Proof.
From (33), it is clear that
[TABLE]
Thus, we have
[TABLE]
where the last inequality is obtained by the error bound of Lemma 2 for the sample size . ∎
Note that the matrix in (33) is not necessarily positive semi-definite. But this does not affect the convexity of the optimization problem (3).
By substituting from (33) into the (3), we can solve the regularized maximum likelihood problem and obtain a sparse solution for the precision matrix . Similar to Theorem 1, we can guarantee that Uncoded method can correctly recover the underlying graph structure with high probability.
Theorem 2**.**
Consider a normal distribution satisfying the incoherence Assumption 1 and 2 with parameter . Let be the solution of the log-determinant program (3) with sample covariance in (33) and regularization parameter for some .
- (a)
If the sample size is lower bounded as
[TABLE]
where
[TABLE]
then with probability at least , the edge set specified by is a subset of the true edge set. 2. (b)
If the sample size satisfies the lower bound
[TABLE]
where
[TABLE]
then,
[TABLE]
Remark 2**.**
Since the channel has real and imaginary parts, 2 samples can be transmitted by each channel access (i.e. each local machine can transmit samples by channel uses). Therefore, if the sample generation rate at the source is less than or equal to twice of the channel’s rate, the machines can transmit all samples without any sample loss.
VI Experiments
In this section, the performance of our proposed methods is evaluated by performing several experiments. In our simulations111The source code is available at https://github.com/ArminKaramzade/distributed-sparse-GGM., the glasso package [19] is used to solve -regularized MLE of the precision matrix. This package is based on the block coordinate descent algorithm proposed by [6].
In our experiments, a sparse random precision matrix is generated as follows. First, we generate a random sparse graph with a fixed probability of the edge presence, say , and also set its maximum node degree to . Then, we choose edge weights uniformly in for the symmetric precision matrix . Next, we make it positive definite matrix by adding a scaled identity matrix. Finally, we normalize the precision matrix to set the variances to . Also, we ensure that the generated matrix satisfies Assumption 1.
We employ the True Positive and False Positive Rates (TPR and FPR, respectively) as our performance measures. TPR is defined as the percentage of the predicted edges (non-zero off-diagonal entries in the precision matrix) that correctly detected. Similarly, FPR is the percentage of the predicted non-edges (zero entries in the precision matrix) that incorrectly detected.
We have experimentally observed that and , where , , and are the best regularization parameter for the signs, uncoded, and original data, respectively.
In order to have a fair comparison between Signs and Uncoded methods, we have used identical parameters for the channel. More precisely, we assume which ensures all the local machines have identical bit rates. We set the bit rate of each local machine, i.e. , to 2 bits. Thus, in the proposed methods, each local machine can transmit bits by channel uses. According to (12), in order to achieve the bit-rate of 2, the signal to noise ratio (SNR) should set to 3 (i.e. ).
In the first experiment, we evaluate the performance of our methods with respect to the dimension . Figure 2 shows TPR and FPR as a function of for sample sizes and . In this experiment, the error curves are averaged over 20 different random graphs and for each graph the TPR and FPR are averaged over 10 different random samples. As can be seen from Figure 2, Signs method outperforms Uncoded method. However, all three methods have approximately the same FPR.
Figure 3 reflects the performance of the methods as a function of the sample size for and . As can be seen, by increasing the sample size , the performance of all methods increases. In this experiment again Signs method outperforms the uncoded scheme.
In Figure 4, the probability of perfect structure recovery for a star-shaped graph is depicted. In this experiment, the underlying star graph consists of nodes. The precision matrix is generated as the inverse of a covariance matrix with for all which satisfies Assumption 1. The probability of perfect recovery is estimated by running the proposed methods 100 times and counting the number of times that the structure is recovered exactly. As can be seen from the figure, all methods recover the structure exactly for large enough sample sizes as claimed by theorems 1 and 2.
In Figure 5, we measure the TPR of Uncoded method for different values of the SNR. The experiment is performed on a random graph with nodes with maximum degree of and . In this experiment, we have generated the channel matrix with entries drawn from i.i.d. standard normal samples. The TPR curve is averaged over 100 different channel matrices. As can be seen from Figure 5, for SNR greater than 5, the performance of Uncoded method is very close to the TPR of the original data.
The FPR curve is not plotted, since the error values were negligible even for small SNR.
VII Conclusion
In this paper, we have studied the sparse structure learning of GGMs where the data are distributed across multiple local machines. Two methods are proposed to send information from the local machines to the central machine, namely, Signs and Uncoded methods. We have analytically and experimentally shown that the central machine can recover the underlying graph if large enough sample sizes are transmitted to the central machine.
Our experiments show that, under the same conditions, Signs method outperforms the uncoded scheme. Both methods have small FPR which is close to the FPR obtained by the original data.
VIII Acknowledgment
We thank Amir Najafi and Amir-Hossein Saberi for their valuable comments that greatly improved the manuscript.
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[1] L. E. Chai, S. K. Loh, S. T. Low, M. S. Mohamad, S. Deris, and Z. Zakaria, “A review on the computational approaches for gene regulatory network construction,” Computers in biology and medicine , vol. 48, pp. 55–65, 2014.
- 2[2] M. Hecker, S. Lambeck, S. Toepfer, E. Van Someren, and R. Guthke, “Gene regulatory network inference: data integration in dynamic models—a review,” Biosystems , vol. 96, no. 1, pp. 86–103, 2009.
- 3[3] R. Xiang, J. Neville, and M. Rogati, “Modeling relationship strength in online social networks,” in Proceedings of the 19th international conference on World wide web . ACM, 2010, pp. 981–990.
- 4[4] P. Ravikumar, M. J. Wainwright, G. Raskutti, B. Yu et al. , “High-dimensional covariance estimation by minimizing ℓ 1 subscript ℓ 1 \ell_{1} -penalized log-determinant divergence,” Electronic Journal of Statistics , vol. 5, pp. 935–980, 2011.
- 5[5] S. Sojoudi, “Equivalence of graphical lasso and thresholding for sparse graphs,” The Journal of Machine Learning Research , vol. 17, no. 1, pp. 3943–3963, 2016.
- 6[6] O. Banerjee, L. E. Ghaoui, and A. d’Aspremont, “Model selection through sparse maximum likelihood estimation for multivariate gaussian or binary data,” Journal of Machine learning research , vol. 9, no. Mar, pp. 485–516, 2008.
- 7[7] C. Chow and C. Liu, “Approximating discrete probability distributions with dependence trees,” IEEE transactions on Information Theory , vol. 14, no. 3, pp. 462–467, 1968.
- 8[8] A. Anandkumar, V. Y. Tan, F. Huang, and A. S. Willsky, “High-dimensional gaussian graphical model selection: Walk summability and local separation criterion,” Journal of Machine Learning Research , vol. 13, no. Aug, pp. 2293–2337, 2012.
