FedTrident: Resilient Road Condition Classification Against Poisoning Attacks in Federated Learning
Organizations: Networked Systems Security Group, KTH Royal Institute of Technology, 114 28 Stockholm, Sweden
Abstract
FL has emerged as a transformative paradigm for ITS, notably camera-based Road Condition Classification (RCC). However, by enabling collaboration, FL-based RCC exposes the system to adversarial participants launching Targeted Label-Flipping Attacks (TLFAs). Malicious clients (vehicles) can relabel their local training data (e.g., from an actual uneven road to a wrong smooth road), consequently compromising global model predictions and jeopardizing transportation safety. Existing countermeasures against such poisoning attacks fail to maintain resilient model performance near the necessary attack-free levels in various attack scenarios due to: 1) not tailoring poisoned local model detection to TLFAs, 2) not excluding malicious vehicular clients based on historical behavior, and 3) not remedying the already-corrupted global model after exclusion. To close this research gap, we propose FedTrident, which introduces: 1) neuron-wise analysis for local model misbehavior detection (notably including attack goal identification, critical feature extraction, and GMM-based model clustering and filtering); 2) adaptive client rating for client exclusion according to the local model detection results in each FL round; and 3) machine unlearning for corrupted global model remediation once malicious clients are excluded during FL. Extensive evaluation across diverse FL-RCC models, tasks, and configurations demonstrates that FedTrident can effectively thwart TLFAs, achieving performance comparable to that in attack-free scenarios and outperforming eight baseline countermeasures by 9.49% and 4.47% for the two most critical metrics. Moreover, FedTrident is resilient to various malicious client rates, data heterogeneity levels, complicated multi-task, and dynamic attacks.
Figures & tables
| Abbreviation | Meaning |
|---|---|
| ITS | Intelligent Transportation Systems |
| DNN | Deep Neural Network |
| RCC | Road Condition Classification |
| FL | Federated Learning |
| TLFA | Targeted Label-Flipping Attack |
| Non-IID | Non-Independent and Identically Distributed |
| Solutions | Detection ∗ | Exclusion † | Remediation ‡ | Highlights (+ pros and - cons) |
|---|---|---|---|---|
| TMean/Median [ 24 ] | ● | ○ | ○ | Removes extremes or takes the median of client updates to limit outlier influence: + Simple, could work when most clients are honest and under IID data - Vulnerable to optimization-based and colluding attacks as well as Non-IID data |
| FoolsGold [ 11 ] | ● | ○ | ○ | Down-weights clients whose gradient histories are overly similar: + Directly exploits behavioral similarity as a robust, model-agnostic signal - Effectiveness drops with sophisticated diversification and non-Sybil strategies |
| FLAME [ 12 ] | ◐ | ○ | ○ | Clusters client updates and injects calibrated noise plus clipping to erase backdoor: + Demonstrates practical backdoor mitigation with modest impact on benign accuracy - Challenged by adaptive attacks and extreme heterogeneity |
| FreqFed [ 13 ] | ◐ | ○ | ○ | Uses frequency-domain analysis of model updates to distinguish poisoned contributions: + Effectively exposes backdoor and untargeted attacks - Heavily tied to assumed spectral patterns and model structures |
| LFGurad [ 7 ] | ● | ○ | ○ | Uses last-layer activations with an SVM to flag malicious participants: + Tailored to vehicular networks with a concrete pipeline for attack detection - Depends on a representative, trusted auxiliary dataset |
| FLDetector [ 39 ] | ◐ | ◐ | ○ | Checks temporal consistency of local updates via Hessian-informed trajectory prediction: + Works without a clean validation set on the server side - Fragile under strong non-IID and large number of malicious clients |
| Notation | Meaning |
|---|---|
| The total number of clients/vehicles | |
| The entire client cluster | |
| the number of road condition classes | |
| The ordered label sequence | |
| The output layer of RCC model | |
| The local dataset of client |
| Term | Value |
|---|---|
| Loss Function | Cross-Entropy |
| Batch Size | 64 |
| Learning Rate (LR) | 0.03 |
| Momentum for LR | 0.5 |
| Optimizer | SGD |
| Local Epoch | 3 |
| Device | Model | Params | Inference | Memory |
|---|---|---|---|---|
| Type | Type | (FP32, M) | Latency (ms) | (Peak, M) |
| Jetson | ResNet-18 | 42.65 | 9.93 | 60.97 |
| ResNet-34 | 81.21 | 16.43 | 101.07 | |
| MobileNet-V3 | 5.81 | 17.26 | 16.53 | |
| EfficientNet-B1 | 24.88 | 38.14 | 43.87 | |
| DenseNet-121 | 26.55 | 50.79 | 45.24 |
| Model Type | Method | RCC @ Friction | RCC @ Material | RCC @ Unevenness | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| SRE | ASR | GAC | GAS | SRE | ASR | GAC | GAS | SRE | ASR | GAC | GAS | ||
| ResNet-18 | FedAvg-NA ‡ [ 3 ] | ||||||||||||
| FedAvg [ 3 ] | |||||||||||||
| Krum [ 23 ] | |||||||||||||
| TMean [ 24 ] | |||||||||||||
| Median [ 24 ] | |||||||||||||
| Model Type | Method | SRE | ASR | GAC | GAS |
|---|---|---|---|---|---|
| ResNet-18 | FedAvg-NA ‡ [ 3 ] | ||||
| FedAvg [ 3 ] | |||||
| Krum [ 23 ] | |||||
| TMean [ 24 ] | |||||
| Median [ 24 ] | |||||
| [ 18 ] | FoolsGold [ 11 ] |
| RCC Task | Method | SRE | ASR | GAC | GAS |
|---|---|---|---|---|---|
| Friction | FedAvg [ 3 ] | ||||
| Krum [ 23 ] | |||||
| TMean [ 24 ] | |||||
| Median [ 24 ] | |||||
| FoolsGold [ 11 ] | |||||
| FLAME [ 12 ] |
| Model Type | Method | RCC @ Friction | RCC @ Material | RCC @ Unevenness | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| SRE | ASR | GAC | GAS | SRE | ASR | GAC | GAS | SRE | ASR | GAC | GAS | ||
| ResNet-18 | Detection | ||||||||||||
| Detection + Exclusion ‡ | 74.39 | 71.03 | 66.56 | ||||||||||
| FedTrident (Full) | 1.72 | 85.16 | 75.12 | 87.47 | 1.89 | 79.77 | 71.76 | 80.17 | 5.68 | 74.16 | 67.21 | ||
| EfficientNet-B1 | Detection | ||||||||||||
| Detection + Exclusion | |||||||||||||
Federated Learning
Federated learning trains shared models across data holders without requiring them to pool their raw data.
See all 943 papers