Performance Assessment of Deep Learning Models on Non-Small Cell Lung Cancer Type Classification

Authors

DOI:

https://doi.org/10.4108/eetpht.10.6423

Keywords:

Lung Cancer Classification, Lung Image Enhancement, Contrast Enhancement, Lung Feature Extraction, Lung Cancer Detection

Abstract

In recent years, lung cancer incidents are very high with equally high mortality rate. The main reason for fatal incidences is the late diagnosis and confirmation of the disease at an advanced stage. Identification of the disease at an early stage using lung Computed Tomography (CT) offers tremendous scope for timely medical intervention. The article illustrates the use of deep transfer learning-based pre-trained models for the diagnosis of Non-Small Cell Lung Cancer (NSCLC). The datasets were chosen from Chest CT Scan Images and the Lung Image Database Consortium (LIDC), containing over 3,179 images depicting three NSCLC types, namely, normal, adenocarcinoma, squamous cell carcinoma, and large cell carcinoma. The process is designed to measure the accuracy of NSCLC detection with an experimental dataset using approaches with and without pre-processing of lung images. The transfer learning models use deep learning and produce good results in prediction and classification. The image dataset was first handled by the convolutional neural networks DenseNet121, ResNet50, InceptionV3, VGG16, Xception, and VGG19. As a second phase, input images were subjected to contrast/brightness enhancement using Multi Level Dualistic Sub Image Histogram Equalization (ML-DSIHE). Enhanced images were further processed using shape-based feature extraction. Finally, those features input to CNN models and the results recorded. Among these models, VGG16 achieved the highest accuracy of 81.42% using the original dataset and 91.64% with the enhanced dataset. The performance of these two approaches was also evaluated using Precision, Recall, F1-Score, Accuracy, and Loss. It is confirmed that VGG16 gives highly reliable accuracy when trained upon enhanced images.

Downloads

Download data is not yet available.

References

A. Khalid et al., "Deep Learning Applications in Cancer Diagnosis and Prognosis," Computational and Structural Biotechnology Journal, vol. 18, pp. 8-17, 2020. doi: 10.1016/j.csbj.2019.12.010.

Wenya and Linda Bi, “Artificial Intelligence in Cancer Imaging: Clinical

Lakshmanaprabu, S. K., Mohanty, S. N., Shankar, K., Arunkumar, N., & Ramirez, G. (2019). Optimal deep learning model for classification of lung cancer on CT images. Future Generation Computer Systems, 92, 374-382.

Nasrullah, N., Sang, J., Alam, M. S., Mateen, M., Cai, B., & Hu, H. (2019). Automated lung nodule detection and classification using deep learning combined with multiple strategies. Sensors, 19(17), 3722.

Sajja, T., Devarapalli, R., &Kalluri, H. (2019). Lung Cancer Detection Based on CT Scan Images by Using Deep Transfer Learning. Traitement du Signal, 36(4), 339-344.

Sang, J., Alam, M. S., & Xiang, H. (2019, May). Automated detection and classification for early-stage lung cancer on CT images using deep learning. In Pattern Recognition and Tracking XXX (Vol. 10995, pp. 200-207). SPIE.

Bhandary, A., Prabhu, G. A., Rajinikanth, V., Thanaraj, K. P., Satapathy, S. C., Robbins, D. E., ... & Raja, N. S. M. (2020). Deep-learning framework to detect lung abnormality–A study with chest X-Ray and lung CT scan images. Pattern Recognition Letters, 129, 271-278.

Ardila, D., Kiraly, A. P., Bharadwaj, S., Choi, B., Reicher, J. J., Peng, L., Tse, D., Etemadi, M., Ye, W., Corrado, G. S., Naidich, D. P., Shetty, S., & Lungren, M. P. (2019). End-to-end lung cancer screening with three-dimensional deep learning on low-dose chest computed tomography. Nature Medicine, 25(6), 954-961.

Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. "Deep Residual Learning for Image Recognition." In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2016.

Gao Huang, Zhuang Liu, and Kilian Q. Weinberger. "Densely Connected Convolutional Networks." In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2017

Karen Simonyan and Andrew Zisserman. "Very Deep Convolutional Networks for Large-Scale Image Recognition." arXiv preprint arXiv:1409.1556 (2014).

Challenges and Applications”, A Cancer Journal for Clinicians, 2019

Zhang, L., Wang, S., & Liu, Q. (2019). A deep learning-based model for lung nodule classification. Computerized Medical Imaging and Graphics, 77, 101639

Wang, S., Shi, W., Chen, X., Jiang, R., & Song, Z. (2018). Lung nodule classification using deep residual learning based on CT images. Computer Methods and Programs in Biomedicine, 157, 19-30

Song, Y., Chen, Y., Liu, Z., & Jiang, H. (2020). Deep learning-based classification of lung nodules using a convolutional neural network with 3D reconstruction. Frontiers in Oncology, 10, 57

Li, X., Zhang, S., Zhang, Q., Wei, X., & Pan, Y. (2018). Lung nodule classification using deep convolutional neural networks on computed tomography images with additional color channels. Computers in biology and medicine, 102, 33-39

Yao, J., Wu, Y., Wang, Z., Zhang, J., & Zhang, Y. (2019). A support vector machine classifier with feature selection for lung nodule detection in CT images. Journal of healthcare engineering, 2019

Qu, J., Wang, S., Wang, Z., Tang, J., Zhang, Y., & Lv, X. (2020). A comparative study on the classification of lung nodules in CT images using deep learning methods. Journal of medical systems, 44(5)

Qu, J., Wang, S., Wang, Z., Tang, J., Zhang, Y., & Lv, X. (2020). A comparative study on the classification of lung nodules in CT images using deep learning methods. Journal of medical systems, 44(5)

Cui, L., Zhang, L., Guo, X., & Wang, Y. (2020). A deep learning-based approach for classification of lung nodules in CT images. BMC medical imaging, 20(1), 1-9

Lee, J. H., Lee, K., Lee, S. M., & Kang, M. J. (2019). A deep learning-based approach for automatic lung nodule detection and classification on CT images. Computers in Biology and Medicine, 109, 217-236. doi: 10.1016/j.compbiomed.2019.04.011

Simonyan, K., & Zisserman, A. (2014). Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556

U.S. National Institute of Health, National Cancer Institute. SEER Cancer Statistics Review, 1975–2015.

Sharma, M., Bhatt, J. S., & Joshi, M. V. (2018, April). Early detection of lung cancer from CT images: nodule segmentation and classification using deep learning. In Tenth International Conference on Machine Vision (ICMV 2017) (Vol. 10696, pp. 226-233). SPIE.

Mhaske, D., Rajeswari, K., &Tekade, R. (2019, September). Deep learning algorithm for classification and prediction of lung cancer using ct scan images. In 2019 5th International Conference on Computing, Communication, Control and Automation (ICCUBEA) (pp. 1-5). IEEE.

Francois Chollet. Xception: Deep Learning with Depthwise Separable Convolutions. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2017.

Szegedy, C., Vanhoucke, V., Ioffe, S., Shlens, J., & Wojna, Z. (2016). Rethinking the Inception architecture for computer vision. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 2818-2826).

Islam, M. T., Raskar, M. A., Islam, M. R., & Roy, C. K. (2020). A Deep Learning Approach for Lung Nodule Classification in CT Images. In 2020 IEEE International Conference on Innovations in Information Technology (IIT) (pp. 115-120). IEEE

Downloads

Published

24-06-2024

How to Cite

1.
Ezhilraja K, Shanmugavadivu P. Performance Assessment of Deep Learning Models on Non-Small Cell Lung Cancer Type Classification. EAI Endorsed Trans Perv Health Tech [Internet]. 2024 Jun. 24 [cited 2024 Jul. 13];10. Available from: https://publications.eai.eu/index.php/phat/article/view/6423