赋能医学影像,AI助力智能化辅助诊断,基于YOLOv3全系列【yolov3tiny/yolov3/yolov3spp】参数模型开发构建磁共振成像、CT扫描场景下医学图像脑肿瘤检测识别系统

Together_CZ 2024-08-15 13:31:01 阅读 100

在医学的浩瀚星空中,肿瘤与癌症等复杂疾病如同夜空中最难以捉摸的星辰,它们的发现与诊断往往依赖于高精度、高灵敏度的医学影像技术,如核磁共振(MRI)和计算机断层扫描(CT)等。这些技术如同现代医学的“火眼金睛”,能够穿透人体,揭示出隐藏在内部的疾病秘密。然而,传统的医学影像分析却面临着诸多挑战:高度依赖专家的主观判断、经验门槛高、分析过程耗时耗力,以及时效性不足等问题,这些都成为了制约医疗服务质量与效率提升的瓶颈。

随着人工智能(AI)技术的飞速发展,一场深刻的医疗变革正在悄然发生。AI以其强大的数据处理能力、学习能力和模式识别能力,为医学影像分析带来了前所未有的机遇。在医学影像领域,AI技术的应用不仅有望解决上述难题,更将推动疾病诊断向更加精准、高效、智能的方向迈进。医学影像数据是医疗大数据的重要组成部分,其海量、多维、复杂的特点为AI模型的训练提供了丰富的素材。通过深度学习等先进算法,AI可以从海量的影像数据中学习并提取出疾病特有的特征模式,进而构建出智能化的检测识别模型。这些模型能够自动分析影像,快速准确地识别出异常区域,为医生提供初步的诊断建议,极大地提高了诊断的效率和准确性。AI在医学影像分析中的应用,使得医生能够借助智能系统快速筛选出疑似病例,减轻了他们的工作负担,同时也降低了人为因素导致的误诊和漏诊风险。医生可以将更多精力集中在复杂病例的深入分析和治疗方案的制定上,从而提高了整体医疗服务的质量。此外,AI系统的实时性和远程服务能力,使得患者能够更快地获得诊断结果,尤其是对于偏远地区的患者而言,这无疑是巨大的福音。

本文从这样的背景出发,想要基于基础的实验尝试来开发构建医学影像场景下的智能化检测识别模型,首先看下实例效果:

接下来看下数据实例:

本文是选择的比较经典的也是比较古老的YOLOv3来进行模型的开发,YOLOv3(You Only Look Once v3)是一种目标检测算法模型,它是YOLO系列算法的第三个版本。该算法通过将目标检测任务转化为单个神经网络的回归问题,实现了实时目标检测的能力。

YOLOv3的主要优点如下:

实时性能:YOLOv3采用了一种单阶段的检测方法,将目标检测任务转化为一个端到端的回归问题,因此具有较快的检测速度。相比于传统的两阶段方法(如Faster R-CNN),YOLOv3能够在保持较高准确率的情况下实现实时检测。

多尺度特征融合:YOLOv3引入了多尺度特征融合的机制,通过在不同层级的特征图上进行检测,能够有效地检测不同尺度的目标。这使得YOLOv3在处理尺度变化较大的场景时表现出较好的性能。

全局上下文信息:YOLOv3在网络结构中引入了全局上下文信息,通过使用较大感受野的卷积核,能够更好地理解整张图像的语义信息,提高了模型对目标的识别能力。

简洁的网络结构:YOLOv3的网络结构相对简洁,只有75个卷积层和5个池化层,使得模型较易于训练和部署,并且具有较小的模型体积。

YOLOv3也存在一些缺点:

较低的小目标检测能力:由于YOLOv3采用了较大的感受野和下采样操作,对于小目标的检测能力相对较弱。当场景中存在大量小目标时,YOLOv3可能会出现漏检或误检的情况。

较高的定位误差:由于YOLOv3将目标检测任务转化为回归问题,较粗糙的特征图和较大的感受野可能导致较高的定位误差。这意味着YOLOv3在需要较高精度的目标定位时可能会受到一定的限制。

YOLOv3是YOLO系列里程碑性质的模型,随着不断地演变和发展,目前虽然已经在性能上难以与YOLOv5之类的模型对比但是不可否认其做出的突出贡献。

训练数据配置文件如下:

# path

train: ./dataset/images/train/

val: ./dataset/images/test/

# number of classes

nc: 2

# class names

names: ['negative', 'positive']

我们开发构建了yolov3全系列的参数模型,包含:yolov3-tiny、yolov3和yolov3-spp,实验阶段保持完全相同的参数设置,等待整体实验完成我们来对其进行整体的性能评估测试。

【Precision曲线】

精确率曲线(Precision Curve)是一种用于评估二分类模型在不同阈值下的精确率性能的可视化工具。它通过绘制不同阈值下的精确率和召回率之间的关系图来帮助我们了解模型在不同阈值下的表现。

精确率(Precision)是指被正确预测为正例的样本数占所有预测为正例的样本数的比例。召回率(Recall)是指被正确预测为正例的样本数占所有实际为正例的样本数的比例。

绘制精确率曲线的步骤如下:

使用不同的阈值将预测概率转换为二进制类别标签。通常,当预测概率大于阈值时,样本被分类为正例,否则分类为负例。

对于每个阈值,计算相应的精确率和召回率。

将每个阈值下的精确率和召回率绘制在同一个图表上,形成精确率曲线。

根据精确率曲线的形状和变化趋势,可以选择适当的阈值以达到所需的性能要求。

通过观察精确率曲线,我们可以根据需求确定最佳的阈值,以平衡精确率和召回率。较高的精确率意味着较少的误报,而较高的召回率则表示较少的漏报。根据具体的业务需求和成本权衡,可以在曲线上选择合适的操作点或阈值。

精确率曲线通常与召回率曲线(Recall Curve)一起使用,以提供更全面的分类器性能分析,并帮助评估和比较不同模型的性能。

【Recall曲线】

召回率曲线(Recall Curve)是一种用于评估二分类模型在不同阈值下的召回率性能的可视化工具。它通过绘制不同阈值下的召回率和对应的精确率之间的关系图来帮助我们了解模型在不同阈值下的表现。

召回率(Recall)是指被正确预测为正例的样本数占所有实际为正例的样本数的比例。召回率也被称为灵敏度(Sensitivity)或真正例率(True Positive Rate)。

绘制召回率曲线的步骤如下:

使用不同的阈值将预测概率转换为二进制类别标签。通常,当预测概率大于阈值时,样本被分类为正例,否则分类为负例。

对于每个阈值,计算相应的召回率和对应的精确率。

将每个阈值下的召回率和精确率绘制在同一个图表上,形成召回率曲线。

根据召回率曲线的形状和变化趋势,可以选择适当的阈值以达到所需的性能要求。

通过观察召回率曲线,我们可以根据需求确定最佳的阈值,以平衡召回率和精确率。较高的召回率表示较少的漏报,而较高的精确率意味着较少的误报。根据具体的业务需求和成本权衡,可以在曲线上选择合适的操作点或阈值。

召回率曲线通常与精确率曲线(Precision Curve)一起使用,以提供更全面的分类器性能分析,并帮助评估和比较不同模型的性能。

【loss曲线】

在深度学习的训练过程中,loss函数用于衡量模型预测结果与实际标签之间的差异。loss曲线则是通过记录每个epoch(或者迭代步数)的loss值,并将其以图形化的方式展现出来,以便我们更好地理解和分析模型的训练过程。

【mAP0.5】

mAP0.5,也被称为mAP@0.5或AP50,指的是当Intersection over Union(IoU)阈值为0.5时的平均精度(mean Average Precision)。IoU是一个用于衡量预测边界框与真实边界框之间重叠程度的指标,其值范围在0到1之间。当IoU值为0.5时,意味着预测框与真实框至少有50%的重叠部分。

在计算mAP0.5时,首先会为每个类别计算所有图片的AP(Average Precision),然后将所有类别的AP值求平均,得到mAP0.5。AP是Precision-Recall Curve曲线下面的面积,这个面积越大,说明AP的值越大,类别的检测精度就越高。

mAP0.5主要关注模型在IoU阈值为0.5时的性能,当mAP0.5的值很高时,说明算法能够准确检测到物体的位置,并且将其与真实标注框的IoU值超过了阈值0.5。

【mAP0.5:0.95】

mAP0.5:0.95,也被称为mAP@[0.5:0.95]或AP@[0.5:0.95],表示在IoU阈值从0.5到0.95变化时,取各个阈值对应的mAP的平均值。具体来说,它会在IoU阈值从0.5开始,以0.05为步长,逐步增加到0.95,并在每个阈值下计算mAP,然后将这些mAP值求平均。

这个指标考虑了多个IoU阈值下的平均精度,从而更全面、更准确地评估模型性能。当mAP0.5:0.95的值很高时,说明算法在不同阈值下的检测结果均非常准确,覆盖面广,可以适应不同的场景和应用需求。

对于一些需求比较高的场合,比如安全监控等领域,需要保证高的准确率和召回率,这时mAP0.5:0.95可能更适合作为模型的评价标准。

综上所述,mAP0.5和mAP0.5:0.95都是用于评估目标检测模型性能的重要指标,但它们的关注点有所不同。mAP0.5主要关注模型在IoU阈值为0.5时的性能,而mAP0.5:0.95则考虑了多个IoU阈值下的平均精度,从而更全面、更准确地评估模型性能。

【F1值曲线】

F1值曲线是一种用于评估二分类模型在不同阈值下的性能的可视化工具。它通过绘制不同阈值下的精确率(Precision)、召回率(Recall)和F1分数的关系图来帮助我们理解模型的整体性能。

F1分数是精确率和召回率的调和平均值,它综合考虑了两者的性能指标。F1值曲线可以帮助我们确定在不同精确率和召回率之间找到一个平衡点,以选择最佳的阈值。

绘制F1值曲线的步骤如下:

使用不同的阈值将预测概率转换为二进制类别标签。通常,当预测概率大于阈值时,样本被分类为正例,否则分类为负例。

对于每个阈值,计算相应的精确率、召回率和F1分数。

将每个阈值下的精确率、召回率和F1分数绘制在同一个图表上,形成F1值曲线。

根据F1值曲线的形状和变化趋势,可以选择适当的阈值以达到所需的性能要求。

F1值曲线通常与接收者操作特征曲线(ROC曲线)一起使用,以帮助评估和比较不同模型的性能。它们提供了更全面的分类器性能分析,可以根据具体应用场景来选择合适的模型和阈值设置。

从整体实验结果上来看:三款不同参数系列的模型没有拉开明显的差距,综合考虑模型参数量这里最终选择使用yolov3-tiny来作为线上推理模型。接下来详细看下yolov3-tiny模型的结果详情。

【离线推理实例如下】

【Batch实例】

【混淆矩阵】

【PR曲线】

【训练可视化】

时代的浪潮在滚滚向前,拥抱科技时代带来的领域变革。展望未来,随着AI技术的不断成熟和医学影像数据的持续积累,AI在医学影像分析中的应用将更加广泛和深入。我们可以预见,未来的医学影像分析将实现全自动化、智能化和个性化,为医生提供更加全面、精准的诊断支持。同时,AI还将与其他医疗技术如基因测序、机器人手术等相结合,共同推动医疗行业的全面升级和转型。AI技术正以前所未有的速度和力量改变着医学影像分析的面貌,为疾病的早期发现、精准诊断和治疗提供了强有力的支持。在这场医疗变革的浪潮中,我们有理由相信,AI将成为推动医学进步的重要力量,引领我们迈向一个更加健康、美好的未来。



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。