Decentralized Rank-Adaptive Matrix Factorization—Part II: Convergence Analysis

Yuchen Jiao,Yuantao Gu,Tsung-Hui Chang,Zhi-Quan (Tom) Luo
DOI: https://doi.org/10.1109/tsp.2024.3465049
IF: 4.875
2024-01-01
IEEE Transactions on Signal Processing
Abstract:The matrix factorization (MF) model has a wide range of applications in signal processing and machine learning. Existing decentralized MF methods require to know the matrix rank a prior, which however is difficult to obtain especially when the data are distributively stored in a network. In the two-part paper, we study a rank-adaptive MF algorithm which proceeds in decentralized setting and meanwhile does not need to know the matrix rank precisely. In the Part-I paper, we have proposed to achieve the rank adaption through a novel l1-norm regularizer, and demonstrated its efficacy via numerical experiments. In this Part-II paper, our goal is to build the convergence conditions and the convergence rate of the proposed rank-adaptive algorithm. In particular, we first consider the decentralized MF algorithm with known rank and show that when the step size is set as $O(t^{-\delta})$ for $\delta \in (1/2, 2/3]$ , the algorithm converges to the global optima with rate $O(t^{-\delta})$ with high probability, whereas when the step-size is a constant, the algorithm converges with a linear rate, but only to the neighborhood of the global optima. Then we extend the analysis to the rank-adaptive algorithm.
What problem does this paper attempt to address?