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