DEeR : Deviation Eliminating and Noise Regulating for Privacy-preserving Federated Low-rank Adaptation
Research output: Journal Publications and Reviews › RGC 21 - Publication in refereed journal › peer-review
Author(s)
Related Research Unit(s)
Detail(s)
Original language | English |
---|---|
Number of pages | 13 |
Journal / Publication | IEEE Transactions on Medical Imaging |
Publication status | Online published - 19 Dec 2024 |
Link(s)
Abstract
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 Deviation Eliminating and Noise Regulating (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. © 1982-2012 IEEE.
Research Area(s)
- Federated Learning, Foundation Models, Low-rank Adaptation, Parameter-efficient Tuning
Citation Format(s)
DEeR: Deviation Eliminating and Noise Regulating for Privacy-preserving Federated Low-rank Adaptation. / Zhu, Meilu; Mao, Axiu; Liu, Jun et al.
In: IEEE Transactions on Medical Imaging, 19.12.2024.
In: IEEE Transactions on Medical Imaging, 19.12.2024.
Research output: Journal Publications and Reviews › RGC 21 - Publication in refereed journal › peer-review