This paper proposes a communication-efficient adaptive federated learning algorithm for heterogeneous defect classification tasks that achieves competitive classification accuracy while reducing single-round training time by up to 70%.
Abstract
Timely and accurate defect classification is critical for ensuring product quality and safety in industrial inspection scenarios. The widespread deployment of Internet of Things (IoT) devices equipped with sensing, computing, and communication capabilities has promoted the development of AI-enabled industrial applications. However, conventional AI approaches typically rely on centralized data collection and processing, which become impractical in real-world IoT environments due to growing privacy concerns and constrained device resources. To address these challenges, this paper proposes a communication-efficient adaptive federated learning algorithm for heterogeneous defect classification tasks. The proposed approach jointly accelerates the training process through three mechanisms: (i) adaptive local updates that balance communication and computation overheads; (ii) parameter compression that trades off communication cost against model accuracy; (iii) joint bandwidth and computation-power allocation that optimizes per-round communication and computation time across participating devices. We further analyze the joint effects of these three mechanisms and provide a convergence analysis. Extensive simulations show that the proposed method achieves competitive classification accuracy while reducing single-round training time by up to 70%.
With the rapid development of the Internet of Things (IoT) and edge computing, the scale and complexity of modern networks have increased significantly, driving the demand for distributed fault diagnosis. Federated learning (FL) effectively addresses the issues of data privacy and dispersion by enabling edge devices to collaboratively train models without sharing raw data. However, existing FL-based fault diagnosis methods still encounter the following challenges. Firstly, static aggregation strategies struggle to balance the contributions of heterogeneous clients dynamically. Secondly, traditional local models are unable to effectively decouple sparse high and low-frequency features in fault signals, thereby limiting the accuracy of fault identification. Finally, the resource constraints of edge devices restrict the deployment of complex diagnostic models. To address these challenges, we propose a federated learning hybrid dynamic weight adjustment method based on delay and model quality, introducing the concept of “accelerated depreciation” in accounting and taxation and the concept of “asset allocation” in economics to improve the communication efficiency of fault diagnosis and reduce the impact of delay differences due to device heterogeneity on the effect of fault diagnosis. Additionally, we propose an adaptive sparse low high frequencies Transformer, introducing a lightweight attention mechanism and an adaptive feature extraction layer, which significantly reduces the computational overhead while maintaining high diagnostic accuracy. The experimental results show that, compared with the most competitive baseline, our method improves the fault diagnosis accuracy by 1% on the Case Western Reserve University Bearing dataset (CWRU), 0.78% on the Xi’an Jiaotong University Gearbox dataset (XJTU), and 0.44% on the Micro service Edge Computing dataset (MICRO).
Jingting Mei, Yang Yang, Celimuge Wu et al.· IEEE Transactions on Cogniti...· 0 citations
The Federated Green Anaconda Optimizer (FedGAO), an innovative FL framework inspired by the behavioral patterns of the Green Anaconda Optimizer (GAO), is proposed, demonstrating superior performance in terms of accuracy, convergence speed, and resource efficiency.
Elahe Eslami, S. A. Shahzadeh Fazeli, J. Abouei et al.· Cluster Computing· 0 citations
FedAnchor is proposed, an anchored and adaptive FL framework for resource-heterogeneous fault diagnosis that couples resource-aware submodel allocation with structurally aligned aggregation and achieves competitive or superior accuracy compared with representative homogeneous and model-heterogeneous FL baselines.
A hierarchical privacy protection and poisoning-robust defense framework for industrial federated learning is proposed and can effectively suppress global-model degradation under multiple poisoning attacks and achieves a favorable balance among privacy protection strength, robustness, and training efficiency.
Huan Yin, Cong Chen, Jing-Yi Zhang et al.· Italian National Conference...· 0 citations
The deployment of Hierarchical Federated Learning (HFL) in resource-constrained Internet of Things (IoT) environments requires careful configuration to balance predictive performance with energy consumption and execution time. This challenge is particularly relevant to smart agriculture, where distributed IoT devices can support automated plant disease classification while operating under limited computational and communication resources. This paper presents a constrained Bayesian Optimization framework for the efficient configuration of HFL deployments. The proposed approach jointly explores the deep learning backbone architecture, aggregation strategy, and number of communication rounds, while the federation size is determined according to the spatial coverage requirements of the agricultural deployment. A weighted objective function captures user-defined trade-offs among energy consumption, execution time, and predictive performance, while explicit constraints ensure compliance with deployment-specific resource and accuracy requirements. The framework is evaluated on an IoT-based plant disease classification task considering multiple deep learning architectures, federated aggregation strategies, and communication-round settings. Experimental results across 30 independent optimization runs show that the proposed approach explores only 11.11% of the search space, while consistently identifying solutions within 1% of the exhaustive-search optimum, with a mean optimality gap of only 0.056%.
Athanasios Papanikolaou, Athanasios Tziouvaras, Apostolos Xenakis et al.· 0 citations
Experiments show that LBSFL achieves competitive model accuracy while substantially reducing computational and communication overhead in most evaluated settings, indicating that LBSFL provides a favorable trade-off between robustness and efficiency for IoT-oriented federated learning.
Wei Ma, Wenjun Tian, Qihang Zhao et al.· Journal of King Saud Univers...· 0 citations
We use cookies to run the site and, with your consent, for analytics and to show ads.
See our Cookie Policy.
