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

回顾更多:利用历史序列更新实现个性化联邦适配器微调

Look Back for More: Harnessing Historical Sequential Updates for Personalized Federated Adapter Tuning

Institute of High Performance Computing (IHPC), Agency for Science, Technology and Research (A*STAR)(新加坡科技研究局高性能计算研究所);EVYD Technology(EVYD科技公司)

Danni Peng, Yuan Wang, Huazhu Fu, Jinpeng Jiang, Yong Liu, Rick Siow Mong Goh, Qingsong Wei

AI总结:

本文提出pFedSeq框架,利用选择性状态空间模型对客户端历史适配器更新序列进行建模,以提升个性化联邦学习中适配器微调的性能。

摘要:

个性化联邦学习(PFL)研究有效的模型个性化,以解决传统联邦学习(FL)中客户端之间的数据异质性问题。现有的PFL方法主要仅依赖客户端最新更新的模型来生成个性化模型,而忽略了它们之前的更新,这可能导致个性化模型学习效果欠佳。为弥补这一不足,我们提出了一种名为pFedSeq的新框架,旨在个性化适配器以微调联邦学习中的基础模型。在pFedSeq中,服务器维护并训练一个序列学习器,该学习器处理来自客户端的过去适配器更新序列,并为个性化适配器生成校准。为有效捕捉先前更新中隐藏的跨客户端和跨步骤关系,并生成高性能的个性化适配器,pFedSeq采用强大的选择性状态空间模型(SSM)作为序列学习器的架构。通过在四个公开基准数据集上的大量实验,我们证明了pFedSeq优于最先进的PFL方法。

英文摘要:

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.

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

↑