arXiv:2606.06154cs.AI2026-06中稿 · International Work…

用超网络生成LoRA参数,让联邦学习更高效精准

Amortizing Federated Adaptation: Hypernetwork Driven LoRA for Personalized Foundation Models

论文配图:Amortizing Federated Adaptation: Hypernetwork Driven LoRA for Personalized Foundation Models
图 1 · 摘自论文原文
  • 用超网络根据客户端特征生成LoRA初始化,避免重复训练
  • 在低秩乘积空间直接聚合更新,解决传统平均的偏差问题
  • 适合个性化模型需求高、数据分布不均的联邦学习场景

使用低秩适应(LoRA)进行联邦微调可实现高效分布式学习。然而现有方法存在两大局限:(1)结构聚合偏差,独立平均低秩因子无法逼近真实联合更新;(2)客户端初始化滞后,每轮通信都需重置LoRA参数,影响收敛速度。本文提出HyperLoRA,通过超网络驱动的LoRA生成与乘积空间聚合,实现参数化适应的摊销。服务器端引入学习型聚合模块,在低秩乘积空间中直接合成更新,消除因子级平均带来的不一致。轻量级残差修正模块进一步提升非独立同分布(non-IID)数据下的稳定性。相比迭代优化和启发式平均,该方法实现高效个性化、无偏聚合与更快收敛。在联邦视觉与视觉-语言基准测试中,HyperLoRA展现出更快收敛速度、更强分布漂移鲁棒性及更优个性化性能。

原文摘要 · Abstract (English)

Federated fine-tuning of foundation models using Low-Rank Adaptation (LoRA) offers a communication efficient solution for distributed learning. However, existing federated LoRA methods suffer from two fundamental limitations: (1) structural aggregation bias, where independently averaging low rank factors fails to approximate the true combined update, and (2) client side initialization lag, as clients repeatedly reinitialize LoRA parameters across communication rounds, slowing convergence. We propose HyperLoRA, a unified framework that addresses both issues through amortized federated adaptation through hypernetwork-driven LoRA generation and product space aggregation. Instead of iterative per-client optimization, HyperLoRA employs a learned generator that maps client distribution signatures to LoRA initializations, effectively amortizing per client adaptation. On the server side, we introduce a learned aggregation module that directly synthesizes updates in the low-rank product space, eliminating the inconsistencies of factor-wise averaging. A lightweight residual correction module further improves stability under heterogenous (non-IID) client distributions.By replacing iterative optimization and heuristic averaging with learned operators, HyperLoRA jointly enables efficient personalization, unbiased aggregation, and faster convergence. Experiments on federated vision and vision-language benchmarks show that HyperLoRA achieves improved convergence speed, greater robustness to distribution shift, and stronger personalization performance compared to prior federated LoRA methods.

联邦学习LoRA超网络个性化

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