arXiv:2506.15711cs.LGcs.AI2025-06被引 15

针对联邦学习中的梯度反演攻击,提出可解释的针对性噪声防御方法。

Shadow defense against gradient inversion attack in federated learning

  • 用可解释的影子模型识别易受攻击的图像区域,实现精准噪声注入
  • 在ChestXRay和EyePACS数据集上,PSNR提升3.73和2.78,SSIM提升0.2和0.166
  • 仅损失不足1%的F1分数,适合医疗等高敏感领域应用

联邦学习(FL)作为一种隐私保护的分布式训练框架,使客户端可在不共享本地数据的情况下协作训练全局模型,尤其适用于医疗等敏感领域。然而,模型更新的通信可能被攻击者利用,引发梯度反演攻击(GIA),从而重建训练图像并泄露患者隐私。现有防御方法通过全局扰动模糊梯度,但缺乏对脆弱梯度或信息类型的精细理解,导致保护过度降低模型精度,或保护不足无法有效防护。为此,本文提出一种基于可解释影子模型的防御框架,能精准识别敏感区域,实现样本级、目标导向的噪声注入。在ChestXRay数据集上,相较无防御情况,PSNR提升3.73,SSIM提升0.2;在EyePACS数据集上,PSNR提升2.78,SSIM提升0.166。该方法对模型性能影响极小,相比最先进方法,F1分数下降不足1%。跨多种医学图像的实验验证了其泛化能力,对FedAvg的防御效果在LPIPS和SSIM上稳定提升超过1.5%,且对多种类型梯度反演攻击具有通用防御力。

原文摘要 · Abstract (English)

Federated learning (FL) has emerged as a transformative framework for privacy-preserving distributed training, allowing clients to collaboratively train a global model without sharing their local data. This is especially crucial in sensitive fields like healthcare, where protecting patient data is paramount. However, privacy leakage remains a critical challenge, as the communication of model updates can be exploited by potential adversaries. Gradient inversion attacks (GIAs), for instance, allow adversaries to approximate the gradients used for training and reconstruct training images, thus stealing patient privacy. Existing defense mechanisms obscure gradients, yet lack a nuanced understanding of which gradients or types of image information are most vulnerable to such attacks. These indiscriminate calibrated perturbations result in either excessive privacy protection degrading model accuracy, or insufficient one failing to safeguard sensitive information. Therefore, we introduce a framework that addresses these challenges by leveraging a shadow model with interpretability for identifying sensitive areas. This enables a more targeted and sample-specific noise injection. Specially, our defensive strategy achieves discrepancies of 3.73 in PSNR and 0.2 in SSIM compared to the circumstance without defense on the ChestXRay dataset, and 2.78 in PSNR and 0.166 in the EyePACS dataset. Moreover, it minimizes adverse effects on model performance, with less than 1\% F1 reduction compared to SOTA methods. Our extensive experiments, conducted across diverse types of medical images, validate the generalization of the proposed framework. The stable defense improvements for FedAvg are consistently over 1.5\% times in LPIPS and SSIM. It also offers a universal defense against various GIA types, especially for these sensitive areas in images.

联邦学习隐私保护梯度攻击医疗AI

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