arXiv:2505.12318cs.LG2025-05被引 1

提出新方法实现高效联邦持续微调,避免遗忘与参数膨胀。

Task-agnostic Low-rank Residual Adaptation for Efficient Federated Continual Fine-Tuning

  • 用通用低秩残差模块持续更新共享模型,不随任务增加参数。
  • 在四个数据集上优于基线,通信与计算开销大幅降低。
  • 适合长期迭代、新增类别的联邦学习场景,如医疗或边缘设备。

联邦参数高效微调(Fed-PEFT)通过仅更新少量参数,实现大模型在联邦学习中的轻量级适配。然而,现有方法通常假设标签空间固定、任务静态,难以应对真实场景中客户端随时间持续遇到新类别的情况,由此产生新兴问题——联邦持续微调(FCFT)。在FCFT中,客户端需协作对预训练模型进行序列化任务微调,每个客户端随时间观察互不重叠的新类别,且推理时无法获知任务身份。该问题挑战在于:非独立同分布的客户端数据导致严重遗忘,任务专用模块引发参数增长与任务特异性推理,以及异构客户端间的聚合不一致。为此,本文提出联邦无任务依赖低秩残差适配(Fed-TaLoRA),一种基于无任务依赖适配、后聚合模型校准和策略性低秩适配位置放置的高效FCFT新方法。Fed-TaLoRA通过持续微调单一共享模块避免任务级参数增长,并引入理论支持的残差权重更新机制,校准聚合后的全局模型,提升聚合保真度。我们对Fed-TaLoRA的收敛性与聚合行为提供了理论分析。在四个基准数据集上的大量实验表明,该方法持续优于强基线,同时显著降低通信与计算成本。

原文摘要 · Abstract (English)

Federated Parameter-Efficient Fine-Tuning (Fed-PEFT) enables lightweight adaptation of large pre-trained models in federated learning settings by updating only a small subset of parameters. However, Fed-PEFT methods typically assume a fixed label space and static downstream tasks, which is restrictive in realistic application scenarios where clients continuously encounter new classes over time. This leads to an emerging problem, known as \emph{Federated Continual Fine-Tuning} (FCFT). In FCFT, clients collaboratively fine-tune a pre-trained model over a sequence of tasks, where each client observes disjoint sets of new classes over time, and task identity is unavailable at inference time. FCFT is challenging because it simultaneously suffers from severe forgetting under non-IID client data distributions, parameter growth and task-specific inference caused by task-wise modules, and aggregation inconsistency across heterogeneous clients. To address these challenges, we propose Federated Task-agnostic Low-rank Residual Adaptation (Fed-TaLoRA), a novel approach for efficient FCFT built on task-agnostic adaptation, post-aggregation model calibration, and strategic low-rank adaptation placement. Fed-TaLoRA continuously fine-tunes a single shared module across sequential tasks to avoid task-wise parameter growth, and further introduces a theoretically grounded residual weight update mechanism to calibrate the aggregated global model and improve aggregation fidelity. We provide a theoretical analysis of the convergence and aggregation behavior of Fed-TaLoRA. Extensive experiments on four benchmark datasets demonstrate that Fed-TaLoRA consistently outperforms strong baselines while reducing communication and computation costs significantly.

联邦学习持续学习低秩适配

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。