第 5 章 总结与展望
5.1 总结
医学图像分割对三维定位、三维可视化、手术规划和计算机辅助诊断等都具有十分重要的意义,是图像处理与分析的热点研究领域之一。随着深度学习技术的快速发展,基于深度学习的医学图像分割方法较于传统方法在特征提取、分割效率和准确度上具有明显优势。本文重点研究的全卷积神经网络是深度学习中普遍应用于医学图像分割的模型,具有重大的研究意义。 本文首先从医学图像分割的研究背景出发,说明了本课题的研究意义,简要分析了国内外研究现状,包括传统医学图像分割方法和基于深度学习的医学图像分割方法。其中重点对基于全卷积神经网络的医学图像分割算法及其改进算法的研究现状进行了介绍。然后为了确保论文的严谨性和方便读者理解后续研究工作,介绍了相关预备知识,包括数据增强方法、卷积神经网络基本结构和基于深度学习的医学图像分割方法的提出、基本原理以及存在的优缺点。最后针对现基于全卷积神经网络的图像分割方法的问题,从加深网络和提高卷积核特征提取能力两方面提出了相应的改进方法,并通过实验结果和数据验证了算法的有效性和先进性。本文的主要工作内容有:
1. 提出了一种基于全卷积 DenseNet 的医学图像分割方法。该方法受 U-Net 和DenseNet 的启发,将 DenseNet 从自然图像迁移到医学图像数据集,作为特征提取部分;并且采用反卷积和类似 U-Net 的全卷积神经网络结构,实现端到端的图像分割。同时引入并改进 Dice 相似性损失函数以解决前列腺 MRI 中背景所占比例远远大于前列腺区域和一些像素难以准确分割等问题。该方法在 PROMISE12 数据集上进行实验,Dice 相似性系数达到 93.25%,Hausdorff 距离小于 1.2mm,相较于目前的主要方法,分割效果更好,所耗时间更短。
2. 基于 U-Net 网络及其改进网络所使用的常规卷积因其固定的几何结构限制了对几何形变的学习能力,上采样方法也存在一定缺陷,针对这些问题提出了一种改进的 U-Net 网络结构。该分割网络采用改进的可变形卷积代替编码器中的常规卷积,并且提出了一种新的上采样方法。该上采样方法是可学习的,并且不需要像反卷积那样填充零。针对生物医学图像中样本分布不均和简单样本过多导致的模型训练困难问题,将 Focal Loss 应用到生物医学图像分割任务中。该分割方法不仅大大降低了网络的参数数量,并且在果蝇细胞数据集和 Warwick-QU 腺体分割数据集上取得了显著的效果,较于现有主要方法具有更强的泛化能力,显著提高了分割精度。
5.2 未来工作展望
本文虽然对基于全卷积神经网络的医学图像分割算法进行研究,提出了两种改进方法,并获得了较好的分割效果。但还有不足,本文内容还有进一步研究与完善的空间,后续的研究工作方向如下:
1. 本文第三章中提出的“基于全卷积 DenseNet 的医学图像分割新算法”,网络结构复杂,参数数量极多,容易出现过拟合。虽然采用了数据增强和 L2 正则化等方法来防止过拟合,但并没有完全解决。为此,更多的防止过拟合方法是进一步的研究方向,包括更多的数据增强方法或者利用生成对抗网络(GAN)来生成图片,添加 DropOut 等。
2. 本文第四章中提出的“基于改进 U-Net 网络结构的医学图像分割方法”,采用了可变形卷积,提高了网络特征提取的能力,并极大地减少了网络的参数数量。但是由于其对输入特征图扩大了九倍,训练时,在反向传播结束前需要保留所有特征图,因此会占用极大的 GPU 资源,导致可变形卷积的数目不能过多,否则会出现内存溢出的情况。为此,可以进一步研究可变形卷积的优化方法,降低GPU 的占用率。
3. 本文所提出的两种改进网络采用的都是 2D 卷积,而很多医学图像是许多切片构成的 3D 图像,2D 卷积很难学习到切片间的空间信息,这些空间信息对目
标的识别有时也是非常重要的。在以后的研究工作中,可以考虑利用 3D 卷积来提取切片间的信息,提高分割结果的准确率。