Communication Compression for Distributed Nonconvex Optimization

• 成果类型: Article
• 署名作者: Yi, Xinlei; Zhang, Shengjun; Yang, Tao; Chai, Tianyou; Johansson, Karl Henrik
• 署名单位: Royal Institute of Technology; Northeastern University - China
• 刊物名称: IEEE TRANSACTIONS ON AUTOMATIC CONTROL
• ISSN/ISSBN: 0018-9286
• DOI: 10.1109/TAC.2022.3225515
• 发表日期: 2023
• 页码: 5477-5492
• 关键词: Communication compression; distributed optimization; linear convergence; Nonconvex Optimization; Polyak-Lojasiewicz (P-L) condition

摘要

In this article, we consider distributed non-convex optimization with the cost functions being distributed over agents. Noting that information compression is a key tool to reduce the heavy communication load for distributed algorithms as agents iteratively communicate with neighbors, we propose three distributed primal-dual algorithms with compressed communication. The first two algorithms are applicable to a general class of compressors with bounded relative compression error and the third algorithm is suitable for two general classes of compressors with bounded absolute compression error. We show that the proposed distributed algorithms with compressed communication have comparable convergence properties as state-of-the-art algorithms with exact communication. Specifically, we show that they can find first-order stationary points with sublinear convergence rate O(1/T) when each local cost function is smooth, where T is the total number of iterations, and find global optima with linear convergence rate under an additional condition that the global cost function satisfies the Polyak-Lojasiewicz condition. Numerical simulations are provided to illustrate the effectiveness of the theoretical results.