Svoboda | Graniru | BBC Russia | Golosameriki | Facebook
Preprint Article Version 1 Preserved in Portico This version is not peer-reviewed

A Comparative Analysis of Machine Learning Models for Lung Cancer Prediction: From Traditional Algorithms to Deep Learning Approaches

Version 1 : Received: 25 May 2024 / Approved: 27 May 2024 / Online: 27 May 2024 (13:18:48 CEST)

How to cite: Asghari Ilani, M.; Kavei, A.; Moftakhar Tehran, S. A Comparative Analysis of Machine Learning Models for Lung Cancer Prediction: From Traditional Algorithms to Deep Learning Approaches. Preprints 2024, 2024051742. https://doi.org/10.20944/preprints202405.1742.v1 Asghari Ilani, M.; Kavei, A.; Moftakhar Tehran, S. A Comparative Analysis of Machine Learning Models for Lung Cancer Prediction: From Traditional Algorithms to Deep Learning Approaches. Preprints 2024, 2024051742. https://doi.org/10.20944/preprints202405.1742.v1

Abstract

Driven by the importance of early detection for improving patient outcomes, this study investigates the effectiveness of various machine learning algorithms in predicting lung cancer based on clinical data. Lung cancer remains a significant global health concern, and this research seeks to contribute to advancements in early diagnosis through the application of machine learning techniques. The dataset consists of 385 training and 97 test samples, with a k-fold cross-validation approach (k=5) utilized to mitigate overfitting. Additionally, the ADASYN oversampling technique is employed to address class imbalance. Our analysis includes traditional algorithms such as Logistic Regression and Decision Trees, as well as ensemble methods like XGBoost, LGBM, and AdaBoost. Furthermore, we explore the effectiveness of deep learning models, particularly Deep Neural Networks (DNN). Results demonstrate promising accuracy rates across all models, with some outperforming others in terms of precision, recall, and F1-score. Our findings reveal that LGBM achieves the highest accuracy among all models, with an accuracy rate of 96.91%. It also exhibits superior performance in terms of precision, recall, and F1-score, particularly in classifying both positive and negative instances of lung cancer.

Keywords

XGBoost; LGBM; AdaBoost; KNN; ML; Lung cancer; DNN

Subject

Public Health and Healthcare, Health Policy and Services

Comments (0)

We encourage comments and feedback from a broad range of readers. See criteria for comments and our Diversity statement.

Leave a public comment
Send a private comment to the author(s)
* All users must log in before leaving a comment
Views 0
Downloads 0
Comments 0


×
Alerts
Notify me about updates to this article or when a peer-reviewed version is published.
We use cookies on our website to ensure you get the best experience.
Read more about our cookies here.