Lung cancer detection using hybrid integration of autoencoder feature extraction and ML techniques

Annemneedi Lakshmanarao, Nirmal Gopal, Nagagopiraju Vullam, Mandapati Sridhar, Modalavalasa Krishna Kanth, Uma Maheswari Rayudu

Abstract


Lung cancer posed a significant global health challenge, necessitating innovative approaches for early detection and accurate diagnosis. In this paper, CT scan images for lung cancer with three classes namely benign, malignant, and normal are collected from Kaggle. We initially applied conventional machine learning (ML) algorithms including support vector machine (SVM), random forests (RF), decision trees (DT), logistic regression (LR), naive bayes (NB), and k-nearest neighbor for lung cancer detection. The results with these conventional algorithms are recorded. Later, we proposed a novel hybrid model that integrated diverse machine learning algorithms to further enhance accuracy. Our approach combined the power of autoencoders for feature extraction. Using Autoencoder technique, features from images are extracted and a new feature vector is created. Later, the same conventional ML classifiers applied and achieved enhanced performance. The hybrid model demonstrated remarkable performance in identifying lung cancer cases when compared to individual classifiers. Through extensive experimentation, we showcased the efficacy of our integrated framework, achieving high accuracy, precision, recall and F1-score metrics across multiple classifiers. This hybrid approach represented a significant advancement in lung cancer detection, offering a versatile and robust solution for early diagnosis and personalized treatment strategies in clinical settings.

Keywords


Autoencoder; Feature extraction; Kaggle; Lung cancer; ML techniques

Full Text:

PDF


DOI: http://doi.org/10.11591/ijeecs.v37.i1.pp416-424

Refbacks

  • There are currently no refbacks.


Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

Indonesian Journal of Electrical Engineering and Computer Science (IJEECS)
p-ISSN: 2502-4752, e-ISSN: 2502-4760
This journal is published by the Institute of Advanced Engineering and Science (IAES) in collaboration with Intelektual Pustaka Media Utama (IPMU).

shopify stats IJEECS visitor statistics