Communication
Khalil Jahani; Behzad Moshiri; Babak Hossein Khalaj
Abstract
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 ...
Read More
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 “AI4I 2020” 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.
