OMS-CNN: Optimized Multi-Scale CNN for Lung Nodule Detection Based on Faster R-CNN

被引:2
作者
Zamanidoost, Yadollah [1 ]
Ould-Bachir, Tarek [1 ]
Martel, Sylvain [1 ]
机构
[1] Polytech Montreal, Dept Comp & Software Engn, Montreal, PQ H3T 1J4, Canada
基金
加拿大自然科学与工程研究理事会;
关键词
Lungs; Feature extraction; Computed tomography; Convolutional neural networks; Accuracy; Three-dimensional displays; Lung cancer; Bioinformatics; Sensitivity; Kernel; Lung nodule detection; Faster R-CNN; computed tomography images; multi-scale features; metaheuristic optimization; advanced parameter-setting-free harmony search; beetle antennae search; CONVOLUTIONAL NEURAL-NETWORK; PULMONARY NODULES; CLASSIFICATION; IMAGES;
D O I
10.1109/JBHI.2024.3507360
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The global increase in lung cancer cases, often marked by pulmonary nodules, underscores the critical importance of timely detection to mitigate cancer progression and reduce morbidity and mortality. The Faster R-CNN approach is a two-stage, high-precision nodule detection method designed for detecting small nodules, particularly in computed tomography (CT) images. This paper presents an improved Faster R-CNN by introducing an optimized multi-scale convolutional neural network (OMS-CNN) technique for feature map generation. This approach aims to achieve an optimal feature map through metaheuristic optimization by combining the last three layers of the VGG16 architecture. The advanced parameter-setting-free harmony search (PSF-HS) algorithm is utilized to implement this method, automatically adjusting the number of channels in the composite layers as a hyperparameter. The beetle antenna search (BAS) optimization algorithm is utilized to effectively initialize the kernel filter weights and biases in the composite layers, thereby enhancing training speed and detection accuracy. In the false-positive reduction stage, a combination of multiple 3D deep convolutional neural networks (3D DCNN) is designed to reduce false-positive nodules. The proposed model was evaluated using the LUNA16 and PN9 datasets. The results demonstrate that the OMS-CNN technique effectively extracted representative features of nodules at various sizes, achieving a sensitivity of 94.89% and a CPM score of 0.892. The comprehensive experiments illustrate that the proposed method can enhance detection sensitivity and manage the number of false positive nodules, thereby offering clinical utility and serving as a valuable point of reference.
引用
收藏
页码:2148 / 2160
页数:13
相关论文
共 53 条
[21]   Feature Pyramid Networks for Object Detection [J].
Lin, Tsung-Yi ;
Dollar, Piotr ;
Girshick, Ross ;
He, Kaiming ;
Hariharan, Bharath ;
Belongie, Serge .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :936-944
[22]   SSD: Single Shot MultiBox Detector [J].
Liu, Wei ;
Anguelov, Dragomir ;
Erhan, Dumitru ;
Szegedy, Christian ;
Reed, Scott ;
Fu, Cheng-Yang ;
Berg, Alexander C. .
COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 :21-37
[23]   TiCNet: Transformer in Convolutional Neural Network for Pulmonary Nodule Detection on CT Images [J].
Ma, Ling ;
Li, Gen ;
Feng, Xingyu ;
Fan, Qiliang ;
Liu, Lizhi .
JOURNAL OF IMAGING INFORMATICS IN MEDICINE, 2024, 37 (01) :196-208
[24]   SANet: A Slice-Aware Network for Pulmonary Nodule Detection [J].
Mei, Jie ;
Cheng, Ming-Ming ;
Xu, Gang ;
Wan, Lan-Ruo ;
Zhang, Huan .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2022, 44 (08) :4374-4387
[25]   Lung nodule detection of CT images based on combining 3D-CNN and squeeze-and-excitation networks [J].
Mkindu, Hassan ;
Wu, Longwen ;
Zhao, Yaqin .
MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 82 (17) :25747-25760
[26]   3-D Convolutional Neural Networks for Automatic Detection of Pulmonary Nodules in Chest CT [J].
Pezeshk, Aria ;
Hamidian, Sardar ;
Petrick, Nicholas ;
Sahiner, Berkman .
IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2019, 23 (05) :2080-2090
[27]  
Qi Dou, 2017, Medical Image Computing and Computer Assisted Intervention MICCAI 2017. 20th International Conference. Proceedings: LNCS 10435, P630, DOI 10.1007/978-3-319-66179-7_72
[28]   Generalized Intersection over Union: A Metric and A Loss for Bounding Box Regression [J].
Rezatofighi, Hamid ;
Tsoi, Nathan ;
Gwak, JunYoung ;
Sadeghian, Amir ;
Reid, Ian ;
Savarese, Silvio .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :658-666
[29]   ImageNet Large Scale Visual Recognition Challenge [J].
Russakovsky, Olga ;
Deng, Jia ;
Su, Hao ;
Krause, Jonathan ;
Satheesh, Sanjeev ;
Ma, Sean ;
Huang, Zhiheng ;
Karpathy, Andrej ;
Khosla, Aditya ;
Bernstein, Michael ;
Berg, Alexander C. ;
Fei-Fei, Li .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2015, 115 (03) :211-252
[30]   Validation, comparison, and combination of algorithms for automatic detection of pulmonary nodules in computed tomography images: The LUNA16 challenge [J].
Setio, Arnaud Arindra Adiyoso ;
Traverso, Alberto ;
de Bel, Thomas ;
Berens, Moira S. N. ;
van den Bogaard, Cas ;
Cerello, Piergiorgio ;
Chen, Hao ;
Dou, Qi ;
Evelina Fantacci, Maria ;
Geurts, Bram ;
van der Gugten, Robbert ;
Heng, Pheng Ann ;
Jansen, Bart ;
de Kaste, Michael M. J. ;
Kotov, Valentin ;
Lin, Jack Yu-Hung ;
Manders, Jeroen T. M. C. ;
Sonora-Mengana, Alexander ;
Carlos Garcia-Naranjo, Juan ;
Papavasileiou, Evgenia ;
Prokop, Mathias ;
Saletta, Marco ;
Schaefer-Prokop, Cornelia M. ;
Scholten, Ernst T. ;
Scholten, Luuk ;
Snoeren, Miranda M. ;
Lopez Torres, Ernesto ;
Vandemeulebroucke, Jef ;
Walasek, Nicole ;
Zuidhof, Guido C. A. ;
van Ginneken, Bram ;
Jacobs, Colin .
MEDICAL IMAGE ANALYSIS, 2017, 42 :1-13