论文部分内容阅读
以黑色素瘤为代表的皮肤癌是一种严重的皮肤疾病,每年不仅患病人数多,而且死亡率高,严重威胁人类的健康。借助皮肤镜图像技术,对皮肤癌进行分割,是为临床医生提供辅助诊断和治疗的重要方式。在当前提出的皮肤癌分割方法中,基于传统图像处理的皮肤癌分割方法效果不够理想,而基于深度学习的皮肤癌分割方法被广泛采用。
针对U-net模型中连续的下采样次数过多,导致精度损失,以及Deeplabv3模型中输出的特征图分辨率太小,导致预测结果的边缘有明显锯齿状的问题,提出了U-DeepLab模型。该模型的整体结构仍然采用编码器-解码器结构,并加入了空洞金字塔模块(ASPP),通过将几个不同空洞率的空洞卷积并联,可以在不过多降低特征图分辨率的前提下,提取图像中的多尺度信息。模型输出的预测结果的尺寸和原输入图像的尺寸相同,这样就保证了分割边缘的精细度和平滑度,同时还提升了在皮肤癌分割上的效果。
在语义分割任务的众多性能评估目标中,Jaccard指数(IOU)是其中最常用的指标。针对交叉熵损失没有直接优化Jaccard指数,以及IOU-loss只适用于离散情况的计算等问题,提出使用Lovasz-hinge损失函数训练皮肤癌的分割模型。Lovasz-hinge损失函数是针对Jaccard指数专门设计的,它在Jaccardloss的基础上进行Lovasz拓展,使其连续可导,且输入空间从离散域变成连续域,从而起到优化模型参数的目的。使用Lovasz-hinge损失函数训练优化U-DeepLab模型,将皮肤癌病变区域的分割效果进一步提升。
使用ISIC2018数据集对模型的分割效果进行评估。在训练过程中,使用最大的批量大小,采用数据增强,预热+余弦退火的学习率调整策略和迁移学习;在模型预测阶段,使用测试时数据增强(TTA)。实验结果显示,U-DeepLab模型在ISIC2018数据集上的JaccardIndex达到了0.826,使用Lovasz-hinge损失函数训练后,U-DeepLab模型的JaccardIndex进一步提升,达到了0.854,超过了皮肤科专家的分割水平。
针对U-net模型中连续的下采样次数过多,导致精度损失,以及Deeplabv3模型中输出的特征图分辨率太小,导致预测结果的边缘有明显锯齿状的问题,提出了U-DeepLab模型。该模型的整体结构仍然采用编码器-解码器结构,并加入了空洞金字塔模块(ASPP),通过将几个不同空洞率的空洞卷积并联,可以在不过多降低特征图分辨率的前提下,提取图像中的多尺度信息。模型输出的预测结果的尺寸和原输入图像的尺寸相同,这样就保证了分割边缘的精细度和平滑度,同时还提升了在皮肤癌分割上的效果。
在语义分割任务的众多性能评估目标中,Jaccard指数(IOU)是其中最常用的指标。针对交叉熵损失没有直接优化Jaccard指数,以及IOU-loss只适用于离散情况的计算等问题,提出使用Lovasz-hinge损失函数训练皮肤癌的分割模型。Lovasz-hinge损失函数是针对Jaccard指数专门设计的,它在Jaccardloss的基础上进行Lovasz拓展,使其连续可导,且输入空间从离散域变成连续域,从而起到优化模型参数的目的。使用Lovasz-hinge损失函数训练优化U-DeepLab模型,将皮肤癌病变区域的分割效果进一步提升。
使用ISIC2018数据集对模型的分割效果进行评估。在训练过程中,使用最大的批量大小,采用数据增强,预热+余弦退火的学习率调整策略和迁移学习;在模型预测阶段,使用测试时数据增强(TTA)。实验结果显示,U-DeepLab模型在ISIC2018数据集上的JaccardIndex达到了0.826,使用Lovasz-hinge损失函数训练后,U-DeepLab模型的JaccardIndex进一步提升,达到了0.854,超过了皮肤科专家的分割水平。