학술논문

Are You Diligent, Inefficient, or Malicious? A Self-Safeguarding Incentive Mechanism for Large-Scale Federated Industrial Maintenance Based on Double-Layer Reinforcement Learning
Document Type
Periodical
Source
IEEE Internet of Things Journal IEEE Internet Things J. Internet of Things Journal, IEEE. 11(11):19988-20001 Jun, 2024
Subject
Computing and Processing
Communication, Networking and Broadcast Technologies
Predictive models
Training
Federated learning
Production facilities
Data models
Servers
Industrial Internet of Things
Deep reinforcement learning
fault prediction
federated learning (FL)
incentive mechanism
Industrial Internet of Things (IIoT)
Language
ISSN
2327-4662
2372-2541
Abstract
Fault prediction is an important application in the Industrial Internet of Things (IIoT) to ensure the safety of industrial systems and factories. Currently, deep-learning-based fault prediction models are more popular, and multifactory co-operation is required to improve the accuracy and generality of fault prediction models. Federated learning can coordinate multiple clients to train models together while protecting client privacy, and thus is widely used for training fault prediction models. How to incentivise more factories to participate in model training is crucial, however, most of the existing incentive mechanisms focus on the problem of fair measurement of client contributions and ignore the problem of incentive allocation in scenarios with limited incentive budgets. In this article, we design a self-safeguarding incentive mechanism for large-scale federated industrial maintenance based on double-layer reinforcement learning, known as dual-layer incentive (DLI). The method enables the central server to achieve higher model training accuracy within a limited incentive budget through rational allocation of incentives, which ultimately reduces the overall cost of model training. In addition, we categorize participating clients into “diligent clients,” “inefficient clients,” and “malicious clients” based on their contributions and design tailor-made incentives for each client type, which saves training costs and enhances the safety of the model training process. Finally, the proposed approach is evaluated through experiments using various data sets. The results show that the method significantly improves the accuracy and safety of industrial fault prediction model compared to other existing methods.