基于U-net的直肠癌智能分割谭俊杰钟妤黄泽斌摘要：近年来，随着医学影像与大数据的发展，通过人工智能来处理图像，从而给医生带来辅助性的诊断参考成为广大学者的研究热点。文章利用U-net网络对直肠癌CT影像进行智能分割，加入了图像增强、批归一化等技巧缓解过拟合现象，并通过多次实验确定最佳初始学习速率和卷积核数目，在验证集上的Dice系数达到0.9329。实验表明，U-Net对小数据集的医学图像分割有很好效果，对正负样本极度偏斜的数据集，使用Dice系数可以准确地衡量分割的相似程度。关键词：直肠癌;图像分割：U-net：Dice系数中图分类号：TP301.6文献标识码：A文章编号：1006-8228（2020）08-18-030引言近年来，我国直肠癌发病率呈显著上升趋势，特别是在大城市，已跃居恶性肿瘤发病率排行榜前三位。研究显示，有65%的患者的发病时间是在40岁之后，其中男女比例为23：1，并且发病率最高的人群年龄分布在40岁-50岁。医学图像分割是临床诊断和治疗的关键技术，能够提供准确可靠的诊断依据。随着深度学习中图像翻译算法的普遍应用，医学图像分割技术也获得了巨大的进展。尤其在U-Net网络结构被提出后，该网络一直在医学图像分割领域表现优秀，众多研究者利用此模型进行医学图像分割[1]，并在此基础上提出改进。本文研究来自于第七届“泰迪杯”数据挖掘挑战赛B题，对直肠CT影像中的肿瘤分割问题提出一种基于深度学习U-Net网络结构的分割方法。1数据预处理1.1数据清洗与划分本数据集共有107个病例，3029张CT影像及对应的掩模图，数据样本为512×512像素。而且每个病例有1745张CT影像序列。其中，有肿瘤区域的CT影像仅有860张且肿瘤区域在CT影像中占比较小，属于正负样本不平衡的数据集。因此，在划分数据集时，为使训练集、验证集和测试集的正负样本分布尽可能地相近，将病例数作为划分对象，训练集、验证集、测试集的个数分别设置为65个（即0.6）、16个和26个。通过掩模图，可观察出肿瘤位置均在图像的一定区域内，因此，本文运用OpenCV库计算出肿瘤区域的最大最小像素值。经统计，长为126-382，宽为202-458的像素范围能包含所有肿瘤特征。因此，最终将数据集裁剪成256x256大小的影像，即过滤了无关特征，同时节省资源，减少计算量。1.2图像增强由于划分后的训练集只有1820张CT影像，对深度学习而言远远不足。因此，需要通过图像增强[2]技术扩充训练集样本量，本文使用水平翻转、垂直翻转、90度旋转以及翻转和旋转的组合将训练集样本量扩充至原来的6倍。图像增强对模型性能的训练有很大的提升，既可以减少过拟合现象的发生，又能提升模型的泛化能力。效果如图l所示。2U-Net模型训练2.1窗宽和窗位由于CT影像的格式为DICOM，首先利用RadiAntDICOMViewer软件得出每个病例CT序列的最佳窗宽和窗位，其次利用SimpleITK库读取CT影像的HU值，将HU值储存为数组后设置其最佳窗宽和窗位，使用灰度直方图均衡化技术提高具体器官的对比度。最后在模型训练前，对样本数据做均值方差归一化，大幅度减少计算量。2.2U-Net结构U-Net网络结构是全卷积神经网络[3]（FCN）基础的改进，主要包括卷积、池化和反卷积三大步骤。其中，卷积操作中每一个卷积块包括两次卷积，然后是最大值池化操作，通过10次卷積操作后先结合对应大小的特征图进行反卷积操作使图像大小翻倍而通道数缩减一倍，再进行两次卷积操作，激活函数为ReLU。本文针对医学图像数据集对U-Net网络结构进行超参数调整，由于肿瘤占图像范围区域小，提取的特征数太多易导致过拟合。经过多次实验，本文对256×256大小的输入图像初始提取两个特征，每经过一次最大值池化，提取的特征数翻倍。每个卷积核的大小设置为3x3，其中步长为l，Padding设置为same，这样即可保证卷积后的图像与原来图像大小一致。反卷积操作后再经过一层卷积层，卷积核大小为1x1，利用Sigmoid函数进行二分类。U-net网络结构如图2所示。本文在卷积和反卷积操作后加入批归一化（Batch_Normalization）的操作，使上一层传递到下一层的参数分布保持一致，这既能加快训练速度，防止低层次神经网络的梯度在反向传播中消失，又能减少参数初始化的影响。2.3衡量指标考虑到这个数据集...