Ensemble Federated Learning Approach for Diagnostics of Multi-Order Lung Cancer

Diagnostics (Basel). 2023 Sep 25;13(19):3053. doi: 10.3390/diagnostics13193053.

Abstract

The early detection and classification of lung cancer is crucial for improving a patient's outcome. However, the traditional classification methods are based on single machine learning models. Hence, this is limited by the availability and quality of data at the centralized computing server. In this paper, we propose an ensemble Federated Learning-based approach for multi-order lung cancer classification. This approach combines multiple machine learning models trained on different datasets allowing for improvising accuracy and generalization. Moreover, the Federated Learning approach enables the use of distributed data while ensuring data privacy and security. We evaluate the approach on a Kaggle cancer dataset and compare the results with traditional machine learning models. The results demonstrate an accuracy of 89.63% with lung cancer classification.

Keywords: decentralized computation; diagnostics; federated learning models; lung cancer classification; optimization; thresholding.

Grants and funding

This research received no external funding.