Rethinking the Transferable Adversarial Attacks and Robust Defense in Federated Learning
Authors: Zuobin Xiong, Deval Mukherjee, Homook Cho, Wei Li
Organizations: Department of Computer Science, University of Nevada Las Vegas, Las Vegas, USA · Cyber Security Research Center at KAIST, Daejeon, South Korea · Department of Computer Science, Georgia State University, Atlanta, Georgia, USA
The development of federated learning (FL) techniques has helped improve the privacy preservation of users' data and extended the applications of machine learning models. However, the involvement of a large number of users in FL also creates open opportunities for different adversaries, such as poisoning attacks, Byzantine attacks, and adversarial example attacks. Yet, recent research has disclosed that existing poisoning attacks and Byzantine attacks can not achieve satisfactory penetration in realistic FL scenarios caused by strong assumptions, e.g., client selection rate, and the ratio of malicious attackers. In this paper, the transferability of adversarial examples among different client models is analyzed to understand the relation between adversarial examples and clients' data distribution. Moreover, to mitigate the attacks of transferable adversarial examples, we design a defense mechanism stemming from the transferability of model robustness by adversarial training. As a result, through theoretical analysis of transferability, we gain insights into adversarial examples and the vulnerability of federated learning systems. Our proposed adversarial attack and defense methods are evaluated via real-life datasets in various settings to show their performance over the existing state-of-the-art methods.
Figures & tables
Fig. 1: Overview of the Federated Learning Framework used in this study.
Fig. 2: Correlation between Model Distance and Attack Success Rate (Fig. 2a , 2b ), and between Distribution Distance and Attack Transferability (Fig. 2c , 2d ) across different datasets.
Method
Clean
VMIFGSM
BIM
VNIFGSM
DIFGSM
VGG11
Standard FedProx
94.49
<1
<1
<1
<1
FAT (Baseline)
93.53
81.57
84.22
81.49
78.01
Ours
92.62
83.26
85.39
83.28
79.87
Centralized (Upper Bound)
95.68
94.57
93.78
93.43
94.96
WRN28
TABLE I: Comparison of defense accuracy (%) against various transfer-based adversarial attacks on CIFAR-10 (IID) .
Method
Clean
VMIFGSM
BIM
VNIFGSM
DIFGSM
VGG11
Standard FedProx
97.78
<1
<1
<1
<1
FAT (Baseline)
95.69
43.13
54.13
42.93
43.05
Ours
95.67
48.71
61.55
48.59
45.55
Centralized (Upper Bound)
98.33
90.20
93.44
89.86
84.75
WRN28
TABLE II: Comparison of defense accuracy (%) against various transfer-based adversarial attacks on SVHN (IID) .
Method
Clean
VMIFGSM
BIM
VNIFGSM
DIFGSM
VGG11
Standard FedProx
87.01
<1
<1
<1
<1
FAT (Baseline)
86.14
47.52
49.00
47.55
45.14
Ours
83.56
52.58
53.59
52.52
49.89
WRN28
Standard FedProx
75.56
<1
<1
<1
<1
TABLE III: Comparison of defense accuracy (%) against various transfer-based adversarial attacks on CIFAR-10 (Non-IID) .
Method
Clean
VMIFGSM
BIM
VNIFGSM
DIFGSM
VGG11
Standard FedProx
95.34
<1
<1
<1
<1
FAT (Baseline)
93.56
27.64
32.90
27.42
27.38
Ours
94.05
28.95
36.30
28.75
28.48
WRN28
Standard FedProx
92.45
<1
<1
<1
<1
TABLE IV: Comparison of defense accuracy (%) against various transfer-based adversarial attacks on SVHN (Non-IID) .
The proliferation of IoT devices has fueled distributed edge systems to collect vast amounts of sensitive data, creating fertile ground for on-device machine learning applications. While federated learning (FL) mitigates privacy concerns by exchanging model parameters instead of raw data, we identify a critical blind spot in current research. We examine the most commonly used personalized federated learning (PFL) methods, which allow clients to maintain private, personalized models to address data heterogeneity across clients. Through systematic analysis, we reveal that PFL methods exhibit heightened vulnerability to transfer-based adversarial attacks compared to centralized learning paradigms. Wherein, malicious clients can exploit local model knowledge to craft adversarial examples that can compromise peer clients' personalized models. We establish this vulnerability through both theoretical analysis and empirical evaluation across multiple benchmark datasets, demonstrating significant accuracy drops across various PFL methods. To address this challenge, we propose a defense framework combining stochastic input noise, input-scaled trace regularization, and parameter sensitivity maximization to improve FL's robustness. Our findings establish the first systematic study of adversarial threats in PFL systems, providing both diagnostic tools and practical countermeasures.
Federated learning (FL) is a popular distributed learning paradigm in machine learning, which enables multiple clients to collaboratively train models under the guidance of a server without exposing private client data. However, FL's decentralized nature makes it vulnerable to poisoning attacks, where malicious clients can submit corrupted models to manipulate the system. To counter such attacks, although various Byzantine-robust methods have been proposed, these methods struggle to provide balanced defense against multiple types of attacks or rely on possessing the dataset in the server. To deal with these drawbacks, thus, we propose an effective multi-layer defensive adaptive aggregation for Bzantine-robust federated learning (AdaBFL) based on a novel three-layer defensive mechanism, which can adaptively adjust the weights of defense algorithms to counter complex attacks. Moreover, we provide convergence properties of our AdaBFL method under the non-convex setting on non-iid data. Comprehensive experiments across multiple datasets validate the superiority of our AdaBFL over the comparable algorithms.
Zehui Tang, Yuchen Liu, Feihu Huang
College of Computer Science and Technology, Nanjing University of Aeronautics and Astronautics, Nanjing, China · MIIT Key Laboratory of Pattern Analysis and Machine Intelligence, Nanjing, China.
Federated Learning enables decentralized model training by exchanging model updates--rather than raw data--with a central parameter server (PS). While most of the existing defenses primarily assume static or independently acting adversaries, we reveal a new class of dynamically adaptive attacks that systematically bypass such protections. We propose Fed-ADR, a holistic attack framework in which a malicious orchestrator server (OS) dynamically coordinates a heterogeneous set of adversarial clients, including both targeted and untargeted attackers. Through real-time coordination by the OS, malicious clients strategically adapt their gradient updates to evade defenses deployed by the PS, while either severely degrading global model performance or steering training toward adversarial objectives.To mitigate this threat, we offer a detection mechanism that estimates each client's true gradient from historical updates, enabling real-time detection of coordinated malicious behavior without additional overhead. We further introduce an in-situ recovery mechanism that restores global model performance without restarting training, preserving convergence and minimizing recovery time. Comprehensive experiments on MNIST, Fashion-MNIST, and CIFAR-10 benchmark datasets demonstrate that Fed-ADR's attack scheme can reduce global accuracy from over 90% to below 10%, bypassing several state-of-the-art defenses. When our detection and recovery modules are employed, they identify malicious clients and restore accuracy to over 90% within a few rounds, at a substantially lower cost than retraining from scratch--achieving a reduction of at least 20x in computational overhead.
Mohamed Shaaban, Ahmed Abdelnaby, Mohamed Elmahallawy
School of Engineering and Applied Sciences, Washington State University, Richland, WA 99354, USA