IoT Federated Blockchain Learning at the Edge

James Calo,Benny Lo

2023-04-06

Abstract:IoT devices are sorely underutilized in the medical field, especially within machine learning for medicine, yet they offer unrivaled benefits. IoT devices are low-cost, energy-efficient, small and intelligent devices. In this paper, we propose a distributed federated learning framework for IoT devices, more specifically for IoMT (Internet of Medical Things), using blockchain to allow for a decentralized scheme improving privacy and efficiency over a centralized system; this allows us to move from the cloud-based architectures, that are prevalent, to the edge. The system is designed for three paradigms: 1) Training neural networks on IoT devices to allow for collaborative training of a shared model whilst decoupling the learning from the dataset to ensure privacy. Training is performed in an online manner simultaneously amongst all participants, allowing for the training of actual data that may not have been present in a dataset collected in the traditional way and dynamically adapt the system whilst it is being trained. 2) Training of an IoMT system in a fully private manner such as to mitigate the issue with confidentiality of medical data and to build robust, and potentially bespoke, models where not much, if any, data exists. 3) Distribution of the actual network training, something federated learning itself does not do, to allow hospitals, for example, to utilize their spare computing resources to train network models.

Machine Learning,Cryptography and Security

What problem does this paper attempt to address?

The paper primarily proposes a new distributed federated learning framework for Internet of Medical Things (IoMT) devices in the healthcare field, aiming to address the following core issues:

1. **Privacy Protection**: Traditional centralized machine learning methods pose a risk of privacy leakage when handling medical data. The proposed method combines federated learning with blockchain technology to achieve a decentralized training process, thereby ensuring data privacy and security.

2. **Edge Computing**: The goal of the paper is to shift from cloud-based architecture to edge-based computing architecture to reduce dependency on network connections, improve system response speed, and alleviate network congestion issues.

3. **Utilization of Computing Resources**: Existing IoMT devices are typically used only for inference and cannot perform model training, which limits their application scope. The proposed method allows these devices to participate in model training, fully utilizing their computing resources.

4. **Improvement of Model Generalization**: By involving different medical institutions in the federated learning process, more generalized models can be trained. This approach not only improves model accuracy but also allows each institution to conduct customized training based on its own needs.

5. **Hardware Limitations**: Existing federated learning frameworks often require specific hardware support (such as GPUs), which increases the cost and energy consumption of devices. The proposed method can run on ordinary CPUs, reducing hardware requirements and enabling more devices to participate.

In summary, this research aims to develop an efficient, secure, and private federated learning system for medical IoMT devices to promote the digital transformation and technological advancement of the healthcare industry.