Adaptive Resource Allocation for Blockchain-Based Federated Learning in Internet of Things

Abstract

The fast development of mobile communication and artificial intelligence (AI) technologies greatly promotes the prosperity of the Internet of Things (IoT), where various types of IoT devices can perform more intelligent tasks. Considering the privacy leakage and limited communication resources, federated learning (FL) has emerged to enable devices to collaboratively train AI models based on their local data without raw data exchanges. Nevertheless, it is still challenging for guaranteeing any FL models to be effective due to the sluggish willingness of IoT devices and the model poisoning attacks in the FL. To address these issues, in this paper, we introduce blockchain technology and propose a blockchain-based FL framework for supporting a trustworthy and reliable FL paradigm in IoT. In the proposed framework, we design a committee-based participant selection mechanism that selects the aggregate node and local model updates dynamically to construct the global model. Moreover, considering the trade-off between the energy consumption and the convergence rate of the FL model, we perform the channel allocation, block size adjustment, and block producer selection jointly. Since the remaining resources, handling transactions, and channel conditions are dynamically varying (i.e., stochastic environment), we formulate the problem as a Markov decision process (MDP) and adopt a deep reinforcement learning (DRL)-based algorithm to solve it. The simulation results demonstrate the effectiveness of the proposed framework and show the superior performance of the DRL-based resource allocation algorithm compared with other baseline methods in terms of energy consumption.