Efficient Lung Nodule Classification Using Transferable Texture Convolutional Neural Network

被引:42
作者
Ali, Imdad [1 ,2 ]
Muzammil, Muhammad [2 ]
Ul Haq, Ihsan [2 ]
Khaliq, Amir A. [2 ]
Abdullah, Suheel [2 ]
机构
[1] Natl Ctr Phys, Islamabad 44000, Pakistan
[2] Int Islamic Univ, Fac Engn & Technol, Islamabad 44000, Islamabad, Pakistan
关键词
Lung; Feature extraction; Cancer; Databases; Computed tomography; Biomedical imaging; Task analysis; cancer detection; computer aided diagnosis; image classification; machine learning; transfer learning; lung nodule; CNN; LIDC-IDRI; LUNGx challenge; COMPUTER-AIDED DIAGNOSIS; SHAPE;
D O I
10.1109/ACCESS.2020.3026080
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Lung nodules are vital indicators for the presence of lung cancer. An early detection enhances the survival rate of the patient by starting treatment at the right time. The detection and classification of malignancy in Computed Tomography (CT) images is a very time-consuming and difficult task for radiologists which lead the researchers to develop algorithms for Computer-Aided Diagnosis (CAD) systems to mitigate this burden. The performance of CAD systems is continuously improving by using various deep learning techniques for screening of lung cancer. In this paper, we proposed transferable texture Convolutional Neural Networks (CNN) to improve the classification performance of pulmonary nodules in CT scans. An Energy Layer (EL) is incorporated in our scheme, which extracts texture features from the convolutional layer. The inclusion of EL reduces the number of learnable parameters of the network, which further reduces the memory requirements and computational complexity. The proposed model has only three convolutional layers and one EL, instead of pooling layer. Overall proposed CNN architecture comprises of nine layers for automatic feature extraction and classification of pulmonary nodule candidates as malignant or benign. Furthermore, the pre-trained model of proposed CNN is also used to handle the smaller dataset classification problem by using transfer learning. This work has been evaluated on publicly available LIDC-IDRI and the LUNGx Challenge database through different evaluation matrices, such as; the accuracy, specificity, error rate and AUC. The proposed model is trained by six-fold cross-validation and achieved an accuracy score of 96.69%+/- 0.72% with only 3.30%+/- 0.72% error rate. Whereas, the measured AUC and recall is 99.11%+/- 0.45% and 97.19%+/- 0.57%, respectively. Moreover, we also tested our proposed technique on the MNIST dataset and achieved state-of-the-art results in terms of accuracy and error rate.
引用
收藏
页码:175859 / 175870
页数:12
相关论文
共 55 条
[1]   Lung nodule classification using deep Local-Global networks [J].
Al-Shabi, Mundher ;
Lan, Boon Leong ;
Chan, Wai Yee ;
Ng, Kwan-Hoong ;
Tan, Maxine .
INTERNATIONAL JOURNAL OF COMPUTER ASSISTED RADIOLOGY AND SURGERY, 2019, 14 (10) :1815-1819
[2]   Using filter banks in Convolutional Neural Networks for texture classification [J].
Andrearczyk, Vincent ;
Whelan, Paulf. .
PATTERN RECOGNITION LETTERS, 2016, 84 :63-69
[3]  
[Anonymous], ICML
[4]  
[Anonymous], 2017, P 34 INT C MACHINE L
[5]  
[Anonymous], 2019, ARXIV190100120
[6]  
[Anonymous], 2016, ARXIV160508350
[7]  
[Anonymous], 2018, ARXIV180600712
[8]   LUNGx Challenge for computerized lung nodule classification [J].
Armato, Samuel G., III ;
Drukker, Karen ;
Li, Feng ;
Hadjiiski, Lubomir ;
Tourassi, Georgia D. ;
Engelmann, Roger M. ;
Giger, Maryellen L. ;
Redmond, George ;
Farahani, Keyvan ;
Kirby, Justin S. ;
Clarke, Laurence P. .
JOURNAL OF MEDICAL IMAGING, 2016, 3 (04)
[9]   Special Section Guest Editorial: LUNGx Challenge for computerized lung nodule classification: Reflections and lessons learned [J].
Armato, Samuel G. ;
Hadjiiski, Lubomir ;
Tourassi, Georgia D. ;
Drukker, Karen ;
Giger, Maryellen L. ;
Li, Feng ;
Redmond, George ;
Farahani, Keyvan ;
Kirby, Justin S. ;
Clarke, Laurence P. .
Journal of Medical Imaging, 2015, 2 (02)
[10]   The Lung Image Database Consortium, (LIDC) and Image Database Resource Initiative (IDRI): A Completed Reference Database of Lung Nodules on CT Scans [J].
Armato, Samuel G., III ;
McLennan, Geoffrey ;
Bidaut, Luc ;
McNitt-Gray, Michael F. ;
Meyer, Charles R. ;
Reeves, Anthony P. ;
Zhao, Binsheng ;
Aberle, Denise R. ;
Henschke, Claudia I. ;
Hoffman, Eric A. ;
Kazerooni, Ella A. ;
MacMahon, Heber ;
van Beek, Edwin J. R. ;
Yankelevitz, David ;
Biancardi, Alberto M. ;
Bland, Peyton H. ;
Brown, Matthew S. ;
Engelmann, Roger M. ;
Laderach, Gary E. ;
Max, Daniel ;
Pais, Richard C. ;
Qing, David P-Y ;
Roberts, Rachael Y. ;
Smith, Amanda R. ;
Starkey, Adam ;
Batra, Poonam ;
Caligiuri, Philip ;
Farooqi, Ali ;
Gladish, Gregory W. ;
Jude, C. Matilda ;
Munden, Reginald F. ;
Petkovska, Iva ;
Quint, Leslie E. ;
Schwartz, Lawrence H. ;
Sundaram, Baskaran ;
Dodd, Lori E. ;
Fenimore, Charles ;
Gur, David ;
Petrick, Nicholas ;
Freymann, John ;
Kirby, Justin ;
Hughes, Brian ;
Casteele, Alessi Vande ;
Gupte, Sangeeta ;
Sallam, Maha ;
Heath, Michael D. ;
Kuhn, Michael H. ;
Dharaiya, Ekta ;
Burns, Richard ;
Fryd, David S. .
MEDICAL PHYSICS, 2011, 38 (02) :915-931