Efficient Lung Nodule Classification Using Transferable Texture Convolutional Neural Network

被引:35
作者
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
来源
IEEE ACCESS | 2020年 / 8卷
关键词
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
    Al-Shabi, Mundher
    Lan, Boon Leong
    Chan, Wai Yee
    Ng, Kwan-Hoong
    Tan, Maxine
    [J]. INTERNATIONAL JOURNAL OF COMPUTER ASSISTED RADIOLOGY AND SURGERY, 2019, 14 (10) : 1815 - 1819
  • [2] Using filter banks in Convolutional Neural Networks for texture classification
    Andrearczyk, Vincent
    Whelan, Paulf.
    [J]. PATTERN RECOGNITION LETTERS, 2016, 84 : 63 - 69
  • [3] [Anonymous], 2015, ARXIV PREPRINT ARXIV
  • [4] [Anonymous], 2017, P 34 INT C MACHINE L
  • [5] [Anonymous], 2019, ARXIV190100120
  • [6] [Anonymous], 2016, ARXIV160508350
  • [7] LUNGx Challenge for computerized lung nodule classification
    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.
    [J]. JOURNAL OF MEDICAL IMAGING, 2016, 3 (04)
  • [8] Special Section Guest Editorial: LUNGx Challenge for computerized lung nodule classification: Reflections and lessons learned
    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.
    [J]. Journal of Medical Imaging, 2015, 2 (02)
  • [9] The Lung Image Database Consortium, (LIDC) and Image Database Resource Initiative (IDRI): A Completed Reference Database of Lung Nodules on CT Scans
    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.
    [J]. MEDICAL PHYSICS, 2011, 38 (02) : 915 - 931
  • [10] Transfer learning on fused multiparametric MR images for classifying histopathological subtypes of rhabdomyosarcoma
    Banerjee, Imon
    Crawley, Alexis
    Bhethanabotla, Mythili
    Daldrup-Link, Heike E.
    Rubin, Daniel L.
    [J]. COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2018, 65 : 167 - 175