Advanced artificial intelligence driven framework for lung cancer diagnosis leveraging SqueezeNet with machine learning algorithms using transfer learning

Abstract

Lung cancer is a severe global public health problem, and early detection is important for improving patient wellbeing. This research presents an advanced Artificial Intelligence (AI) driven framework that integrates deep learning and machine learning techniques to enhance lung cancer classification in chest Computed Tomography (CT) scans. Leveraging transfer learning, SqueezeNet a lightweight Convolutional Neural Network (CNN) is employed for feature extraction, which is then processed by Machine Learning (ML) classifiers. A dataset comprising 950 chest scans from 110 test cases is used to classify tumors into benign, malignant, and normal categories. Among the tested models, SqueezeNet combined with Logistic Regression (LR) achieves the highest accuracy of 92.9 ​%. Performance evaluation is conducted using multiple classification metrics, including Confusion Matrix and Calibration plots, demonstrating the model's reliability in early lung cancer detection. The proposed AI-driven hybrid framework offers a promising approach to improving diagnostic accuracy, ultimately benefiting both patients and the healthcare system.