Blockchain-Based Two-Stage Federated Learning With Non-IID Data in IoMT System

【Author】 Lian, Zhuotao; Zeng, Qingkui; Wang, Weizheng; Gadekallu, Thippa Reddy; Su, Chunhua

【Source】IEEE TRANSACTIONS ON COMPUTATIONAL SOCIAL SYSTEMS

【影响因子】4.747

【Abstract】The Internet of Medical Things (IoMT) has a bright future with the development of smart mobile devices. Information technology is also leading changes in the healthcare industry. IoMT devices can detect patient signs and provide treatment guidance and even instant diagnoses through technologies, such as artificial intelligence (AI) and wireless communication. However, conventional centralized machine learning approaches are often difficult to apply within IoMT devices because of the difficulty of large-scale collection of patient data and the potential risk of privacy breaches. Therefore, we propose a blockchain-based two-stage federated learning approach that allows IoMT devices to train a global model collaboratively without gathering the data to a central server. Specifically, to address the problem of poor training performance on non-independent identically distributed (non-IID) data, we design a blockchain-based data-sharing scheme that can significantly improve the model's accuracy without threatening user privacy. We also design a client selection mechanism to further improve the system's efficiency. Finally, we validate the feasibility and effectiveness of our system through simulation experiments on three popular datasets (i.e., MNIST, Fashion-MNIST, and CIFAR-10).

【Keywords】Blockchain; federated learning; Internet of Medical Things (IoMT); non-independent identically distributed (non-IID) data; privacy preservation

【发表时间】

【收录时间】2022-12-17

【文献类型】实验仿真

【主题类别】

区块链应用-实体经济-医疗领域