Communication-Efficient Distributed SVD via Local Power Iterations
Abstract
We study distributed computing of the truncated singular value decomposition problem. We develop an algorithm that we call \texttt{LocalPower} for improving communication efficiency. Specifically, we uniformly partition the dataset among nodes and alternate between multiple (precisely ) local power iterations and one global aggregation. In the aggregation, we propose to weight each local eigenvector matrix with orthogonal Procrustes transformation (OPT). As a practical surrogate of OPT, sign-fixing, which uses a diagonal matrix with entries as weights, has better computation complexity and stability in experiments. We theoretically show that under certain assumptions \texttt{LocalPower} lowers the required number of communications by a factor of to reach a constant accuracy. We also show that the strategy of periodically decaying helps obtain high-precision solutions. We conduct experiments to demonstrate the effectiveness of \texttt{LocalPower}.
Keywords
Cite
@article{arxiv.2002.08014,
title = {Communication-Efficient Distributed SVD via Local Power Iterations},
author = {Xiang Li and Shusen Wang and Kun Chen and Zhihua Zhang},
journal= {arXiv preprint arXiv:2002.08014},
year = {2021}
}
Comments
9 pages, 7 figures, accepted by 2021 ICML