利用历史更新信息提升联邦学习个性化模型性能
Look Back for More: Harnessing Historical Sequential Updates for Personalized Federated Adapter Tuning
- 通过序列学习捕捉客户端历史更新模式
- 在4个数据集上优于现有最优方法
- 适合关注个性化联邦学习的研究者
个性化联邦学习(PFL)旨在解决传统联邦学习中客户端间数据异构性问题。现有PFL方法主要依赖客户端最新更新模型,忽略历史更新,可能导致个性化效果不佳。为此,我们提出pFedSeq框架,用于在联邦学习中微调基础模型的适配器。在pFedSeq中,服务器维护并训练一个序列学习器,处理来自客户端的历史适配器更新序列,并生成个性化适配器的校准。为有效捕捉历史更新中隐藏的跨客户端和跨步关系,pFedSeq采用强大的选择性状态空间模型(SSM)作为序列学习器架构。在四个公开基准数据集上的大量实验表明,pFedSeq显著优于当前最先进的PFL方法。
原文摘要 · Abstract (English)
Personalized federated learning (PFL) studies effective model personalization to address the data heterogeneity issue among clients in traditional federated learning (FL). Existing PFL approaches mainly generate personalized models by relying solely on the clients' latest updated models while ignoring their previous updates, which may result in suboptimal personalized model learning. To bridge this gap, we propose a novel framework termed pFedSeq, designed for personalizing adapters to fine-tune a foundation model in FL. In pFedSeq, the server maintains and trains a sequential learner, which processes a sequence of past adapter updates from clients and generates calibrations for personalized adapters. To effectively capture the cross-client and cross-step relations hidden in previous updates and generate high-performing personalized adapters, pFedSeq adopts the powerful selective state space model (SSM) as the architecture of sequential learner. Through extensive experiments on four public benchmark datasets, we demonstrate the superiority of pFedSeq over state-of-the-art PFL methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。