摘要
arXiv:2502.07750v1 Announce Type: cross
摘要:分布式学习(DL)能够在多个设备上训练机器学习模型,但面临着非同态数据分布和设备能力差异等挑战,这些挑战会妨碍训练效率。传统的联邦学习(FL)设置中,通信瓶颈进一步复杂化了这些问题。为了解决这些问题,我们提出了个性化去中心化选择训练的联邦学习(PFedDST)框架。PFedDST 通过允许设备根据综合通信得分战略性地评估和选择同伴来增强模型训练。该得分结合了损失、任务相似性和选择频率,确保最佳的同伴连接。这种选择策略旨在增加本地个性化,并促进有益的同伴合作,以增强训练过程的稳定性和效率。我们的实验表明,PFedDST 不仅提高了模型准确性,还加速了收敛。该方法在处理数据异质性方面优于现有方法,能够在多种分散系统中实现更快、更有效的训练。