Comparative Analysis of Machine Learning and Deep Learning Models for Lung Cancer Prediction Based on Symptomatic and Lifestyle Features
Source abstract
Abstract
Lung cancer remains a leading cause of global mortality, with early detection being critical for improving the patient survival rates. However, applying machine learning and deep learning effectively for lung cancer prediction using symptomatic and lifestyle data requires the careful consideration of feature selection and model optimization, which is not consistently addressed in the existing research. This research addresses this gap by systematically evaluating and comparing the predictive efficacy of several machine learning and deep learning models, employing rigorous data preprocessing, including feature selection with Pearson’s correlation, outlier removal, and normalization, on a patient symptom and lifestyle factor dataset from Kaggle. Machine learning classifiers, including Decision Trees, K-Nearest Neighbors, Random Forest, Naïve Bayes, AdaBoost, Logistic Regression, and Support Vector Machines, were implemented using Weka simultaneously with neural network models with 1, 2, and 3 hidden layers, which were developed in Python within a Jupyter Notebook environment. The model performance was assessed using K-fold cross-validation and 80/20 train/test splitting. The results highlight the importance of feature selection for enhancing the model accuracy and demonstrate that the single-hidden-layer neural network, trained for 800 epochs, achieved a prediction accuracy of 92.86%, outperforming the machine learning models. This study contributes to developing more effective computational methods for early lung cancer detection, ultimately supporting improved patient outcomes.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Radiomics and Machine Learning in Medical Imaging · Artificial Intelligence in Healthcare · AI in cancer detection
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department of Information Technology and Management, English Taught Program in Smart Service Management, Shih Chien University, Taipei 104, Taiwan
References (25)
Show source-provided references
1. Maurya, S.P., Sisodia, P.S., Mishra, R., and Singh, D.P. (2024). Performance of machine learning algorithms for lung cancer prediction: A comparative approach. Sci. Rep., 14.10.1038/s41598-024-58345-8 ↗
2. Lung microbiome: New insights into the pathogenesis of respiratory diseases10.1038/s41392-023-01722-y ↗
3. Garg, P., Singhal, S., Kulkarni, P., Horne, D., Malhotra, J., Salgia, R., and Singhal, S.S. (2024). Advances in Non-Small Cell Lung Cancer: Current Insights and Future Directions. J. Clin. Med., 13.10.3390/jcm13144189 ↗
4. (2025, January 05). CDC, Available online: https://www.cdc.gov/lung-cancer/risk-factors/index.html#:~:text=In%20the%20United%20States%2C%20cigarette,Many%20are%20poisons.
5. Liu, Y., Geng, Q., and Lin, X. (2024). Benefits, harms, and cost-effectiveness of risk model-based and risk factor-based low-dose computed tomography screening strategies for lung cancer: A systematic review. BMC Cancer, 24.10.1186/s12885-024-13356-6 ↗
6. Chakraborty, C., Bhattacharya, M., Pal, S., and Lee, S.S. (2024). From machine learning to deep learning: Advances of the recent data-driven paradigm shift in medicine and healthcare. Curr. Res. Biotechnol., 7.10.1016/j.crbiot.2023.100164 ↗
7. Sogandi, F. (2024). Identifying diseases symptoms and general rules using supervised and unsupervised machine learning. Sci. Rep., 14.10.1038/s41598-024-69029-8 ↗
8. Review of deep learning: Concepts, CNN architectures, challenges, applications, future directions10.1186/s40537-021-00444-8 ↗
9. Liu, Y., Wu, R., and Yang, A. (2023). Research on Medical Problems Based on Mathematical Models. Mathematics, 11.10.3390/math11132842 ↗
10. Lung cancer immunotherapy: Progress, pitfalls, and promises10.1186/s12943-023-01740-y ↗
11. A comparison of machine learning algorithms for diabetes prediction10.1016/j.icte.2021.02.004 ↗
12. Protić, D., Stanković, M., Prodanović, R., Vulić, I., Stojanović, G.M., Simić, M., Ostojić, G., and Stankovski, S. (2023). Numerical Feature Selection and Hyperbolic Tangent Feature Scaling in Machine Learning-Based Detection of Anomalies in the Computer Network Behavior. Electronics, 12.10.3390/electronics12194158 ↗
13. Graphical representation of data prediction potential: Correlation graphs and correlation chains10.1007/s00371-023-03240-y ↗
14. Patra, R. (2020, January 26–27). Prediction of lung cancer using machine learning classifier. Proceedings of the International Conference on Computing Science, Communication and Security, Gujarat, India.10.1007/978-981-15-6648-6_11 ↗
15. Radhika, P., Nair, R.A., and Veena, G. (2019, January 20–22). A comparative study of lung cancer detection using machine learning algorithms. Proceedings of the 2019 IEEE International Conference on Electrical, Computer and Communication Technologies (ICECCT), Prague, Czech Republic.
16. Faisal, M.I., Bashir, S., Khan, Z.S., and Khan, F.H. (2018, January 18–20). An evaluation of machine learning classifiers and ensembles for early stage prediction of lung cancer. Proceedings of the 2018 3rd International Conference on Emerging Trends in Engineering, Sciences and Technology (ICEEST), Thrissur, Kerala, India.10.1109/ICEEST.2018.8643311 ↗
17. Deep learning-enabled medical computer vision10.1038/s41746-020-00376-2 ↗
18. Vieira, E., Ferreira, D., Neto, C., Abelha, A., and Machado, J. (2021). Data Mining Approach to Classify Cases of Lung Cancer. World Conference on Information Systems and Technologies, Springer.10.1007/978-3-030-72657-7_49 ↗
19. Aldoseri, A., Al-Khalifa, K.N., and Hamouda, A.M. (2023). Re-Thinking Data Strategy and Integration for Artificial Intelligence: Concepts, Opportunities, and Challenges. Appl. Sci., 13.10.20944/preprints202305.1565.v2 ↗
20. Jiang, D., Wang, Z., and Li, H. (2024). Experiences of patient delay among lung cancer patients in South China. BMC Cancer, 24.10.1186/s12885-024-13295-2 ↗
21. Vij, A., and Kaswan, K.S. (2023, January 27–29). Prediction of lung cancer using convolution neural networks. Proceedings of the 2023 IEEE International Conference on Artificial Intelligence and Smart Communication (AISC), Greater Noida, India.10.1109/AISC56616.2023.10085058 ↗
22. Performance of Lung Cancer Prediction Methods Using Different Classification Algorithms
23. SVMVGGNet-16: A Novel Machine and Deep Learning Based Approaches for Lung Cancer Detection using Combined SVM and VGGNet-16
24. Prediction of Lung Cancer using Ensemble Classifiers10.1088/1742-6596/2161/1/012007 ↗
25. Prediction of survival and progression-free survival using machine learning in stage III lung cancer: A pilot study