arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.02064cs.LG
收藏

你的数据,我的模型:在联邦学习中学习谁真正有帮助

Your Data, My Model: Learning Who Really Helps in Federated Learning

Aalto University(阿尔托大学);York University(约克大学)

Shamsiiat Abdurakhmanova, Amirhossein Mohammadi, Yasmin SarcheshmehPour, Alexander Jung

AI总结:

针对联邦学习中确定最有益协作节点的挑战,提出一种保护隐私、模型无关且数据驱动的节点选择方法,通过评估单次梯度步进后的模型改善程度来选择协作者,并支持非参数模型扩展。

摘要:

许多重要的机器学习应用涉及设备网络(如可穿戴设备或智能手机),这些设备生成局部数据并训练个性化模型。一个关键挑战是确定哪些节点对协作最有益。我们提出了一种简单且保护隐私的方法,通过评估模型在使用另一设备数据进行单次梯度步进后的改善程度来选择相关协作者,且无需共享原始数据。该方法通过用非参数泛化替代梯度步进,自然地扩展到非参数模型。我们的方法为个性化联邦学习(PersFL)实现了模型无关、数据驱动的节点选择。

英文摘要:

Many important machine learning applications involve networks of devices-such as wearables or smartphones-that generate local data and train personalized models. A key challenge is determining which peers are most beneficial for collaboration. We propose a simple and privacy-preserving method to select relevant collaborators by evaluating how much a model improves after a single gradient step using another devices data-without sharing raw data. This method naturally extends to non-parametric models by replacing the gradient step with a non-parametric generalization. Our approach enables model-agnostic, data-driven peer selection for personalized federated learning (PersFL).

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

↑