Tensor-RT-Based Transfer Learning Model for Lung Cancer Classification

被引:12
作者
Bishnoi, Vidhi [1 ]
Goel, Nidhi [1 ]
机构
[1] Indira Gandhi Delhi Tech Univ Women, New Delhi, Delhi, India
关键词
DICOM; Computed tomography; Lung cancer; Computer-aided techniques; Transfer learning; Nvidia tensor-RT; PULMONARY NODULE DETECTION; CT IMAGES;
D O I
10.1007/s10278-023-00822-z
中图分类号
R8 [特种医学]; R445 [影像诊断学];
学科分类号
1002 ; 100207 ; 1009 ;
摘要
Cancer is a leading cause of death across the globe, in which lung cancer constitutes the maximum mortality rate. Early diagnosis through computed tomography scan imaging helps to identify the stages of lung cancer. Several deep learning-based classification methods have been employed for developing automatic systems for the diagnosis and detection of computed tomography scan lung slices. However, the diagnosis based on nodule detection is a challenging task as it requires manual annotation of nodule regions. Also, these computer-aided systems have yet not achieved the desired performance in real-time lung cancer classification. In the present paper, a high-speed real-time transfer learning-based framework is proposed for the classification of computed tomography lung cancer slices into benign and malignant. The proposed framework comprises of three modules: (i) pre-processing and segmentation of lung images using K-means clustering based on cosine distance and morphological operations; (ii) tuning and regularization of the proposed model named as weighted VGG deep network (WVDN); (iii) model inference in Nvidia tensor-RT during post-processing for the deployment in real-time applications. In this study, two pre-trained CNN models were experimented and compared with the proposed model. All the models have been trained on 19,419 computed tomography scan lung slices, which were obtained from the publicly available Lung Image Database Consortium and Image Database Resource Initiative dataset. The proposed model achieved the best classification metric, an accuracy of 0.932, precision, recall, an F1 score of 0.93, and Cohen's kappa score of 0.85. A statistical evaluation has also been performed on the classification parameters and achieved a p-value <0.0001 for the proposed model. The quantitative and statistical results validate the improved performance of the proposed model as compared to state-of-the-art methods. The proposed framework is based on complete computed tomography slices rather than the marked annotations and may help in improving clinical diagnosis.
引用
收藏
页码:1364 / 1375
页数:12
相关论文
共 49 条
[1]   Automatic classification of solitary pulmonary nodules in PET/CT imaging employing transfer learning techniques [J].
Apostolopoulos, Ioannis D. ;
Pintelas, Emmanuel G. ;
Livieris, Ioannis E. ;
Apostolopoulos, Dimitris J. ;
Papathanasiou, Nikolaos D. ;
Pintelas, Panagiotis E. ;
Panayiotakis, George S. .
MEDICAL & BIOLOGICAL ENGINEERING & COMPUTING, 2021, 59 (06) :1299-1310
[2]   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
[3]   Performance of a Novel Chaotic Firefly Algorithm with Enhanced Exploration for Tackling Global Optimization Problems: Application for Dropout Regularization [J].
Bacanin, Nebojsa ;
Stoean, Ruxandra ;
Zivkovic, Miodrag ;
Petrovic, Aleksandar ;
Rashid, Tarik A. ;
Bezdan, Timea .
MATHEMATICS, 2021, 9 (21)
[4]  
Bezdan Timea, 2021, Intelligent and Fuzzy Techniques: Smart and Innovative Solutions. Proceedings of the INFUS 2020 Conference. Advances in Intelligent Systems and Computing (AISC 1197), P955, DOI 10.1007/978-3-030-51156-2_111
[5]   Lung nodules detection using grey wolf optimization by weighted filters and classification using CNN [J].
Bilal, Anas ;
Sun, Guangmin ;
Li, Yu ;
Mazhar, Sarah ;
Latif, Jahanzaib .
JOURNAL OF THE CHINESE INSTITUTE OF ENGINEERS, 2022, 45 (02) :175-186
[6]   Wrapper-Based Best Feature Selection Approach for Lung Cancer Detection [J].
Bishnoi, Vidhi ;
Goel, Nidhi ;
Tayal, Akash .
ARTIFICIAL INTELLIGENCE AND SUSTAINABLE COMPUTING FOR SMART CITY, AIS2C2 2021, 2021, 1434 :175-186
[7]  
Bishnoi Vidhi, 2022, INT J SYST ASSUR ENG, V1, P01, DOI [10.1504/IJMEI.2022.10047638, DOI 10.1504/IJMEI.2022.10047638]
[8]   Integration of convolutional neural networks for pulmonary nodule malignancy assessment in a lung cancer classification pipeline [J].
Bonavita, Ilaria ;
Rafael-Palou, Xavier ;
Ceresa, Mario ;
Piella, Gemma ;
Ribas, Vicent ;
Gonzalez Ballester, Miguel A. .
COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2020, 185
[9]   The Cancer Imaging Archive (TCIA): Maintaining and Operating a Public Information Repository [J].
Clark, Kenneth ;
Vendt, Bruce ;
Smith, Kirk ;
Freymann, John ;
Kirby, Justin ;
Koppel, Paul ;
Moore, Stephen ;
Phillips, Stanley ;
Maffitt, David ;
Pringle, Michael ;
Tarbox, Lawrence ;
Prior, Fred .
JOURNAL OF DIGITAL IMAGING, 2013, 26 (06) :1045-1057
[10]   A practical tutorial on the use of nonparametric statistical tests as a methodology for comparing evolutionary and swarm intelligence algorithms [J].
Derrac, Joaquin ;
Garcia, Salvador ;
Molina, Daniel ;
Herrera, Francisco .
SWARM AND EVOLUTIONARY COMPUTATION, 2011, 1 (01) :3-18