一种改进的更快的 R-CNN 算法,用于辅助肺结节检测。

An improved faster R-CNN algorithm for assisted detection of lung nodules.

机构信息

School of Statistics and Mathematics, Zhejiang Gongshang University, Hangzhou, 310018, China; Collaborative Innovation Center of Statistical Data Engineering, Technology & Application, Zhejiang Gongshang University, Hangzhou, 310018, China.

School of Statistics and Mathematics, Zhejiang Gongshang University, Hangzhou, 310018, China; Collaborative Innovation Center of Statistical Data Engineering, Technology & Application, Zhejiang Gongshang University, Hangzhou, 310018, China.

出版信息

Comput Biol Med. 2023 Feb;153:106470. doi: 10.1016/j.compbiomed.2022.106470. Epub 2022 Dec 28.

Abstract

The morbidity and mortality of lung cancer are increasing rapidly in every country in the world, and pulmonary nodules are the main symptoms of lung cancer in the early stage. If we can diagnose pulmonary nodules in time at the early stage and follow up and treat suspicious patients, we can effectively reduce the incidence of lung cancer. CT (Computed Tomography) has been applied to the screening of many diseases because of its high resolution. Pulmonary nodules show white round shadows in CT images. With the popularity of CT equipment, doctors need to review a large number of imaging results every day. Doctors will misjudge and miss the lesions because of reviewing CT scanning results for a long time. At this time, the method of automatic detection of pulmonary nodules by computer can relieve the pressure of doctors in reviewing CT scan results. Traditional lung nodule detection methods, such as gray threshold method and region growing method, divide the detection process into two steps: extracting candidate regions and eliminating false regions. In addition, the traditional detection method can only operate on a single image, which leads to the inability of this method to detect the batch scanning results in real time. With the continuous development of computer equipment performance and artificial intelligence, the relationship between medical image processing and deep learning is getting closer and closer. In deep learning, object detection methods such as Faster R-CNN、YOLO can complete parallel detection of batch images, and deep structure can fully extract the features of input images. Compared with traditional lung nodule detection methods, it has the characteristics of high efficiency and high precision. Faster R-CNN is a classical and high-precision two-stage object detection method. In this paper, an improved Faster R-CNN model is proposed. On the basis of Faster R-CNN, multi-scale training strategy is used to fully mine the features of different scale spaces and perform path augmentation on lower-dimensional features, which improves the small object detection ability of the model. Through Online Hard Example Mining (OHEM), the loss value is used to quantify the difficulty of candidate region detection, and the training times of the region to be detected are adaptively adjusted. Make full use of prior information to customize the size and proportion of preset boundary anchor boxes. Using deformable convolution to improve the visual field to enhance the global features and enhance the ability to extract the feature information of pulmonary nodules in the same scale space. The new model was tested on LUNA16 (Lung Nodule Analysis 2016) dataset. The detection precision of the improved Faster R-CNN model for pulmonary nodules increased from 76.4% to 90.7%, and the recall rate increased from 40.1% to 56.8% Compared with the mainstream object detection algorithms YOLOv3 and Cascade R-CNN, the improved model is superior to the above models in every index.

摘要

肺癌的发病率和死亡率在世界上的每个国家都在迅速上升,肺部结节是肺癌早期的主要症状。如果我们能在早期及时诊断肺部结节,并对可疑患者进行随访和治疗,就能有效地降低肺癌的发病率。CT(计算机断层扫描)由于其高分辨率已被应用于许多疾病的筛查。肺部结节在 CT 图像中呈现为白色圆形阴影。随着 CT 设备的普及,医生每天需要审查大量的成像结果。由于长时间审查 CT 扫描结果,医生可能会误判和遗漏病变。此时,计算机自动检测肺结节的方法可以减轻医生审查 CT 扫描结果的压力。传统的肺结节检测方法,如灰度阈值法和区域生长法,将检测过程分为两步:提取候选区域和消除伪区域。此外,传统的检测方法只能对单个图像进行操作,这导致该方法无法实时检测批量扫描结果。随着计算机设备性能和人工智能的不断发展,医学图像处理与深度学习的关系越来越密切。在深度学习中,Faster R-CNN、YOLO 等目标检测方法可以并行检测批量图像,深度学习结构可以充分提取输入图像的特征。与传统的肺结节检测方法相比,它具有高效、高精度的特点。Faster R-CNN 是一种经典的高精度两阶段目标检测方法。本文提出了一种改进的 Faster R-CNN 模型。在 Faster R-CNN 的基础上,采用多尺度训练策略充分挖掘不同尺度空间的特征,并对低维特征进行路径增强,提高了模型对小目标的检测能力。通过在线硬例挖掘(OHEM),使用损失值量化候选区域检测的难度,并自适应调整待检测区域的训练次数。充分利用先验信息定制预设边界锚框的大小和比例。使用可变形卷积提高视野,增强全局特征,增强同一尺度空间内肺结节特征信息的提取能力。新模型在 LUNA16(肺结节分析 2016)数据集上进行了测试。改进后的 Faster R-CNN 模型对肺结节的检测精度从 76.4%提高到 90.7%,召回率从 40.1%提高到 56.8%。与主流目标检测算法 YOLOv3 和级联 R-CNN 相比,改进后的模型在各项指标上均优于上述模型。

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索