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

FedSelect:通过定制化选择参数进行微调的个性化联邦学习

FedSelect: Personalized Federated Learning with Customized Selection of Parameters for Fine-Tuning

Lapis Labs(拉皮斯实验室);University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);Bar-Ilan University(巴伊兰大学)

Rishub Tamirisa, Chulin Xie, Wenxuan Bao, Andy Zhou, Ron Arel, Aviv Shamsian

AI总结:

本文提出FedSelect,一种借鉴彩票假设迭代子网络发现的个性化联邦学习算法,通过增量扩展个性化子网络并全局聚合其余参数,在数据异质性和分布偏移下优于现有PFL方法。

摘要:

当客户端数据分布具有足够的异质性时,标准联邦学习方法会受到影响。近期方法通过个性化联邦学习(PFL)解决客户端数据异质性问题;PFL是一类联邦学习算法,旨在将学习到的全局知识个性化,以更好地适配客户端的本地数据分布。现有PFL方法通常通过在特定层(即分类器头)上执行个性化、并对网络其余部分进行全局聚合,来解耦深度神经网络中的全局更新。然而,预先选择网络层进行个性化可能导致全局知识的次优存储。在本文中,我们提出FedSelect,这是一种受彩票假设所用迭代子网络发现过程启发的新型PFL算法。FedSelect增量扩展子网络以个性化客户端参数,同时对其余参数执行全局聚合。该方法能够在训练过程中同时实现客户端参数和子网络结构的个性化。最后,我们表明FedSelect在具有挑战性的客户端数据异质性设置下优于近期最先进的PFL算法,并对各种真实世界分布偏移表现出鲁棒性。我们的代码可在https://github.com/lapisrocks/fedselect获取。

英文摘要:

Standard federated learning approaches suffer when client data distributions have sufficient heterogeneity. Recent methods addressed the client data heterogeneity issue via personalized federated learning (PFL) - a class of FL algorithms aiming to personalize learned global knowledge to better suit the clients' local data distributions. Existing PFL methods usually decouple global updates in deep neural networks by performing personalization on particular layers (i.e. classifier heads) and global aggregation for the rest of the network. However, preselecting network layers for personalization may result in suboptimal storage of global knowledge. In this work, we propose FedSelect, a novel PFL algorithm inspired by the iterative subnetwork discovery procedure used for the Lottery Ticket Hypothesis. FedSelect incrementally expands subnetworks to personalize client parameters, concurrently conducting global aggregations on the remaining parameters. This approach enables the personalization of both client parameters and subnetwork structure during the training process. Finally, we show that FedSelect outperforms recent state-of-the-art PFL algorithms under challenging client data heterogeneity settings and demonstrates robustness to various real-world distributional shifts. Our code is available at https://github.com/lapisrocks/fedselect.

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

↑