SCRUB-FL: バックドアの忘却による表現の浄化とクレンジング
SCRUB-FL: Sanitizing and Cleansing Representations via Unlearning of Backdoors
連合学習におけるバックドア攻撃を、訓練中に疑わしいサンプルを検出し、収束後に生成モデルで近似したトリガーを用いて機械忘却を適用することで除去する二段階の防御手法を提案した。
著者: Osama Wehbi, Sarhad Arisdakessian, Omar Abdel Wahab, Azzam Mourad, Hadi Otrok
分類: cs.LG, cs.AI, cs.CR, cs.CV
原文アブストラクト
Federated Learning (FL) enables collaborative model training without sharing raw data, making it a promising paradigm for privacy-sensitive applications. However, its decentralized nature makes it inherently vulnerable to backdoor attacks, where malicious clients embed hidden triggers into local training data to manipulate model predictions. Existing defenses mainly operate during before and during aggregation cannot fully eliminate backdoor behaviors that persist in the converged global model. Moreover, the effectiveness of post-training sanitization is often limited by the server's lack of knowledge of trigger patterns or poisoned clients after convergence, resulting in residual backdoor behaviors or accuracy degradation due to neuron entanglement. To address this limitation, we propose SCRUB-FL (Sanitizing and Cleansing Representations via Unlearning of Backdoors), a two-phase solution for post-training backdoor removal in FL. During training, clients identify suspicious samples using spectral analysis and activation clustering, then train lightweight Wasserstein Generative Adversarial Network with Gradient Penalty (WGAN-GP) models to capture trigger-related distributions. The generator parameters are aggregated server-side to construct a global representation of suspicious patterns without exposing raw data. After convergence, the server synthesizes trigger-approximating samples and applies machine unlearning to erase the trigger-target association by redistributing predictions toward a uniform distribution. Experimental evaluations on CIFAR-10 and GTSRB across three attack types and up to 40% malicious participation demonstrate that SCRUB-FL reduces the backdoor attack success rate to as low as 3.88% while maintaining over 91% normal task accuracy, outperforming state-of-the-art defenses without requiring prior trigger knowledge or a large clean proxy dataset at the server.
関連論文
- 連合学習のための鍵付き来歴透かしと相補的格子ベース安全集約連合学習/セキュリティ
- FedLNS: レイヤー正規化シグネチャモデリングを活用した連合LLMにおける敵対的操作の緩和連合学習/セキュリティ
- 垂直連合学習におけるバックドア脆弱性の理解:研究と実践のギャップ連合学習/セキュリティ