arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.03383cs.LGcs.DC

非IID数据下的去中心化学习动态拓扑优化

Dynamic Topology Optimization for Non-IID Data in Decentralized Learning

Delft University of Technology(代尔夫特理工大学)

Bart Cox, Antreas Ioannou, Jérémie Decouchant

AI总结:

Morph通过动态拓扑优化提升非IID数据下去中心化学习的准确率和稳定性,实现更快收敛和更少通信轮次。

摘要:

去中心化学习(DL)使一组节点能够协作训练模型,而无需中央协调,这在隐私和可扩展性方面具有优势。然而,当数据分布是非独立和同分布(非IID)且通信拓扑静态时,DL难以训练高精度模型。为解决这些问题,我们提出了Morph,一种用于DL的拓扑优化算法。在Morph中,节点根据最大模型差异性自适应地选择同行进行模型交换。Morph保持固定的入度,通过基于谣言的同行发现和多样性驱动的邻居选择动态重塑通信图,从而提高对数据异质性的鲁棒性。在CIFAR-10和FEMNIST上进行的实验表明,Morph在100个节点的情况下始终优于静态和流行基线,同时接近完全连接的上界。在CIFAR-10上,Morph相比最先进的基线在测试准确率上实现了1.12倍的相对提升。在FEMNIST上,Morph的准确率比流行学习高1.08倍。相似的趋势在50个节点部署中也成立,其中Morph在CIFAR-10上将与完全连接上界之间的差距缩小到0.5个百分点。这些结果表明,Morph在最终准确率、收敛速度和稳定性(通过较低的节点间方差量化)方面均优于基线,同时所需的通信轮次更少,且不需要全局知识。

英文摘要:

Decentralized learning (DL) enables a set of nodes to train a model collaboratively without central coordination, offering benefits for privacy and scalability. However, DL struggles to train a high accuracy model when the data distribution is non-independent and identically distributed (non-IID) and when the communication topology is static. To address these issues, we propose Morph, a topology optimization algorithm for DL. In Morph, nodes adaptively choose peers for model exchange based on maximum model dissimilarity. Morph maintains a fixed in-degree while dynamically reshaping the communication graph through gossip-based peer discovery and diversity-driven neighbor selection, thereby improving robustness to data heterogeneity. Experiments on CIFAR-10 and FEMNIST with up to 100 nodes show that Morph consistently outperforms static and epidemic baselines, while closely tracking the fully connected upper bound. On CIFAR-10, Morph achieves a relative improvement of 1.12x in test accuracy compared to the state-of-the-art baselines. On FEMNIST, Morph achieves an accuracy that is 1.08x higher than Epidemic Learning. Similar trends hold for 50 node deployments, where Morph narrows the gap to the fully connected upper bound within 0.5 percentage points on CIFAR-10. These results demonstrate that Morph achieves higher final accuracy, faster convergence, and more stable learning as quantified by lower inter-node variance, while requiring fewer communication rounds than baselines and no global knowledge.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑