arXiv:2410.12926cs.CV2024-10被引 15

解决联邦学习中低秩适配的偏差与噪声放大问题,提升医疗场景隐私保护效果。

DEeR: Deviation Eliminating and Noise Regulating for Privacy-preserving Federated Low-rank Adaptation

  • 通过交替优化确保客户端LoRA参数一致,消除聚合偏差。
  • 设计噪声调节器降低差分隐私噪声放大,提升隐私保护鲁棒性。
  • 在公开医疗数据集上优于现有方法,适合医疗联邦学习场景。

将低秩适配(LoRA)与联邦学习(FL)结合以实现隐私保护的分布式微调,已成为近期研究热点,旨在将预训练基础模型(FMs)适配到下游医疗任务。然而,当前方法因直接结合LoRA与FL,面临两个核心问题:聚合偏差和差分隐私(DP)噪声放大。为此,本文提出新型隐私保护联邦微调框架DEeR。理论证明消除聚合偏差的必要条件是客户端间LoRA参数等价。基于此,设计偏差消除器,利用交替最小化算法迭代优化零初始化与非零初始化的LoRA参数矩阵,确保训练全程偏差为零。进一步分析噪声放大机制,发现其主要源于DP噪声与LoRA参数间的线性关系。为此提出噪声调节器,引入双调节因子解耦二者关系,实现强隐私保护与优异微调性能。大量消融实验验证了各模块有效性,DEeR在多个公开医疗数据集上优于现有先进方法。代码已开源。

原文摘要 · Abstract (English)

Integrating low-rank adaptation (LoRA) with federated learning (FL) has received widespread attention recently, aiming to adapt pretrained foundation models (FMs) to downstream medical tasks via privacy-preserving decentralized training. However, owing to the direct combination of LoRA and FL, current methods generally undergo two problems, i.e., aggregation deviation, and differential privacy (DP) noise amplification effect. To address these problems, we propose a novel privacy-preserving federated finetuning framework called \underline{D}eviation \underline{E}liminating and Nois\underline{e} \underline{R}egulating (DEeR). Specifically, we firstly theoretically prove that the necessary condition to eliminate aggregation deviation is guaranteing the equivalence between LoRA parameters of clients. Based on the theoretical insight, a deviation eliminator is designed to utilize alternating minimization algorithm to iteratively optimize the zero-initialized and non-zero-initialized parameter matrices of LoRA, ensuring that aggregation deviation always be zeros during training. Furthermore, we also conduct an in-depth analysis of the noise amplification effect and find that this problem is mainly caused by the ``linear relationship'' between DP noise and LoRA parameters. To suppress the noise amplification effect, we propose a noise regulator that exploits two regulator factors to decouple relationship between DP and LoRA, thereby achieving robust privacy protection and excellent finetuning performance. Additionally, we perform comprehensive ablated experiments to verify the effectiveness of the deviation eliminator and noise regulator. DEeR shows better performance on public medical datasets in comparison with state-of-the-art approaches. The code is available at https://github.com/CUHK-AIM-Group/DEeR.

联邦学习隐私保护低秩适配医疗AI

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