Secure PDM: A Novel Byzantine Fault Tolerant Federated Learning Framework using a Robust PCA-Based Anomaly Detection Approach

With the proliferation of federated learning programs as a suitable framework for protecting user privacy and reducing the computational overhead of AI algorithms, various industries have also turned to the widespread use of this framework in industrial applications such as improving predictive maintenance (PDM). However, despite its increasing applications, several security challenges, such as Byzantine attacks, make the application of federated learning in industries questionable. Byzantine attacks in FL can degrade model performance by injecting malicious updates, causing model divergence or biased learning. This reduces accuracy, and can introduce security vulnerabilities such as backdoors. To address this problem, we propose a Byzantine Fault Tolerant (BFT) federated learning algorithm designed to improve PDM in industrial applications. Our proposed approach uses a PCA-based anomaly detection algorithm to detect and mitigate local Byzantine updates. Also, a game theory-based reward mechanism is designed to promote honest participation and discourage malicious behavior among federated users. The proposed framework is evaluated using the predictive maintenance datasets “AI۴I ۲۰۲۰” and “NASA Acoustics and Vibration”. The results show that our proposed framework effectively detects and mitigates Byzantine attacks, enhancing the overall reliability of PDM in industrial applications.

کلیدواژه ها:

Federated Learning ، byzantine fault tolerant ، Predictive Maintenance ، Anomaly detection