对比两种分布式学习算法在不同网络下的表现,发现多路径随机游走更优。
A Tale of Two Learning Algorithms: Multiple Stream Random Walk and Asynchronous Gossip
- 提出多流异步随机游走算法,分析其计算、时间与通信效率
- 在长径图中迭代收敛更快,通信开销更低,尤其适合大规模网络
- 适用于数据异构性强、网络结构复杂的分布式学习场景
尽管基于广播和随机游走的分布式学习算法广为人知,但对其在不同图拓扑和数据异构性下的相对性能,理论与实验分析仍不充分。本文设计并分析了一种基于多路径随机游走的异步算法(Multi-Walk, MW),给出了其在迭代次数、实际运行时间和通信开销上的收敛性分析。同时对异步广播算法进行了系统性分析,填补了文献中缺乏全面收敛性与资源开销评估的空白。结果表明:在大直径图(如环形图)中,MW 的迭代收敛速度优于异步广播;在小直径图(如完全图)中,其性能取决于路径数与数据异构程度。在实际运行时间上,两者均表现出线性加速特性,分别随路径数和节点数提升。此外,除极端数据异构的小直径图外,MW 在通信开销上始终优于异步广播。这些发现揭示了两类算法在不同场景下的适用优势。代码已开源以支持复现。
原文摘要 · Abstract (English)
Although gossip and random walk-based learning algorithms are widely known for decentralized learning, there has been limited theoretical and experimental analysis to understand their relative performance for different graph topologies and data heterogeneity. We first design and analyze a random walk-based learning algorithm with multiple streams (walks), which we name asynchronous "Multi-Walk (MW)". We provide a convergence analysis for MW w.r.t iteration (computation), wall-clock time, and communication. We also present a convergence analysis for "Asynchronous Gossip", noting the lack of a comprehensive analysis of its convergence, along with the computation and communication overhead, in the literature. Our results show that MW has better convergence in terms of iterations as compared to Asynchronous Gossip in graphs with large diameters (e.g., cycles), while its relative performance, as compared to Asynchronous Gossip, depends on the number of walks and the data heterogeneity in graphs with small diameters (e.g., complete graphs). In wall-clock time analysis, we observe a linear speed-up with the number of walks and nodes in MW and Asynchronous Gossip, respectively. Finally, we show that MW outperforms Asynchronous Gossip in communication overhead, except in small-diameter topologies with extreme data heterogeneity. These results highlight the effectiveness of each algorithm in different graph topologies and data heterogeneity. Our codes are available for reproducibility.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。