TSFed: A three-stage optimization mechanism for secure and efficient federated learning in industrial IoT networks

Abstract

This paper presents a three-stage optimization mechanism designed to enhance Federated Learning (FL) in Industrial Internet of Things (IIoT) networks. Traditional FL optimizations, which typically focus on a single aspect, fall short in IIoT environments. Our approach integrates a multi-criteria enhancement: first, an Ensembled Client Selection Mechanism (ECSM) selects participants based on accuracy, reputation, and randomness. Second, Adaptive Distributed Client Training (ADCT) dynamically adjusts based on participant performance. Lastly, a Secure and Efficient Communication Channel (SECC), backed by blockchain, meets IIoT’s stringent security demands. The evaluation shows TSFed outperforms baseline methods, enhancing FL performance by increasing accuracy and F1-score. Importantly, TSFed improves the efficiency of achieving 80% accuracy on the MNIST dataset by 29.09% over baseline methods, showcasing significant gains in both security and efficiency. This mechanism also exhibits robustness against malicious attacks, setting a new benchmark for FL in IIoT environments.