计算机辅助诊断(CAD)有近五十年的历史,已经帮助许多临床医生诊断。随着技术的发展,最近,研究使用深度学习方法来获得高精度的结果在CAD系统。与CAD计算机输出可以作为第二选择对放射科医生和医生做最后的正确决策。胸部异常检测是一个典型的检测和分类问题;研究者需要分类常见的胸肺疾病和本地化关键的发现。检测问题,深刻的学习方法有两种:单程法和两阶段法。在我们的论文中,我们介绍和分析一些代表性的模型,如RCNN, SSD, YOLO系列的意思。为了更好地解决胸部异常检测的问题,我们提出了一种新的基于YOLOv5和ResNet50模型。YOLOv5是最新YOLO系列,意思是比单程检测算法更灵活。YOLOv5的功能在我们的纸是本地化异常区域。 On the other hand, we use ResNet, avoiding gradient explosion problems in deep learning for classification. And we filter the result we got from YOLOv5 and ResNet. If ResNet recognizes that the image is not abnormal, the YOLOv5 detection result is discarded. The dataset is collected via VinBigData’s web-based platform, VinLab. We train our model on the dataset using Pytorch frame and use the mAP, precision, and
由于技术的发展,与传统诊断方法,放射科医生可能使用成像技术诊断和治疗的医疗条件如CT和PET扫描,核磁共振,x射线,当然,当病人去医院(gydF4y2Ba
为此,由于成像技术的迅速发展和计算机的计算能力,诞生了一个新的研究维度,称为计算机辅助诊断(CAD)系统(gydF4y2Ba
CAD系统可以分为两个重要的方面:“检测”和“诊断”[gydF4y2Ba
机器学习方法在CAD系统分析成像数据和开发模型匹配的输入数据和输出疾病之间的关系从病人的人口使用成像数据gydF4y2Ba
机器学习技术的方法来分析病人数据获得决策支持,适用于任何病人护理过程,如疾病或损伤检测、表征,癌症分期、治疗计划、治疗反应评估、监控、复发和预后预测(gydF4y2Ba
也同样,在本文中,我们开发一个应用程序的疾病检测的应用深度学习在CT图像上。我们的任务是本地化和14个类型进行分类的胸异常胸片。我们的贡献是给自动胸部检测解决方案。特别是,我们将检测方法分为两个步骤包括检测步骤和一个分类的步骤。分类步骤是用于过滤结果的第一步。gydF4y2Ba
我们描述了近年来临床诊断与计算机和计算机的历史版本gydF4y2Ba
在本部分中,我们首先介绍胸部x线摄影的定义,然后大致解释CAD的发展。最后,我们描述了一些算法的目标检测用于我们对胸部异常检测的任务。gydF4y2Ba
胸部x线摄影是最常用的诊断成像过程gydF4y2Ba
CAD系统已经发展了许多年的传统的机器学习方法,现在,深度学习。当务之急是一个不可阻挡的趋势在临床过程中使用CAD系统。文献[gydF4y2Ba
最近,越来越多的应用程序对目标检测(gydF4y2Ba
两阶段方法:例如,表示是RCNN算法(gydF4y2Ba
单程方法:表示YOLO和SSD,意思是意识到一个端到端的模型直接得到最终的检测结果(gydF4y2Ba
候选人地区一代:使用选择性搜索方法来生成1 K∼2 K的候选区域从图像第二步gydF4y2Ba
特征提取:对于每个候选人地区提供的第一步,是深卷积网络用于提取特征gydF4y2Ba
类别的判断:使用SVM分类器,输入第二步到分类器中提供的功能gydF4y2Ba
位置优化:使用精细的回归正确的候选帧的位置gydF4y2Ba
然而,RCNN有两个缺点gydF4y2Ba
快RCNN仍然使用选择性搜索选择2000个候选框(gydF4y2Ba
没有支持向量机分类器和回归量在快速RCNN [gydF4y2Ba
使用网络而不是使用的选择性搜索项快速RCNN提取候选区域相当于快RCNN =的RPN +快速RCNN,以及RPN和快速RCNN分享卷积层(gydF4y2Ba
快RCNN具有以下特征(gydF4y2Ba
多尺度目标:使用RPN网络候选区域和使用锚不同大小和纵横比的解决多尺度问题gydF4y2Ba
计算的交点的借据锚和真正的框架和建立积极的和消极的样本通过阈值gydF4y2Ba
样本不平衡:在每批随机样本256名主播边境回归训练,确保正负样本的数量是相同的,尽可能避免问题的梯度规则太多造成的负样本gydF4y2Ba
速度快:与两步的目标检测方法相比,YOLOv1使用端到端方法,更快gydF4y2Ba
使用全局特征的推理。由于使用全局上下文信息,与滑动窗口和意见箱方法相比,背景是更准确的判断gydF4y2Ba
泛化:训练模型仍然有好的结果在新领域或意想不到的输入情况gydF4y2Ba
使用多尺度特征图进行检测。SSD利用大规模的特征图谱检测小目标,反之亦然。gydF4y2Ba
利用卷积的检测。SSD直接使用卷积来提取不同特征图谱检测结果。gydF4y2Ba
设置一个先天的盒子。SSD利用更快的锚RCNN和设置一个先天的框与不同尺度或纵横比为每个单元。预测的边框都是基于这些盒子之前,这样可以减少培训在一定程度上的困难。一般来说,每个单元将多个先天的盒子,和他们的尺度和比例方面是不同的。gydF4y2Ba
SSD使用VGG16作为基本模型,然后添加一个新的卷积层VGG16的基础上获得更多的特征图谱检测(gydF4y2Ba
有五个主要优势的SSD (gydF4y2Ba
实时:它比YOlOv1更快,因为完全连接层gydF4y2Ba
标签方案:通过预测类别信心和前一帧的偏差的相对固定的尺度,不同尺度的影响损失可以有效地平衡gydF4y2Ba
多尺度:多尺度目标预测是由使用多个特征图和锚定帧对应不同的尺度gydF4y2Ba
数据增强:增强执行的数据随机裁剪来提高模型的鲁棒性gydF4y2Ba
样本不平衡:通过困难样本采矿、最高的先验盒负样本用于训练,信心和积极的和消极的比例样本设置为1:3,这使得该模型训练收敛速度更快gydF4y2Ba
批规范化:它大大提高了性能gydF4y2Ba
高分辨率分类器:它使pretraining分类任务分辨率与目标探测分辨率一致gydF4y2Ba
卷积与锚箱:使用一个完全卷积神经网络预测偏差,而不是特定的坐标,模型更容易收敛gydF4y2Ba
维度集群:设置锚架的规模通过聚类算法获得一个更好的先天的框架和减轻不同尺度对损失的影响gydF4y2Ba
细粒度功能:集成底层图像特征通过简单的加法gydF4y2Ba
多尺度培训:通过使用一个完整的卷积网络,多个尺度的模型支持输入图像和火车gydF4y2Ba
构造Darknet-19而不是VGG16骨干与更好的性能gydF4y2Ba
实时:与RetinaNet相比,YOLOv3牺牲检测准确性和使用Darknet特征提取主干网络代替ResNet101获得更快的检测速度gydF4y2Ba
多尺度:与YOLOv1-v2相比,相同的红外系统网络RetinaNet用作增强特征提取网络来获得更高的检测精度gydF4y2Ba
目标重叠:通过使用逻辑回归和分类预测损失函数两叉,每个候选帧与多个标签分类解决的可能性,一个检测帧可能包含多个目标同时gydF4y2Ba
实时:借鉴CSPNet网络结构,提高Darknet53 CSPDarknet53使模型参数,计算时间短gydF4y2Ba
多尺度:颈部分别介绍了锅和小型网络结构作为一个增强的特征提取网络,它可以有效的多尺度特性和精度高于红外系统的引入网络gydF4y2Ba
数据增强:马赛克的引入数据增强可以有效降低batch_size当使用BN的影响gydF4y2Ba
模型训练:借据,GIoU DIoU,意识作为回归的目标框架,具有检测精度高于平方差异损失由YOLOv3使用gydF4y2Ba
在本节中,我们介绍我们关于胸部异常检测方法。我们使用两步方法。第一步是使用一些传统的目标检测方法如YOLOv5执行目标探测。第二步是使用图像分类器来执行两个分类(是否有异常),如果是认识到,图像不是异常,YOLOv5的检测结果就会被丢弃。gydF4y2Ba
整个结构YOLOv5 [gydF4y2Ba
YOLOv5架构。gydF4y2Ba
YOLO家族的意思模型包含三个主要的建筑块:骨干,颈部和头部。gydF4y2Ba
YOLOv5支柱:它采用CSPDarknet为骨干组成的特征提取的图像cross-stage局部网络gydF4y2Ba
YOLOv5脖子:它使用PANet生成特性金字塔网络上执行聚合特性并将其传递给预测gydF4y2Ba
YOLOv5头:层,生成预测锚箱的对象检测gydF4y2Ba
除此之外,YOLOv5使用以下选择培训(gydF4y2Ba
激活和优化:YOLOv5使用漏ReLU乙状结肠激活和SGD亚当作为优化器选择gydF4y2Ba
损失函数:它使用二叉叉分对数的损失gydF4y2Ba
YOLOv5有多个品种的pretrained模型如上面我们可以看到。他们之间的差异之间的权衡模型的大小和推理时间。轻量级模型版本YOLOv5s只是14 MB但不是很准确。另一方面,我们有YOLOv5x的大小是168 MB,但最准确的版本的家庭(gydF4y2Ba
而YOLO系列,意思YOLOv5薮猫照明点(gydF4y2Ba
多尺度:使用红外系统提高特征提取的网络而不是锅,使模型更简单和更快gydF4y2Ba
目标重叠:使用舍入方法找到附近的位置,所以目标映射到多个中央网格点gydF4y2Ba
ResNet [gydF4y2Ba
剩余模块。gydF4y2Ba
剩余模块由两个致密层和一个跳过连接。每两个致密层的激活函数是ReLU函数。gydF4y2Ba
为了解决胸部异常检测,我们设计一个新的混合模型,结合YOLOv5和ResNet50。处理原始图像后,我们在YOLOv5和ResNet50输入它们。然后我们输入的过滤器。过滤器的作用主要是去除YOLOv5无法发现的异常由ResNet分类。我们的模型的整体结构如图gydF4y2Ba
我们的模型的整体结构。gydF4y2Ba
在本节中,我们介绍我们使用的数据集和重要性能指标的研究。gydF4y2Ba
我们的数据集从VinBigData获得,这是一个组织促进基础研究和研究小说和高度适用技术。VinBigData医学成像小组进行研究收集、处理、分析和理解医疗数据。他们致力于构建大规模、高精度医疗成像解决方案基于人工智能的最新发展,促进有效的临床工作流。gydF4y2Ba
构建VinDr-CXR数据集是三个步骤的过程gydF4y2Ba
数据收集:当病人接受胸部影像学检查,医疗机构可以收集原始图像DICOM格式图像,然后让鉴定保护病人的隐私。gydF4y2Ba
数据过滤:因为并不是所有的图片都是有效的,有必要对原始图像进行过滤。例如,其他形式的图片,其他身体部位,低质量,或不正确的方向都需要过滤掉的基于机器学习的分类器。gydF4y2Ba
数据标签:开发一个基于web的标记工具,VinLab,存储、远程管理和注释DICOM数据。gydF4y2Ba
在我们的任务,我们需要做的是分类,数据集包含14个重要影像学结果。(0)主动脉扩张,(1)肺不张,(2)钙化,(3)心脏肥大,(4)合并,(5)ILD,(6)渗透,(7)肺不透明度,(8)结节/质量,其他病变(9),(10)胸腔积液,胸膜增厚(11),(12)气胸,肺纤维化(13)。和“没有找到”观察(14)是捕获其余发现除了上述14发现。gydF4y2Ba
18000扫描数据集包含注释的有经验的放射科医生。gydF4y2Ba
我们使用15000扫描作为训练数据集,和其他3000个扫描的测试数据集。此外,我们的火车。csv是火车组元数据,为每一个对象一行,包括一个类和一个边界框。它包含8列,它们独特的形象标识符,检测到的对象的类的名称,类的检测对象的ID, ID的放射科医生的观察,和对象的边界框的最小坐标,分别。一些图片在测试和训练有多个对象。数据gydF4y2Ba
输入数据的一个例子。gydF4y2Ba
输出数据的一个例子。gydF4y2Ba
在本节中,我们描述了一些评价指标用于我们的实验。众所周知,我们,在CAD系统中,主要是检测。共同的标准来衡量分类算法的性能包括精度、灵敏度(回忆),特异性,精度、f值,ROC曲线,记录损失,借据(gydF4y2Ba
据统计机器学习理论,精密两分的统计指标的公式gydF4y2Ba
此外,有必要定义TP, FP和FN的检测任务。gydF4y2Ba
真阳性(TP):借据>(公式)(在本文中,(公式)0.6)检测帧的数量(此句地面真理只计算一次)gydF4y2Ba
假阳性(FP):复选框的数量借据< =(公式),或冗余的复选框的数量检测同一地面实况gydF4y2Ba
假阴性(FN):真理不检测的数量gydF4y2Ba
借据是衡量两个检测帧之间的重叠度(目标探测),公式如下:gydF4y2Ba
B_gt代表实际地面坐标系(地面实况,GT)的目标,和B_p代表预测框架。通过计算两者的借据,它可以判断预测检测框架是否符合条件。照片显示了借据如下。gydF4y2Ba
了解这些知识后,我们引入了地图。美联社是计算某种类型的P-R曲线下的面积,和地图是计算平均P-R曲线下的面积的类型。gydF4y2Ba
我们YOLOv5模型的重要参数如表所示gydF4y2Ba
参数及其值。gydF4y2Ba
| 参数gydF4y2Ba | 价值gydF4y2Ba |
|---|---|
| 批量大小gydF4y2Ba | 8gydF4y2Ba |
| 图像大小gydF4y2Ba | 512年gydF4y2Ba |
| 时代gydF4y2Ba | 20.gydF4y2Ba |
| 学习速率gydF4y2Ba | 1gydF4y2Ba |
我们画一个类分布的直方图来表示数据集显然在图gydF4y2Ba
类分布的直方图。gydF4y2Ba
和图gydF4y2Ba
此外,评估模型的性能,我们选择一些以前的经典模型和比较它们使用相同的数据集和评价指标。我们在地图的指标进行比较,精度。我们已经介绍了上述定义。我们选择的经典模型YOLOv5,快速RCNN、高效。表gydF4y2Ba
与其他模型的结果比较。gydF4y2Ba
| 模型gydF4y2Ba | Map@0.6gydF4y2Ba | 精度gydF4y2Ba |
|---|---|---|
| YOLOv5 + ResNet50gydF4y2Ba | 0.254gydF4y2Ba | 0.512gydF4y2Ba |
| YOLOv5gydF4y2Ba | 0.244gydF4y2Ba | 0.494gydF4y2Ba |
| 快RCNNgydF4y2Ba | 0.234gydF4y2Ba | 0.485gydF4y2Ba |
| EfficientDetgydF4y2Ba | 0.231gydF4y2Ba | 0.479gydF4y2Ba |
我们工作的动机是开发一个系统来自动检测胸部异常使用深度学习技术。我们的工作可以帮助医生提高诊断和更快的做出决定。介绍,计算机辅助诊断(CAD)的背景和一些相关的作品。在引言部分的结束,我们的方法的建议。检测方法包含两个步骤。第一步是使用目标检测算法如YOLO和EfficientDet找到位置意思(边界框)的CT扫描图像。很可能的结果是一组有信心大于前一个分数。第二步是使用二进制CNN分类器像ResNet删除的图像异常产生的第一步。在第一步中,我们提到古典像RCNN检测神经网络,快速RCNN RCNN更快,SSD, YOLO系列的意思。这些模型的结构和一些特点是仔细描述。 In the experiment section, the VinBig Dataset is firstly introduced. The training parameters of models, evaluation metrics, and the figures of training process are also given for a repeating experiment. Table
所有数据用于支持本研究的结果包括在本文中。gydF4y2Ba
于罗和一帆张co-first作者。gydF4y2Ba
作者宣称没有利益冲突。gydF4y2Ba