Taking Advantage of the Mistakes: Rethinking Clustered Federated Learning for IoT Anomaly Detection
Jiamin Fan, Kui Wu, Guoming Tang, Yang Zhou, Shengqiang Huang · IEEE Transactions on Parallel and Distributed Systems · 2024
Clustered federated learning (CFL) is a promising solution to address the non-IID problem in the spatial domain for federated learning (FL). However, existing CFL solutions overlook the non-IID issue in the temporal domain and lack consideration of time efficiency. In this work, we propose a novel approach, calledClusterFLADS, which takes advantage of the false predictions of the inappropriate global models, together with knowledge of temperature scaling and catastrophic forgetting to reveal distributional similarities between the training data (of different clusters) and the test data. Additionally, we design an efficient feature extraction scheme by exploiting the role of each layer in a neural network's learning process. By strategically selecting model parameters and using PCA for dimensionality reduction,ClusterFLADSeffectively improves clustering speed. We evaluateClusterFLADSusing real-world IoT trace data in various scenarios. Our results show thatClusterFLADSaccurately and efficiently clusters clients, achieving a$100\%$true positive rate and low false positives across various data distributions in both the spatial and temporal domains.