FedCare: Federated Learning for Resource-Constrained Healthcare Devices in IoMT System

ABSTRACT

In social IoMT systems, resource-constrained devices face the challenges of limited computation, bandwidth, and privacy in the deployment of deep learning models. Federated learning (FL) is one of the solutions to user privacy and provides distributed training among several local devices. In addition, it reduces the computation and bandwidth of transferring videos to the central server in camera-based IoMT devices. In this work, we design an edge-based federated framework for such devices. In contrast to traditional methods that drop the resource-constrained stragglers in a federated round, our system provides a methodology to incorporate them. We propose a new phase in the FL algorithm, known as split learning. The stragglers train collaboratively with the nearest edge node using split learning. We test the implementation using heterogeneous computing devices that extract vital signs from videos. The results show a reduction of 3.6 h in the training time of videos using the split learning phase with respect to the traditional approach. We also evaluate the performance of the devices and system with key parameters, CPU utilization, memory consumption, and data rate. Furthermore, we achieve 87.29% and 60.26% test accuracy at the nonstragglers and stragglers, respectively, with a global accuracy of 90.32% at the server. Therefore, FedCare provides a straggler-resistant federated method for a heterogeneous system for social IoMT devices.