CoCFL: A Lightweight Blockchain-based Federated Learning Framework in IoT Context

Jianrong Wang,Yang Shi,Dengcheng Hu,Keqiu Li,Xiulong Liu

DOI: https://doi.org/10.1109/icdcs60910.2024.00104

2024-01-01

Abstract:One notable drawback of traditional Federated Learning (FL) is its susceptibility to single point of failures. In recent years, Blockchain-based Federated Learning (BCFL) has been proposed as an effective solution to address this issue. However, existing BCFL frameworks face challenges in heterogeneous IoT scenarios. The heterogeneity of IoT devices poses challenges to the adaptation of blockchain consensus. The integration of blockchain imposes constraints on the learning scalability of systems, making it challenging to accommodate a large number of heterogeneous IoT devices. On the other hand, current blockchain consensus fail to sufficiently measure the contributions and destructions among heterogeneous devices in terms of learning quality, leading to low learning security and insufficient incentive fairness. To overcome the limitations of prior art, this paper introduces CoCFL, a novel blockchain-based federated learning framework based on multi-chain collaborative model. CoCFL enhances learning scalability by adopting a multi-chain asynchronous collaboration approach that partitions both learning and communication granularity of the system. Within each sub chain, CoCFL introduces a lightweight, secure and incentive-fair blockchain-based federated learning consensus, called Proof of Contribution to FL (PoCFL). In PoCFL, partic-ipants' contributions to the learning and the consensus process form the basis for delegating consensus responsibility and dis-tributing rewards. Furthermore, we introduce a novel malicious model detection algorithm into PoCFL, called the Trustee Nearest Algorithm. Through Trustee Nearest, PoCFL effectively mitigates poisoning attacks. Experimental results demonstrate that CoCFL exhibits better learning scalability compared to traditional FL and and avdanced BCFL frameworks in the same scenarios and can effectively withstand poisoning attacks initiated by at least 40% of malicious participants. Moreover, CoCFL demonstrated good incentive fairness during the learning process.