[发明专利]一种融合通道和空间注意力的卷积神经网络息肉分割方法

说明书

本发明公开了一种融合通道和空间注意力的卷积神经网络息肉分割方法，包括：数据预处理，用于对息肉图像的分辨率进行调整以及数据增强；编码器阶段，使用残差模块对息肉图像进行多尺度特征提取；瓶颈层阶段，利用改进的挤压激励模块，增强对息肉图像高级特征的提取；解码器阶段，使用通道和空间组合注意力模块，提高分割精度；同时对于瓶颈层和解码器的输出使用跨层融合；并在上采样过程中使用深监督技术，减轻在训练时的梯度消失或爆炸现象，加速模型的收敛。本发明解决了息肉与周围组织边界模糊不清以及小的息肉区域难以分割的问题，同时深监督机制的引入优化了网络模型的梯度，加快了网络模型的收敛，缩短了网络模型的训练时间。

技术领域

本发明涉及医学图像分割和人工智能技术领域，具体涉及一种融合通道和空间注意力的卷积神经网络息肉分割方法。

背景技术

结直肠癌(CRC)是临床医学中最常见的诊断癌症之一，而结肠息肉往往是病变的最初表现形式。早期的息肉大多为良性，但如果不能够得到及时的治疗，随着时间的推移，部分结肠息肉会发展为结肠癌。息肉分割是对结肠镜采集的视频或者图像进行分割，辅助医生进行诊断和病理分析，以达到提高诊断效率和减少误诊率的作用，在结直肠癌的筛查和诊疗过程中扮演者不可或缺的角色。但由于息肉的形态差异大、边界模糊，使得息肉很难被精确的分割，导致漏检、误检的情况时有发生。由此可见，精确的息肉分割是一项具有挑战性的医学图像分析任务，对临床预防结直肠癌有着重要的意义。

传统的息肉分割方法主要基于颜色，纹理，形状等浅层特征或者通过组合这些特征来进行处理。这类方法不仅费时费力，分割效果往往较差，难以真正应用于临床实践。自全卷积网络(FCN)被Long提出以来，神经网络在图像分割领域取得了巨大的成功。FCN是首个用于语义分割的网络模型，将卷积神经网络开创性地引入到了语义分割领域，并且在图像分割中取得了不错的结果。Akbari等将改进的FCN应用到息肉分割中，相较于FCN提高了分割的准确性。进一步地，U-Net在FCN的基础上，使用对称的编码器-解码器结构，在医学分割领域取得了不错的成绩。在U-Net的基础上，发展了很多改进算法，并且都在医学图像分割领域取得了良好的效果。UNet++通过将不同深度的U-Net嵌入到同一个网络中，增强了编码器特征与解码器的特征之间的语义匹配程度，并且使用深监督策略对网络进行剪枝，提高了推理的速度。Double U-Net使用两个U-Net网络进行图像分割，利用迁移学习从第一个U-Net中学习特征然后逐级传递到第二个U-Net中，并利用第二个U-Net捕获更多的语义信息。ResUNet++在U-Net的基础上将残差单元、空间金字塔和SE引入到网络之中，提高了对语义的准确预测。这些U-Net变体尽管取得了不错的分割效果，但都十分依赖内存，需要较高的内存才能使用。

发明内容

发明目的：本发明的目的在于解决U-Net网络层数不够深，训练时容易出现过拟合，而且对前景像素的敏感度不足，难以获得足够的全局信息和局部语义信息的问题，提出一种融合通道和空间注意力的卷积神经网络息肉分割方法，即CSCA U-Net。

技术方案：一种融合通道和空间注意力的卷积神经网络息肉分割方法，包括以下步骤：

(1)准备数据集，分为训练和测试数据集；

(2)利用残差模块(Residual Block,RB)作为编码器提取图像的特征；

(3)利用双重挤压激励模块(Double Squeeze and Excited,DSE)增强瓶颈层对高级特征的提取能力；

(4)在解码器使用通道和空间组合的注意力模块(Channel and Space Compound,CSCA)，从通道和空间两个层面上关注关键区域，从提取到的特征选择最相关的特征，抑制不相关的特征，提高分割精度；此外使用跨层融合(Cross-Layer Fusion)的方式，捕获上下文信息，细化分割的边界，通过解码器，最终生成全局预测图Sg；