vix.ing · top · new · best · stats · spec

BaPa: A Novel Approach of Improving Load Balance in Parallel Matrix Factorization for Recommender Systems

2020/05/25 by Ruixin Guo, Feng Zhang, Lizhe Wang +4
Computer Science · Decision Sciences · Engineering · #Recommender Systems and Techniques #Advanced Bandit Algorithms Research #Advanced MIMO Systems Optimization

paper · doi:10.1109/tc.2020.2997051

Abstract

A simplified approach to accelerate matrix factorization of big data is to parallelize it. A commonly used method is to divide the matrix into multiple non-intersecting blocks and concurrently calculate them. This operation causes the Load balance problem, which significantly impacts parallel performance and is a big concern. A general belief is that the load balance across blocks is impossible by balancing rows and columns separately. We challenge the belief by proposing an approach of “Balanced Partitioning (BaPa)”. We demonstrate under what circumstance independently balancing rows and columns can lead to the balanced intersection of rows and columns, why, and how. We formally prove the feasibility of BaPa by observing the variance of rating numbers across blocks, and empirically validate its soundness by applying it to two standard parallel matrix factorization algorithms, DSGD and CCD++. Besides, we establish a mathematical model of “Imbalance Degree” to explain further why BaPa works well. BaPa is applied to synchronous parallel matrix factorization, but as a general load balance solution, it has significant application potential.

Citations

Related