中文

FedSVD:面向私有联邦学习的自适应正交化

机器学习 2025-05-20 v1

摘要

低秩适应(LoRA)通过将两个可训练的低秩矩阵的乘积引入到冻结的预训练权重中,广泛用于联邦学习(FL)中语言模型的高效微调。然而,当与差分隐私随机梯度下降(DP-SGD)结合时,LoRA 面临巨大的噪声放大:DP-SGD 对每个样本的梯度进行扰动,而 LoRA 更新的矩阵乘法(BABA)会加剧这种效应。冻结一个矩阵(如 AA)可减少噪声,但会限制模型的表达能力,常导致适应不佳。为此,我们提出了 FedSVD\texttt{FedSVD},一种简单而有效的方法,引入基于奇异值分解(SVD)的全局重参数化。在我们的方法中,每个客户端仅优化 BB 矩阵并将其传输给服务器。服务器聚合 BB 矩阵,使用先前的 AA 计算乘积 BABA,并通过 SVD 对结果进行重分解。这得到一个新的自适应 AA,由 BABA 的正交右奇异向量组成,以及包含剩余 SVD 成分的更新后的 BB。这种重参数化避免了二次噪声放大,同时使 AA 能更好地捕获聚合更新的主方向。此外,AA 的正交结构限制了 BBAA 的梯度范数,在 DP-SGD 下保留更多信号,正如我们的理论分析所证明的。因此,FedSVD\texttt{FedSVD} 在各种隐私设置和基准测试中始终能够提高稳定性和性能,在私有和非私有 regimes 下均优于相关基线。

关键词

引用

@article{arxiv.2505.12809,
  title  = {Koopman Autoencoders Learn Neural Representation Dynamics},
  author = {Nishant Suresh Aswani and Saif Eddin Jabari},
  journal= {arXiv preprint arXiv:2505.12809},
  year   = {2025}
}