首先在当前的乳房 X 光照片上应用并评估 YOLO 技能,以检测不同的乳腺病变并将其分为肿块、钙化或布局扭曲,其余为正常。其次,考虑两种图像到图像技能 Pix2Pix 和 CycleGAN,以学习当前乳房 X 光照片与其对应的先前乳房 X 光照片之间的映射。如下图所示,生成新的合成先前乳房 X 光照片以降服由于时间和纹理厘革导致的筛查之间的错位。
接下来,利用第一步练习的模型来预测平移后的先前乳房 X 光照片上乳房病变的位置和范例。预测先前乳房 X 光照片中“未来癌症”可疑病变的边界框具有挑战性。因此,将全部诊断信息整合到一个框架中,该框架探索可能表明“未来癌症”风险的隐形模式证据。推理模型直接应用于翻译后的先前乳房 X 光照片,并利用 真实边界框的位置 及其 对应的当前乳房 X 光照片的类标签 进行评估。
4. 结果
4.1 数据集
在本研究中,我们利用了康涅狄格大学中心 (UCHC) 的一组私家数据集,名为 UCHC DigiMammo (UCHCDM) 数据库。该数据集包含 230 名患者的筛查乳房 X 线照片,此中每个病例都有一次初步筛查,称为先前检查,以及 1 至 6 年之间的第二次随访筛查,称为当前检查。
数据会合的每次筛查都会获取两个不同的视图,即 CC 和 MLO。全部图像均以医学数字成像和通信 (DICOM) 格式生存,由专业放射科医生在形貌文本文件中解释,并附上乳房 X 线检查发现的相应病理(即肿块、钙化、布局扭曲、正常)。总共考虑了 413 张乳房 X 线照片,分别用于当前检查和先前检查,它们的平均尺寸为 2950 × 3650 像素。
4.2 数据准备
全部乳房 X 光检查图像均利用数字 X 射线乳房 X 光检查工具网络,以 DICOM 格式压缩和存储图像。因此,在练习过程之前对全部原始图像应用了一些预处置惩罚步调,利用去噪和直方图均衡方法以提高质量。由于原始 DICOM 图像尺寸较大,全部乳房 X 光检查图像均利用 4×4 邻域的双三次插值进行下采样。在实验中,利用的图像大小为 448×448 像素(即根据 YOLO-V3 的 DarkNet 主干架构可以被 32 整除)。最后,全部练习图像都被归一化以使强度值在 [0, 1] 范围内。原始图像和预处置惩罚图像的样本如下图所示。
具体来说,输入图像的宽度和高度都应该是32的整数倍。这是因为 DarkNet 在进行特性提取和降采样时,会利用步长为32的卷积或池化操作,以确保特性图的大小与网络架构兼容。
由于很难网络和标志医学图像,医学数据集缺乏带解释的图像。为相识决这个问题,重要提出了数据增强技能,通过旋转或翻转实例来增长数据集的大小。本文中,将原始图像旋转了四次,角度为。因此,为 UCHCDM 数据集生成了总共 1,652 张乳房 X 光照片来练习和测试模型。每个种别的原始样本如下图所示。
4.3 评估指标和实验设置
在研究中,利用对象检测和分类指标来权衡基于 YOLO 的模型的性能。为了评估乳房病变在乳房 X 光检查中的位置及其范例的检测,首先丈量每个检测到的框与其对应的地面真相(即 (x, y, h, w) 坐标和类标签)之间的交并比 (IoU) 分数,然后验证它是否凌驾置信度分数阈值 0.35。
IoU 分数公式:
之后报告了一个最终的客观指标,称为检测准确率,考虑了真实检测框的预测种别概率。受 Samuelson 等人[49] 的工作启发,计算病变范例(即肿块、钙化、布局扭曲)和正常图像中真实检测到的图像数量与利用的乳房 X 线照片总数,如下面的公式中界说。
可以通过不同范例乳腺病变在形状、大小和质地方面的差异来表明。众所周知,钙化不会出现在标准的形状和位置,它们可以是双侧的、厚的、聚集的、多形性的和血管性的等等。由于它们的大小和位置不规则,利用图像到图像合成复制此类异常无助于在先前视图中检测和识别钙化病变。钙化通常很小且聚集在一起,必要平滑的像素分布,但是 X 射线图像可能会降级,并且在早期状态下很难识别乳房钙化。 别的,比力了 1 至 6 年之间厘革的 Prior 检查时间内的早期检测和分类结果。下图利用最佳报告实验(纵然用 Pix2Pix 转换)提供了对每个种别标签正确预测的 Prior 乳房 X 线照片百分比的视觉观察。 很明显,1 年的随访检查时间具有最高的预测图像率。强调了我们的方法在早期定位和识别通常被认为最难诊断的病变方面的乐成。另一个观察结果是,我们的方法捕获了 3 年以后才要求随访的肿块病变,这可能对诊断患有肿块性乳腺病变的患者来说为时已晚。
5. 讨论和结论
提出利用 YOLO 架构模型来检测和分类乳房 X 光检查中的可疑病变。 根据近来的工作,展示了利用基于 YOLO 的融合模型来正确定位和识别三种不同范例的病变的优势:肿块、钙化和布局扭曲。别的,进一步开发了所提出的框架,以整合全部利用的后续筛查中的先前乳房 X 光检查,并对初始筛查的乳房 X 光检查进行早期检测和分类。这项工作强调了对先前诊断为正常但在后期报告有明显异常发现和进展的乳房 X 光检查进行回顾性预测的能力。 类似的方法办理了这个问题,并利用乳房 X 光检查对来增强 CAD 体系对当前乳房 X 光检查的结果,方法是包罗地域配准的时间特性 [35],或在 SVM 分类器 [36] 或 CNN 模型 [37] 中添加配对之间的时间减法。然而,我们的研究只采用了一个模型,该模型在当前视图上进行练习和测试,然后在其对应的先前视图上进行推断。通过将生存的基于 YOLO 的融合模型以不同的方式直接应用于利用图像到图像转换技能生成的原始和合成先前乳房 X 光片上,强调了提出的方法的性能。在乳房 X 光片对(先前、当前)之间练习和验证了两种开始辈的模型 CycleGAN 和 Pix2Pix,以创建新的转换先前乳房 X 光片,从而降服由于时间和纹理厘革导致两次筛查之间的错位。