arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.00569cs.LGcs.AI
收藏

重新定义贡献:Shapley驱动的联邦学习

Redefining Contributions: Shapley-Driven Federated Learning

Mohamed bin Zayed University of Artificial Intelligence (MBZUAI)(穆罕默德·本·扎耶德人工智能大学(MBZUAI))

Nurbek Tastan, Samar Fares, Toluwani Aremu, Samuel Horvath, Karthik Nandakumar

AI总结:

本文提出ShapFed方法,利用Shapley值对联邦学习中参与者贡献进行细粒度评估,并引入ShapFed-WA加权聚合方法,在类别不平衡场景下提升模型效用、效率和公平性。

摘要:

联邦学习(FL)已成为机器学习中的关键方法,使多个参与者能够在不共享原始数据的情况下协作训练全局模型。尽管FL在医疗和金融等多个领域得到应用,但当参与者贡献不均或不够诚实时,确保全局模型收敛具有挑战性。为克服这一挑战,需要有原则的机制来评估FL环境中个体参与者的贡献。现有的贡献评估解决方案依赖于一般准确率评估,往往无法捕捉细微的动态和特定类别的影响。本文提出了一种名为ShapFed的新型贡献评估方法,用于对FL中参与者贡献进行细粒度评估。我们的方法利用合作博弈论中的Shapley值,提供对特定类别影响的精细理解。基于ShapFed,我们引入了一种名为ShapFed-WA的加权聚合方法,其性能优于传统的联邦平均,尤其是在类别不平衡场景中。根据参与者的贡献个性化其更新,通过提供与参与者贡献相称的差异化模型,进一步增强了协作公平性。在CIFAR-10、Chest X-Ray和Fed-ISIC2019数据集上的实验证明了我们的方法在提升FL系统的效用、效率和公平性方面的有效性。代码可在https://github.com/tnurbek/shapfed找到。

英文摘要:

Federated learning (FL) has emerged as a pivotal approach in machine learning, enabling multiple participants to collaboratively train a global model without sharing raw data. While FL finds applications in various domains such as healthcare and finance, it is challenging to ensure global model convergence when participants do not contribute equally and/or honestly. To overcome this challenge, principled mechanisms are required to evaluate the contributions made by individual participants in the FL setting. Existing solutions for contribution assessment rely on general accuracy evaluation, often failing to capture nuanced dynamics and class-specific influences. This paper proposes a novel contribution assessment method called ShapFed for fine-grained evaluation of participant contributions in FL. Our approach uses Shapley values from cooperative game theory to provide a granular understanding of class-specific influences. Based on ShapFed, we introduce a weighted aggregation method called ShapFed-WA, which outperforms conventional federated averaging, especially in class-imbalanced scenarios. Personalizing participant updates based on their contributions further enhances collaborative fairness by delivering differentiated models commensurate with the participant contributions. Experiments on CIFAR-10, Chest X-Ray, and Fed-ISIC2019 datasets demonstrate the effectiveness of our approach in improving utility, efficiency, and fairness in FL systems. The code can be found at https://github.com/tnurbek/shapfed.

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

↑