arXiv:2508.20295cs.LG2025-08Conference of the …被引 1

通过局部聚合提升联邦学习中的表示微调效率与稳定性。

FedReFT: Federated Representation Fine-Tuning with All-But-Me Aggregation

  • 在联邦学习中直接调整隐藏表示,仅更新少量参数以降低开销。
  • 在多个基准上表现优于主流方法,参数效率提升1至49倍。
  • 采用排除自身的聚合策略,有效缓解任务异构带来的语义错位问题。

参数高效微调(PEFT)通过仅更新少量参数来适应大型预训练模型。近年来,表示微调(ReFT)作为一种新范式兴起,将微调重点从模型权重转向捕捉丰富语义信息的隐藏表示,在独立设置下表现优于现有先进PEFT方法。然而,其在联邦学习(FL)中的应用面临客户端数据分布、模型能力与计算资源异构的挑战。为此,我们提出联邦表示微调(FedReFT),通过稀疏干预层直接调控客户端的隐藏表示,实现轻量级且语义丰富的微调,适用于边缘设备。然而,表示级更新在任务异构下易受聚合不匹配影响,简单平均可能导致语义对齐破坏。为此,我们提出全然除外自身(All-But-Me, ABM)聚合机制,使每个客户端接收其他客户端的聚合更新并部分融合,从而在保持本地个性化的同时稳定获取全局知识。我们进一步设计一种受测试时计算(TTC)启发的自适应更新策略,在异构条件下平衡本地与全局贡献。FedReFT在常识推理、算术推理和GLUE基准上达到领先性能,同时相比主流LoRA方法实现1-49倍更高的参数效率。

原文摘要 · Abstract (English)

Parameter-efficient fine-tuning (PEFT) adapts large pre-trained models by updating only a small subset of parameters. Recently, Representation Fine-Tuning (ReFT) has emerged as an effective alternative. ReFT shifts the fine-tuning paradigm from updating model weights to directly manipulating hidden representations that capture rich semantic information, and outperforms state-of-the-art PEFTs in standalone settings. However, its application in Federated Learning (FL) remains challenging due to heterogeneity in clients' data distributions, model capacities, and computational resources. To address these challenges, we introduce Federated Representation Fine-Tuning (FedReFT), a novel approach to fine-tune clients' hidden representations. FedReFT applies sparse intervention layers to steer hidden representations directly, offering a lightweight and semantically rich fine-tuning alternative ideal for edge devices. However, representation-level updates are especially vulnerable to aggregation mismatch under different task heterogeneity, where naive averaging can corrupt semantic alignment. To mitigate this issue, we propose All-But-Me (ABM) aggregation, where each client receives the aggregated updates of others and partially incorporates them, enabling stable and personalized learning by balancing local focus with global knowledge. We further design an adaptive update strategy inspired by Test-Time Computing (TTC) to balance local and global contributions under heterogeneous conditions. FedReFT achieves state-of-the-art performance on commonsense reasoning, arithmetic reasoning, and GLUE benchmarks, while delivering 1-49 times higher parameter efficiency compared to leading LoRA-based methods.

联邦学习表示微调参数效率边缘计算

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