Reliability and Energy Balanced Computation Resource Allocation Mechanism in Federated Learning System for AI-Enabled IoT Businesses

Yonghao Qi,Siya Xu,Feng Qi,Peng Yu
DOI: https://doi.org/10.1109/globecom54140.2023.10437237
2023-01-01
Abstract:With the advancement and widespread adoption of Artificial Intelligence (AI) and Internet of Things (IoT) technology, machine learning (ML) can be leveraged in au-tomated factories to enable intelligent robot fault detection and recognition. As an emerging distributed machine learning framework, federated learning (FL) enables collaborative model training while safeguarding the privacy of user data. However, FL encounters various challenges, including the lack of stable energy supply to maintain continuous training and the existence of malicious terminals in the system. These situations could result in issues such as free-riding attacks and robots running out of energy during the training process. Hence, this paper presents a reliability and energy balanced computation resource allocation mechanism in FL system for IoT businesses. Firstly, a training latency and model contribution based reputation evaluation model is proposed to reveal the reliability of IoT terminals. Then, a training client reliable selection and delayed admission strategy is proposed, which takes into account both the reputation and energy consumption, aiming at enabling terminals with high-quality data and low energy capacity to keep contributing to the model during the later training stage. Especially, we integrate reputation and computing capability of IoT terminals to decide the freshness level of the allocated FL global model, which can effectively defend against free-riding attacks. The simulation results verify that the proposed mechanism outperforms non-delayed/partial-delayed admission mechanisms in terms of model quality and system stability.
What problem does this paper attempt to address?