Abstract

Lung cancer has become one of the most common deaths amongst the cancer patients. World Health Organisation states that lung cancer is the second most fatal cancer all over the world in 2014. Alarmingly, most of the lung cancer patients are diagnosed at the later stages where the cancer has spreads. Thus, early screening via Computed Tomography scan particularly among active smokers is encouraged. Manual diagnosis of the cancer is made feasible through the integration of Computer Aided Diagnosis system. For the past few years, deep learning method leads most of the artificial based intelligence applications including CAD systems. This paper aims to investigate the performance of five newly established Convolutional Neural Network architectures; GoogleNet, SqueezeNet, DenseNet, ShuffleNet and MobileNetV2 to classify lung tumours into malignant and benign categories using LIDC-IDRI datasets. Their performances are measured in terms of accuracy, sensitivity, specificity and area under the curve of the receiver operating characteristic curve. Experimental results show that GoogleNet is the best CNN architecture for CT lung tumour classification wih an accuracy of 94.53%, specificity 99.06%, sensitivity of 65.67% and AUC 86.84%.

Keyword

Computed tomography, Convolution neural network, Deep learning, Lung cancer.

Cite this article

Ashhar SM, Mokri SS, Rahni AA, Huddin AB, Zulkarnain N, Azmi NA, Mahaletchumy T

Refference

[1][1]Kan Chan Siang MD, John CK. A review of lung cancer research in Malaysia. Medical Journal of Malaysia. 2016; 71(1):70-8.

[2][2]Rossi F, Mokri SS, Rahni AA. Development of a semi-automated combined PET and CT lung lesion segmentation framework. In medical imaging biomedical applications in molecular, structural, and functional imaging 2017 (p. 101370B). International Society for Optics and Photonics.

[3][3]Shukor MZ, Mokri SS. Comparison of segmentation techniques for nonsmall cell lung cancer (NSCLC) In PET images. Solid State Technology. 2020; 63(1s):958-66.

[4][4]Song Q, Zhao L, Luo X, Dou X. Using deep learning for classification of lung nodules on computed tomography images. Journal of Healthcare Engineering. 2017:1-8.

[5][5]Chan HP, Hadjiiski LM, Samala RK. Computer‐aided diagnosis in the era of deep learning. Medical Physics. 2020; 47(5): e218-27.

[6][6]Lee JG, Jun S, Cho YW, Lee H, Kim GB, Seo JB, Kim N. Deep learning in medical imaging: general overview. Korean Journal of Radiology. 2017; 18(4):570-84.

[7][7]Nasrullah N, Sang J, Alam MS, Mateen M, Cai B, Hu H. Automated lung nodule detection and classification using deep learning combined with multiple strategies. Sensors. 2019; 19(17):1-19.

[8][8]Shen S, Han SX, Aberle DR, Bui AA, Hsu W. An interpretable deep hierarchical semantic convolutional neural network for lung nodule malignancy classification. Expert Systems with Applications. 2019; 128:84-95.

[9][9]Abraham GK, Bhaskaran P, Jayanthi VS. Lung nodule classification in CT images using convolutional neural network. In 9th international conference on advances in computing and communication 2019 (pp. 199-203). IEEE.

[10][10]Tran GS, Nghiem TP, Nguyen VT, Luong CM, Burie JC. Improving accuracy of lung nodule classification using deep learning with focal loss. Journal of Healthcare Engineering. 2019: 1-9.

[11][11]Al-Shabi M, Lee HK, Tan M. Gated-dilated networks for lung nodule classification in CT scans. IEEE Access. 2019; 7:178827-38.

[12][12]Szegedy C, Liu W, Jia Y, Sermanet P, Reed S, Anguelov D, et al. Going deeper with convolutions. In proceedings of the IEEE conference on computer vision and pattern recognition 2015 (pp. 1-9).

[13][13]Iandola FN, Han S, Moskewicz MW, Ashraf K, Dally WJ, Keutzer K. SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and< 0.5 MB model size. arXiv preprint arXiv:1602.07360. 2016.

[14][14]Zhang X, Zhou X, Lin M, Sun J. Shufflenet: an extremely efficient convolutional neural network for mobile devices. In proceedings of the IEEE conference on computer vision and pattern recognition 2018 (pp. 6848-56).

[15][15]Huang G, Liu Z, Van Der Maaten L, Weinberger KQ. Densely connected convolutional networks. In proceedings of the IEEE conference on computer vision and pattern recognition 2017 (pp. 4700-8).

[16][16]Sandler M, Howard A, Zhu M, Zhmoginov A, Chen LC. Mobilenetv2: inverted residuals and linear bottlenecks. In proceedings of the IEEE conference on computer vision and pattern recognition 2018 (pp. 4510-20).

[17][17]Armato III SG, McLennan G, Bidaut L, McNitt‐Gray MF, Meyer CR, Reeves AP, et al. The lung image database consortium (LIDC) and image database resource initiative (IDRI): a completed reference database of lung nodules on CT scans. Medical Physics. 2011; 38(2):915-31.