联营
计算机科学
人工智能
模式识别(心理学)
医学影像学
肺癌筛查
肺癌
计算机视觉
数据挖掘
医学
病理
作者
Zhanlin Ji,Yun Wang,Xinyi Zeng,Yongli An,Li Zhao,Zhiwu Wang,Иван Ганчев
出处
期刊:IEEE Access
[Institute of Electrical and Electronics Engineers]
日期:2023-01-01
卷期号:11: 76371-76387
被引量:5
标识
DOI:10.1109/access.2023.3296530
摘要
Lung cancer has become one of the malignant tumors with the highest morbidity and mortality rate worldwide.The early images of lung cancer are pulmonary nodules, and early detection and diagnosis can help reduce the incidence of lung cancer.However, due to the large differences in shape, size and location of pulmonary nodules in medical imaging, the detection of small nodules is very challenging.This paper proposes a new model, based on improved YOLOv5s, called YOLOv5-CASP.The proposed YOLOv5s improvements include: (i) incorporating Convolutional Block Attention Modules (CBAM) to suppress the interference features of the medical images through a channel dimension and spatial dimension, and to improve the detection accuracy of the model; (ii) optimizing the Spatial Pyramid Pooling -Fast (SPPF) module of YOLOv5s to Atrous Spatial Pyramid Pooling (ASPP) as to increase the model's receptive field of images of different sizes and extract multi-scale contextual information for improving the detection performance on small lung nodules; and (iii) introducing a Contextual Transformer (CoT) module to optimize part of the CSPDarknet53 module of YOLOv5s in order to enhance the characteristics of the model while removing redundant operations extraction capacity.Experimental results conducted on two public datasets confirm that the proposed YOLOv5-CASP model outperforms the original YOLOv5s model and other five state-of-the-art models (Faster R-CNN, SSD, YOLOv4-Tiny, DETR-R50, Deformable DETR-R50), in terms of the mean average precision (mAP) and F1 score, by achieving corresponding values of 72% and 0.740 on the LUNA16 dataset, and 79.44% and 0.766 on the X-Nodule dataset.
科研通智能强力驱动
Strongly Powered by AbleSci AI