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 条
[1]   Two-stage lung nodule detection framework using enhanced UNet and convolutional LSTM networks in CT images [J].
Agnes, S. Akila ;
Anitha, J. ;
Solomon, A. Arun .
COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 149
[2]  
Almahasneh M, 2024, Arxiv, DOI arXiv:2407.14464
[3]  
Bauckhage C., 2017, Tech. Rep
[4]  
Bhavanishankar K., 2015, International Journal on Cybernetics Informatics, V4, P27, DOI DOI 10.5121/IJCI.2015.4103
[5]   Cascade R-CNN: Delving into High Quality Object Detection [J].
Cai, Zhaowei ;
Vasconcelos, Nuno .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :6154-6162
[6]   A Novel Convolutional Neural Network Model Based on Beetle Antennae Search Optimization Algorithm for Computerized Tomography Diagnosis [J].
Chen, Dechao ;
Li, Xiang ;
Li, Shuai .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2023, 34 (03) :1418-1429
[7]   A Convolutional Neural Network Approach for Post-Processing in HEVC Intra Coding [J].
Dai, Yuanying ;
Liu, Dong ;
Wu, Feng .
MULTIMEDIA MODELING (MMM 2017), PT I, 2017, 10132 :28-39
[8]  
Dobrenkii A, 2017, 2017 IEEE CONFERENCE ON COMPUTATIONAL INTELLIGENCE IN BIOINFORMATICS AND COMPUTATIONAL BIOLOGY (CIBCB), P190
[9]   Parameter-setting-free harmony search algorithm [J].
Geem, Zong Woo ;
Sim, Kwee-Bo .
APPLIED MATHEMATICS AND COMPUTATION, 2010, 217 (08) :3881-3889
[10]   Automatic lung nodule detection using a 3D deep convolutional neural network combined with a multi-scale prediction strategy in chest CTs [J].
Gu, Yu ;
Lu, Xiaoqi ;
Yang, Lidong ;
Zhang, Baohua ;
Yu, Dahua ;
Zhao, Ying ;
Gao, Lixin ;
Wu, Liang ;
Zhou, Tao .
COMPUTERS IN BIOLOGY AND MEDICINE, 2018, 103 :220-231