A Blockchain-Empowered Multiaggregator Federated Learning Architecture in Edge Computing With Deep Reinforcement Learning Optimization

【Author】 Li, Xiao; Wu, Weili

【Source】IEEE TRANSACTIONS ON COMPUTATIONAL SOCIAL SYSTEMS

【影响因子】4.747

【Abstract】Federated learning (FL) is emerging as a sought-after distributed machine learning architecture, offering the advantage of model training without direct exposure to raw data. With advancements in network infrastructure, FL has been seamlessly integrated into edge computing. However, the limited resources on edge devices introduce security vulnerabilities to FL in the context. While blockchain technology promises to bolster security, practical deployment on resource-constrained edge devices remains a challenge. Moreover, the exploration of FL with multiple aggregators in edge computing is still new in the literature. Addressing these gaps, we introduce the blockchain-empowered heterogeneous multiaggregator federated learning architecture (BMA-FL). We design a novel lightweight Byzantine consensus mechanism, namely PBCM, to enable secure and fast model aggregation and synchronization in BMA-FL. We study the heterogeneity problem in BMA-FL that the aggregators are associated with varied number of connected trainers with non- IID data distributions and diverse training speed. We propose a multiagent deep reinforcement learning algorithm (MASBDRL) to help aggregators decide the best training strategies. Experiments on real-word datasets demonstrate the efficiency of BMA-FL to achieve better models faster than baselines, showing the efficacy of PBCM and MASB-DRL.

【Keywords】Training; Computational modeling; Computer architecture; Servers; Data models; Peer-to-peer computing; Security; Performance evaluation; Costs; Blockchain; deep reinforcement learning (DRL); distributed machine learning; edge computing; federated learning (FL); federated learning (FL)

【发表时间】2024 2024 OCT 30

【收录时间】2024-11-16

【文献类型】案例研究

【主题类别】

区块链技术-协同技术-边缘计算