Federated Learning for Bronchus Cancer Detection Using Tiny Machine Learning Edge Devices

Abstract

In deep learning, acquiring sufficient data is crucial for making informed decisions. However, due to concerns regarding security and privacy, obtaining enough data for training models in the era of deep learning is challenging. There is a growing need for machine learning (ML) solutions that can derive accurate conclusions from small data while preserving privacy. Smartphones, which are widely used and generate large amounts of data, can serve as an excellent source for data generation. One suitable approach for regularly evaluating real-world data from edge devices is Tiny Machine Learning (TinyML). With the increasing number of edge devices involved in transmitting private data, it's vital to have a method that allows computations to be performed on edge devices and pushed to the edge rather than over the network. Considering these obstacles, the combination of TinyML edge devices and Federated Learning can be applied in the early treatment of Bronchus Cancer. Under the framework of federated learning, local edge devices are trained independently and then integrated into the server without exchanging edge device data. This approach enables the creation of secure models without sharing information, resulting in a highly efficient solution with enhanced data security and accessibility. This article provides a comprehensive discussion of the key challenges addressed in recent literature, accompanied by an extensive examination of relevant studies. Additionally, a novel model based on edge devices and federated learning is proposed.