Distributed Bayesian Matrix Decomposition for Big Data Mining and Clustering

Chihao Zhang,Yang Yang,Wei Zhang,Shihua Zhang
DOI: https://doi.org/10.48550/arXiv.2002.03703
2020-02-10
Abstract:Matrix decomposition is one of the fundamental tools to discover knowledge from big data generated by modern applications. However, it is still inefficient or infeasible to process very big data using such a method in a single machine. Moreover, big data are often distributedly collected and stored on different machines. Thus, such data generally bear strong heterogeneous noise. It is essential and useful to develop distributed matrix decomposition for big data analytics. Such a method should scale up well, model the heterogeneous noise, and address the communication issue in a distributed system. To this end, we propose a distributed Bayesian matrix decomposition model (DBMD) for big data mining and clustering. Specifically, we adopt three strategies to implement the distributed computing including 1) the accelerated gradient descent, 2) the alternating direction method of multipliers (ADMM), and 3) the statistical inference. We investigate the theoretical convergence behaviors of these algorithms. To address the heterogeneity of the noise, we propose an optimal plug-in weighted average that reduces the variance of the estimation. Synthetic experiments validate our theoretical results, and real-world experiments show that our algorithms scale up well to big data and achieves superior or competing performance compared to other distributed methods.
Machine Learning,Computer Vision and Pattern Recognition,Distributed, Parallel, and Cluster Computing
What problem does this paper attempt to address?
The problem that this paper attempts to solve is: How to efficiently perform matrix decomposition of large - scale data in a distributed system for big data mining and cluster analysis. Specifically, the paper proposes solutions to the following challenges: 1. **Low efficiency of single - machine processing**: Traditional matrix decomposition methods are inefficient or infeasible when processing very large data sets on a single machine. 2. **Data distributed storage**: Big data generated by modern applications is usually stored distributedly on different machines, and this data often has strong heterogeneous noise (that is, data from different sources has large differences in noise characteristics). 3. **Communication problems**: In a distributed system, the communication cost between nodes is high, and it is necessary to consider how to reduce communication overhead. To solve these problems, the author proposes a Distributed Bayesian Matrix Decomposition model (DBMD) and adopts three strategies to achieve distributed computing: - **Accelerated Gradient Descent (AGD)** - **Alternating Direction Method of Multipliers (ADMM)** - **Communication - Efficient Accurate Statistical Estimation (CEASE)** In addition, to deal with the heterogeneous noise problem, the author proposes an optimal weighted average method to improve the model performance by minimizing the estimated variance. ### Formula Representation The key formulas involved in the paper are as follows: - The objective function of matrix decomposition: \[ \min_{W, H_c} \sum_{c = 1}^C\frac{1}{2\sigma_c^2}\|X_c - WH_c\|_F^2+\lambda\|W\|_1-\sum_{c = 1}^C\sum_{k, j}(\alpha_k - 1)\ln(h_{kj}^{(c)})+\sum_{c = 1}^Cmn_c\ln\sigma_c \] where $\|X_c - WH_c\|_F^2$ represents the Frobenius norm, $\lambda\|W\|_1$ is the L1 - regularization term, and $\alpha_k$ is the parameter of the Dirichlet distribution. - The update formula of Accelerated Gradient Descent: \[ W_k=\arg\min_Wg(W)+\frac{L}{2}\left\|W-(Y_k-\frac{1}{L}\nabla\tilde{f}_c(Y_k))\right\|_F^2 \] \[ Y_{k + 1}=W_k+\frac{\nu_{k - 1}}{\nu_{k + 1}}(W_k - W_{k - 1}) \] - The update formula of ADMM: \[ W_c^{(k + 1)}=(X_cH_c^T+U_c^{(k)}/\rho+W^{(k)})(I_r+H_cH_c^T/\rho)^{-1} \] \[ W^{(k + 1)}=S_{\lambda/C\rho}(\bar{W}_c^{(k + 1)}-\bar{U}_c^{(k)}/\rho) \] - The update formula of CEASE: \[ W_c^{(k)}=\arg\min_W\tilde{f}_c(W)=f_c(W)-\langle\nabla f_c(W_k)-\nabla f(W_k), W\rangle+\frac{\gam