arXiv:2411.00365cs.LG2024-11被引 1

用博弈论权重优化去中心化学习,提升模型鲁棒性

ROSS: RObust decentralized Stochastic learning based on Shapley values

  • 基于谢林值为邻居梯度加权,动态评估贡献
  • 理论证明线性收敛速度,实验验证高效稳定
  • 适合数据异构、含噪声或恶意数据场景

在去中心化学习中,各智能体协同使用分布式数据训练全局模型,但面临数据分布异质性挑战,如非独立同分布、含噪声或被污染数据。本文提出一种基于谢林值的鲁棒去中心化随机学习算法 ROSS。每轮中,各智能体聚合邻接智能体的交叉梯度信息(即自身局部模型对邻居数据的梯度),并以谢林值衡量其贡献进行加权更新,类似动量机制。理论分析表明,该算法具有线性收敛速度。大量实验证明,在面对数据多样性挑战时,相比现有最先进方法,ROSS 在收敛速度和预测准确率上均具显著优势。

原文摘要 · Abstract (English)

In the paradigm of decentralized learning, a group of agents collaborate to learn a global model using a distributed dataset without a central server; nevertheless, it is severely challenged by the heterogeneity of the data distribution across the agents. For example, the data may be distributed non-independently and identically, and even be noised or poisoned. To address these data challenges, we propose ROSS, a novel robust decentralized stochastic learning algorithm based on Shapley values, in this paper. Specifically, in each round, each agent aggregates the cross-gradient information from its neighbors, i.e., the derivatives of its local model with respect to the datasets of its neighbors, to update its local model in a momentum like manner, while we innovate in weighting the derivatives according to their contributions measured by Shapley values. We perform solid theoretical analysis to reveal the linear convergence speedup of our ROSS algorithm. We also verify the efficacy of our algorithm through extensive experiments on public datasets. Our results demonstrate that, in face of the above variety of data challenges, our ROSS algorithm has significant advantages over existing state-of-the-art proposals in terms of both convergence and prediction accuracy.

去中心化学习谢林值鲁棒性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。