基于多维度特征提取网络的肝脏图像分割

Multi-dimensional feature extraction network for liver image segmentation

随着计算机技术的发展,基于深度学习的医学图像自动分割已经成为人工智能辅助医疗的重要研究方向.为弥补现有神经网络结构对信息提取不足而产生的边缘细节丢失问题,构建了一种基于多维度特征提取网络(RDD-UNet)模型,该模型是基于残差UNet和混合损失函数的三维分割网络,以向肝脏肿瘤分割方法提供高精度的脏器分割结果.首先,该网络从原始CT数据的3个轴向提取信息,以长短跳跃连接的组合形式融合多尺度语义特征,保证了层内和层间信息的充分利用.其次,网络中设计了不平衡深度可分离空洞卷积模块,在提升三维网络计算效率的同时,扩大了体素级别的特征感受范围.最后,针对小尺寸分割目标数据不平衡问题提出了混合损失函数,并与深度监督结构相结合,提升了边缘细节的分割效果.该网络模型从体素、轴向和网络层级3个维度上充分提取特征信息,提高了肝脏分割的准确率,在公共数据集LiTS 2017上的Dice分数达到0.9652,与其他方法相比达到了较高的精度水平.

With the development of computer technology,automatic medical image segmentation based on deep learning has become an important research field of artificial intelligence assisted medicine.Yet,many existing neural network structures could not be able to integrate enough semantic feature information,which led to the loss of marginal details.In order to solve this problem,a multi-dimensional feature extraction network model(RDD-UNet)is proposed.This model is based on 3 D residual UNet with multi-loss,which can provide high-precision organ segmentation results for liver tumor segmentation methods.At first,the network extracts information from the three axes of the original CT data,and long and short skip connections are added into the network ensuring the full usage of inter-slice and intra-slice information.In addition,the unbalanced depth-wise separable dilated convolution block is designed to improve the calculation efficiency of the 3 D network and expand the receptive field at the voxel level.Finally,a new multi-loss function is proposed to solve the problem of imbalanced data label on small size objections,which is combined with deep supervision structure to improve the segmentation effect of edge details.The proposed method can adequately extract feature information from voxel-wise,axis-wise and hierarchy-wise,and improve the accuracy of liver segmentation.This method gets the Dice score of 0.9652 on the liver segmentation results of LiTS dataset,which achieves a higher accuracy level compared with other methods.