skip to main content
10.1145/3653081.3653105acmotherconferencesArticle/Chapter ViewAbstractPublication PagesiotaaiConference Proceedingsconference-collections
research-article

A Federated Learning Model for Fault Diagnosis of IIoT Using a Modified PSO Algorithm Customized by Taguchi Method

Authors Info & Claims
Published:03 May 2024Publication History

ABSTRACT

The Industrial Internet of Things (IIoT) represents the deployment of Internet of Things (IoT) technology in industrial applications. In this article, we address the challenges of fault diagnosis and data privacy protection within the IIoT environment. We present a novel fault diagnosis model that combines federated learning and a particle swarm optimization algorithm. Firstly, we introduce a three-tier federated learning model designed to safeguard the data privacy of each participant in a real industrial control network structure. Subsequently, we enhance the particle swarm optimization algorithm to augment its global exploration capabilities and convergence performance, enabling it to collect federated learning model weights in lieu of traditional techniques. Furthermore, we employ the Taguchi method to tailor an optimal solution for the modified particle swarm optimization algorithm (TMPSO), thereby enhancing the algorithm's efficiency and robustness. Additionally, we propose a neural network model utilizing small convolutional kernels (SVGG) for fault diagnosis within the IIoT framework, thereby improving the model's feature learning capabilities. Experimental validation was conducted using actual industrial rolling bearing datasets and CIFAR-10 datasets. The results of these experiments demonstrate that our proposed TPMPSO-SVGG model outperforms other methods in terms of fault identification accuracy and communication cost.

References

  1. Lu, Y., Huang, X., Zhang, K., Maharjan, S., & Zhang, Y. 2020. Communication-efficient federated learning for digital twin edge networks in industrial IoT. IEEE Transactions on Industrial Informatics, 17(8), 5709-5718.Google ScholarGoogle ScholarCross RefCross Ref
  2. Wu, W., Peng, M., Chen, W., & Yan, S. 2020. Unsupervised deep transfer learning for fault diagnosis in fog radio access networks. IEEE Internet of Things Journal, 7(9), 8956-8966.Google ScholarGoogle ScholarCross RefCross Ref
  3. Jiang, Q., Bao, B., Hou, X., Huang, A., Jiang, J., & Mao, Z. 2023. Feature Mining and Sensitivity Analysis with Adaptive Sparse Attention for Bearing Fault Diagnosis. Applied Sciences, 13(2), 718.Google ScholarGoogle ScholarCross RefCross Ref
  4. McMahan, B., Moore, E., Ramage, D., Hampson, S., & y Arcas, B. A. 2017. Communication-efficient learning of deep networks from decentralized data. In Artificial intelligence and statistics (pp. 1273-1282). PMLR.Google ScholarGoogle Scholar
  5. Zhang, Z., Guan, C., Chen, H., Yang, X., Gong, W., & Yang, A. 2021. Adaptive privacy-preserving federated learning for fault diagnosis in internet of ships. IEEE Internet of Things Journal, 9(9), 6844-6854.Google ScholarGoogle ScholarCross RefCross Ref
  6. Weng, J., Weng, J., Zhang, J., Li, M., Zhang, Y., & Luo, W. 2019. Deepchain: Auditable and privacy-preserving deep learning with blockchain-based incentive. IEEE Transactions on Dependable and Secure Computing, 18(5), 2438-2455.Google ScholarGoogle ScholarDigital LibraryDigital Library
  7. Sun, W., Lei, S., Wang, L., Liu, Z., & Zhang, Y. 2020. Adaptive federated learning and digital twin for industrial internet of things. IEEE Transactions on Industrial Informatics, 17(8), 5605-5614.Google ScholarGoogle ScholarCross RefCross Ref
  8. Mao, Y., Zhao, Z., Yan, G., Liu, Y., Lan, T., Song, L., & Ding, W. 2022. Communication-efficient federated learning with adaptive quantization. ACM Transactions on Intelligent Systems and Technology (TIST), 13(4), 1-26.Google ScholarGoogle ScholarDigital LibraryDigital Library
  9. Zhai, R., Chen, X., Pei, L., & Ma, Z. 2023. A Federated Learning Framework against Data Poisoning Attacks on the Basis of the Genetic Algorithm. Electronics, 12(3), 560.Google ScholarGoogle ScholarCross RefCross Ref
  10. Junior, F. E. F., & Yen, G. G. 2019. Particle swarm optimization of deep neural networks architectures for image classification. Swarm and Evolutionary Computation, 49, 62-74.Google ScholarGoogle ScholarCross RefCross Ref
  11. Li, W., Liang, P., Sun, B., Sun, Y., & Huang, Y. 2023. Reinforcement learning-based particle swarm optimization with neighborhood differential mutation strategy. Swarm and Evolutionary Computation, 78, 101274.Google ScholarGoogle ScholarCross RefCross Ref
  12. Vaidya, G., Nalla, B. T., Sharma, D. K., Thangaraja, J., Devarajan, Y., & Ponnappan, V. S. 2022. Production of biodiesel from phoenix sylvestris oil: process optimisation technique. Sustainable Chemistry and Pharmacy, 26, 100636.Google ScholarGoogle ScholarCross RefCross Ref
  13. Park, S., Suh, Y., & Lee, J. 2021. FedPSO: Federated learning using particle swarm optimization to reduce communication costs. Sensors, 21(2), 600.Google ScholarGoogle ScholarCross RefCross Ref
  14. Zhang, W., Li, C., Peng, G., Chen, Y., & Zhang, Z. 2018. A deep convolutional neural network with new training methods for bearing fault diagnosis under noisy environment and different working load. Mechanical systems and signal processing, 100, 439-453.Google ScholarGoogle Scholar

Index Terms

  1. A Federated Learning Model for Fault Diagnosis of IIoT Using a Modified PSO Algorithm Customized by Taguchi Method

    Recommendations

    Comments

    Login options

    Check if you have access through your login credentials or your institution to get full access on this article.

    Sign in
    • Published in

      cover image ACM Other conferences
      IoTAAI '23: Proceedings of the 2023 5th International Conference on Internet of Things, Automation and Artificial Intelligence
      November 2023
      902 pages
      ISBN:9798400716485
      DOI:10.1145/3653081

      Copyright © 2023 ACM

      Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].

      Publisher

      Association for Computing Machinery

      New York, NY, United States

      Publication History

      • Published: 3 May 2024

      Permissions

      Request permissions about this article.

      Request Permissions

      Check for updates

      Qualifiers

      • research-article
      • Research
      • Refereed limited
    • Article Metrics

      • Downloads (Last 12 months)3
      • Downloads (Last 6 weeks)3

      Other Metrics

    PDF Format

    View or Download as a PDF file.

    PDF

    eReader

    View online with eReader.

    eReader

    HTML Format

    View this article in HTML Format .

    View HTML Format