Skip to content
Open access

Adaptive Federated Baseline K-Means for Lightweight IoT Intrusion Detection: Auto-Thresholding and Robust Statistics Aggregation

Aug 2026 · IoT · 0 citations · 24 references

TL;DR

AF-BKM is presented, an Adaptive Federated Baseline K-Means that repairs the federated mechanism with two label-free, statistics-only enhancements, and identifies merge-induced precision decay under non-IID workers as an open gap.

Abstract

Federated, semi-supervised novelty detection is well suited for intrusion detection on resource-constrained Internet of Things (IoT) nodes: each device learns a model of benign traffic, shares only summary statistics, and does not transmit raw traffic samples. A previously published cross-layer federated detector, Baseline K-Means, showed that periodically merging worker statistics through a coordinator raises the detection rate, but it also exhibited a systematic side effect: after every merge, the precision decays, and the false-positive rate (FPR) climbs because the coordinator recomputes its threshold from streaming distances filtered by the closest observed anomaly, so tightens after every merge, flagging progressively more benign traffic; the threshold was also hand-tuned. We present AF-BKM, an Adaptive Federated Baseline K-Means that repairs the federated mechanism with two label-free, statistics-only enhancements, denoted as E1 and E2: (i) an adaptive decision threshold read from the benign Mahalanobis-distance distribution, requiring no manual percentile search and no attack labels (E1), and (ii) a robust, benignly anchored aggregation that blends worker means under quality weighting and outlier-worker filtering and recalibrates the threshold on a trusted benign anchor to a stable, anchor-referenced false-positive level, which a target-FPR rule can make operator-selectable instead of tightening it toward the nearest anomaly (E2). With MinMax scaling fit only on benign baseline data and non-IID federated streams on NSL-KDD, UNSW-NB15 and the N-BaIoT corpus of real traffic from commercial IoT devices, AF-BKM removes the merge-induced precision decay (the first-to-last-epoch precision change improves from −0.134 to −0.002 on NSL-KDD, from −0.121 to −0.014 on UNSW-NB15, and from −0.170 to −0.009 on N-BaIoT) and reduces the mean FPR by 30–64%, depending on the dataset; all central improvements are significant across 10 seeds (Wilcoxon p=0.002, large effect sizes). AF-BKM preserves recall on NSL-KDD and N-BaIoT and, on the harder UNSW-NB15, exposes an explicit precision–recall trade-off through a benign target-FPR knob. In fp32, the deployed model serializes to 5.5–52 KB, a packet is classified in 11–27 µs on a desktop CPU, and each merge round uploads a d+3-value summary (160–472 B) 94.6–98.3% smaller than the same summary extended with the covariance upper triangle. A robustness study covering selected faulty-worker updates, contamination of the commissioning anchor, and detector-level white-box evasion reports the measured degradation patterns: fabricated threshold candidates have no direct path to the threshold, although a fabricated mean still reaches it indirectly through the blended centroid, and the anchor-referenced false-positive level remains stable under percent-level anchor contamination, while recall sensitivity is dataset-dependent and the evasion budget tracks the benign–attack margin of each dataset. We frame the contribution with a focused taxonomy that identifies merge-induced precision decay under non-IID workers as an open gap. Code is released for reproducibility.

Read PDF

Similar papers

Open access Aug 2026

A Three-Stage Federated Distillation Framework for Robust Intrusion Detection in Heterogeneous IoT/Edge Networks

Internet of Things(IoT)/edge intrusion-detection systems operate on distributed traffic and system-state data whose distributions vary across gateways, services, and attack conditions. We study a server-assisted federated setting in which a teacher reference is fitted from a permitted server-accessible training pool and explicitly distinguish this simulation assumption from fully decentralized deployment. The proposed framework evaluates progressive local training through boundary stabilization, confidence-weighted decision distillation, representation alignment, and validation-quality-aware aggregation. The evaluation uses a leakage-controlled protocol: server and client validation subsets are held out before federated training, update quality and early stopping use validation data only, and the final-test split is evaluated once. Results on NSL-KDD, CIC-IDS2017, Edge-IIoTset, and the ToN-IoT network dataset show competitive primary performance and stronger robustness in several severe label-skew settings. On the Telemetry of Things(ToN-IoT) with Dirichlet alpha = 0.1, the proposed method achieves 91.46 ± 5.54 F1, compared with 53.73 ± 49.00 for FedAvg and 53.77 ± 48.92 for FedProx. The results do not establish universal superiority or a universally optimal stage order: competing methods remain stronger in selected stable and attack-shift settings. The framework is therefore presented as a bounded, server-assisted robustness-oriented training strategy for heterogeneous IoT/edge intrusion detection.

Xudong Yang, Zikui Lin, Qiuyan Li et al. · 0 citations
Conference Open access 2026

Enhanced Intrusion Detection in IoT Networks using Federated Learning

The results show a success in implementing a real time, scalable, privacy-preserving, and adaptive IDS in large-scale IoT deployments through intelligent workload distribution between edge and cloud layers.

Chidera Winifred John, Eduediuyai Ekerete Dan, P. Asuquo et al. · 0 citations
Preprint Aug 2026

FBID: Adaptive Personalized Federated Learning for Robust Out-of-Distribution Attack Detection in IoT Networks

Federated Bandit Intrusion Detection (FBID), a novel adaptive PFL framework to address this limitation through server-side personalization control, employs a contextual multi-armed bandit at the server to dynamically regulate each client's local training intensity according to its observed behavior and update quality.

A. Bui, Cong T. Nguyen, Hoang-Anh Pham et al. · 0 citations
Open access Jul 2026

Mean/Std: Lightweight Distribution-Aware Aggregation for Federated IoT Botnet Detection

Federated learning (FL) is a promising paradigm for privacy-preserving IoT intrusion detection, but its effectiveness can be substantially degraded by the combination of heterogeneous non-IID client distributions and severe multi-class imbalance. Under such conditions, conventional size-based aggregation may overemphasize large yet highly skewed clients, limiting the representation of minority attack classes in the global model. To address this issue, we propose Mean/Std, a lightweight distribution-aware aggregation strategy that combines a client-size proxy with two complementary statistics of local label distributions, namely the standard deviation and the dominance gap of class proportions, while preserving a communication footprint comparable to FedAvg. Experiments on the N-BaIoT benchmark, comprising seven heterogeneous IoT clients and eleven traffic classes, are conducted under a privacy-oriented update-perturbation setting inspired by secure aggregation workflows. The results show that Mean/Std consistently provides the strongest imbalance-aware performance among the evaluated FL baselines, achieving a Macro-F1 score of 0.8418 and a Balanced Accuracy of 0.8722 while improving the representation of minority attack classes. Additional experiments across five independent random seeds and a comprehensive hyperparameter sensitivity analysis further confirm the robustness and stability of the proposed aggregation mechanism. Overall, the results demonstrate that lightweight distribution-aware aggregation offers an effective, robust, and practically deployable solution for mitigating aggregation bias under simultaneous non-IID heterogeneity and severe multi-class imbalance in FL-based IoT botnet detection.

Yassine El Yamani, Youssef Baddi, Najib El Kamoun · 0 citations
Open access Aug 2026

A Reliability-Aware Edge–Cloud Framework for Early Intrusion Detection in IoT Networks

Gateway-resident intrusion detection can act before IoT traffic reaches cloud services, but early decisions are based on incomplete flow prefixes. This paper presents a reliability-aware edge–cloud framework that treats early detection as a sequential routing problem. At each checkpoint, a lightweight gated recurrent unit (GRU) maps causal packet-prefix features to a malicious-probability estimate. Temperature scaling, asymmetric benign and malicious thresholds, and an eight-packet minimum-evidence gate determine whether a flow exits locally, remains under observation, or is sent for cloud refinement. Short and unresolved flows are classified by regularized logistic regression using a compact 97-feature causal representation. The edge model contains 19,777 parameters, and each cloud submission carries 388 bytes of float32 features. The principal evaluation uses all 309 CIC-IoT-2023 PCAP files under four outer PCAP-disjoint folds, with separate edge-training, calibration, cloud-development, and final-test roles. Across 2,286,754 pooled out-of-fold flows with 88.54% malicious prevalence, the framework resolves 422,190 flows at the edge and routes 1,864,564 for cloud refinement, reducing cloud submissions by 18.46%. The final policy attains 4.47% FPR, 1.89% FNR, 96.82% balanced accuracy, and 98.76% F1 score. Observation-budget analysis identifies 32 packets as a corpus-specific compromise, whereas controlled delays in post-eight-packet information expose the limits of short-prefix detection. On the balanced CICIDS2017 test set, in-domain development attains 97.03% balanced accuracy; zero-shot transfer falls to 86.30%, and target-calibration-only adaptation improves it to 91.65%. Ablation results identify the minimum-evidence gate and cloud-refinement stage as the main reliability controls. Benign false alarms, delayed post-eight-packet information, cross-dataset shift, and scenario/file-level labels remain the principal limitations.

Siraj Azam, Farheen Naaz, Mikail Mohammed Salim · 0 citations