arXiv:2503.03920cs.LGcs.CL2025-03被引 9

针对异构数据,自动学习每客户端最优适配秩,实现高效个性化联邦微调。

Personalized Federated Fine-tuning for Heterogeneous Data: An Automatic Rank Learning Approach via Two-Level LoRA

  • 两级LoRA架构:共享适配层+个性化适配层,同时学习公共与个体特征。
  • 自适应确定秩:根据客户端数据特性动态调整适配器秩,提升适配精度。
  • 轻量高效:第二级适配参数极少,内存开销低,适合资源受限设备。

我们研究在语言模型背景下,客户端在不共享本地数据的前提下协同微调语言模型(如BERT、GPT)以实现个性化的问题。尽管现有方法已将低秩适配(LoRA)等参数高效微调技术应用于联邦设置,但通常采用预设最大/最小秩的单一或多个独立低秩适配器,难以适应不同客户端的异构数据。为此,我们提出PF2LoRA,一种基于新型两层LoRA的自动秩学习方法。在冻结预训练模型权重的前提下,该算法同时学习两个层级的适配:第一层为所有客户端共享的通用适配器,第二层用于各客户端个性化。关键优势在于,可依据每个客户端的数据自适应确定合适秩,而非依赖对数据异构性无感的预设秩。我们通过合成示例验证了其能自动学习各客户端的真实秩,精准匹配其数据特性。值得注意的是,该方法引入极小额外内存开销,因第二级适配参数远少于第一级。在自然语言理解与生成任务上的实验表明,PF2LoRA显著优于现有联邦微调方法。

原文摘要 · Abstract (English)

We study the task of personalized federated fine-tuning with heterogeneous data in the context of language models, where clients collaboratively fine-tune a language model (e.g., BERT, GPT) without sharing their local data, achieving personalization simultaneously. While recent efforts have applied parameter-efficient fine-tuning techniques like low-rank adaptation (LoRA) in federated settings, they typically use single or multiple independent low-rank adapters with predefined maximal and minimal ranks, which may not be optimal for diverse data sources over clients. To address this issue, we propose PF2LoRA, a new personalized federated fine-tuning algorithm built on a novel \emph{automatic rank learning approach via two-level LoRA}. Given the pretrained language model whose weight is frozen, our algorithm aims to learn two levels of adaptation simultaneously: the first level aims to learn a common adapter for all clients, while the second level fosters individual client personalization. A key advantage of PF2LoRA is its ability to adaptively determine a suitable rank based on an individual client's data, rather than relying on a predefined rank that is agnostic to data heterogeneity. We present a synthetic example that highlights how PF2LoRA automatically learns the ground-truth rank for each client, tailoring the adaptation to match the properties of their individual data. Notably, this approach introduces minimal additional memory overhead, as the second-level adaptation comprises a small number of parameters compared to the first level. Our experiments on natural language understanding and generation tasks demonstrate that PF2LoRA significantly outperforms existing federated fine-tuning methods.

联邦学习个性化LoRA自适应

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