arXiv:2607.18343cs.LGcs.AI2026-07

用小模型生成大模型权重,通信量压缩8718倍仍保持高精度。

Federated Lightweight Fine-Tuning

  • 用可训练潜变量生成权重,平均潜变量即平均权重,无需传完整模型。
  • 每轮仅传1280个浮点数(约5KB),比全量微调减少8718倍通信量。
  • 适合资源受限设备参与联邦学习,尤其在数据分布不均时表现更优。

联邦微调受通信瓶颈制约:FedAvg和伪梯度方法传输量随模型大小增长,梯度压缩仅降低常数倍。本文提出新思路:映射网络通过可训练潜变量与固定仿射投影生成权重,因映射共享且为仿射,潜变量平均等价于权重平均。为此改进两点:采用低秩、可种子再生的投影分解,将生成器内存从约80 GB降至约10 MB;引入增量形式 θ = θ^pre + UV^T z,学习围绕中央预训练基础模型的增量修正,实现可扩展的联邦微调。冻结正交分类头进一步移除头部参数传输并提升准确率。在CIFAR-100上使用ResNet-18+GroupNorm,本方法(FLITE)每客户端每轮仅传输1280个浮点数(约5KB),较全量微调减少8718倍,达到74.67%准确率,仅比全量微调低约0.5个百分点。平均恒等性在浮点精度下成立(6×10^-8);该方法位于带宽-精度帕累托前沿的低位,优于PowerSGD和top-k一到两个数量级,在强非独立同分布偏斜下性能持平或超越全量微调。使用int4潜变量可降至648字节/轮而准确率不变,而int4全量微调则退化至随机水平。

原文摘要 · Abstract (English)

Federated fine-tuning is bottlenecked by communication: FedAvg and pseudo-gradient schemes transmit a payload that scales with the model, and gradient compression shrinks it by only a constant factor. We take a different lever. Mapping networks generate a network's weights from a small trainable latent through a frozen affine projection; because the map is shared and affine, averaging latents is exactly averaging the generated weights. We turn this into a practical low-bandwidth federated channel with two changes: a low-rank, seed-regenerable factorisation of the projection (cutting generator memory from ~80 GB to ~10 MB), and a delta formulation $θ= θ^{\mathrm{pre}} + U V^{\top} z$ that learns an additive correction around a shared centrally-pretrained base -- federated fine-tuning, which is what makes the method work at scale. A frozen orthogonal classifier head further removes the head from the payload while improving accuracy. On CIFAR-100 with ResNet-18+GroupNorm, our method (FLITE, Federated Low-rank Iterative Training Engine) communicates 1,280 floats (~5 KB) per client per round -- an 8718x reduction -- and reaches 74.67%, within ~0.5 pp of full-weight FedAvg. The averaging identity holds to floating-point precision ($6 \times 10^{-8}$); the method sits one to two orders of magnitude below PowerSGD and top-k on the bandwidth-accuracy Pareto; it matches or exceeds full-weight FedAvg under strong non-IID skew. int4 latents reach 648 bytes per round at unchanged accuracy, whereas int4 full-weight FedAvg collapses to chance.

联邦学习轻量化通信压缩模型生成

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