Journal of Hebei University(Natural Science Edition) ›› 2021, Vol. 41 ›› Issue (4): 426-435.DOI: 10.3969/j.issn.1000-1565.2021.04.013

Previous Articles     Next Articles

Multi-dimensional feature extraction network for liver image segmentation

LIU Rui1,XU Xinying1,XIE Jun2   

  1. 1.College of Electrical and Power Engineering, Taiyuan University of Technology, Taiyuan 030024, China; 2.College of Information and Computer, Taiyuan University of Technology, Jinzhong 030600, China
  • Received:2021-04-07 Published:2021-09-03

Abstract: 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 3D 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 3D 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- DOI:10.3969/j.issn.1000-1565.2021.04.013基于多维度特征提取网络的肝脏图像分割刘蕊1,续欣莹1,谢珺2(1.太原理工大学 电气与动力工程学院,山西 太原 030024;2.太原理工大学 信息与计算机学院,山西 晋中 030600)摘 要:随着计算机技术的发展,基于深度学习的医学图像自动分割已经成为人工智能辅助医疗的重要研究方向.为弥补现有神经网络结构对信息提取不足而产生的边缘细节丢失问题,构建了一种基于多维度特征提取网络(RDD-UNet)模型,该模型是基于残差UNet和混合损失函数的三维分割网络,以向肝脏肿瘤分割方法提供高精度的脏器分割结果.首先,该网络从原始CT数据的3个轴向提取信息,以长短跳跃连接的组合形式融合多尺度语义特征,保证了层内和层间信息的充分利用.其次,网络中设计了不平衡深度可分离空洞卷积模块,在提升三维网络计算效率的同时,扩大了体素级别的特征感受范围.最后,针对小尺寸分割目标数据不平衡问题提出了混合损失函数,并与深度监督结构相结合,提升了边缘细节的分割效果.该网络模型从体素、轴向和网络层级3个维度上充分提取特征信息,提高了肝脏分割的准确率,在公共数据集LiTS 2017上的Dice分数达到0.965 2,与其他方法相比达到了较高的精度水平.关键词:三维肝脏图像分割;残差连接;混合损失函数;深度可分离空洞卷积中图分类号:TP391 文献标志码:A 文章编号:1000-1565(2021)04-0426-10Multi-dimensional feature extraction network for liver image segmentationLIU Rui1,XU Xinying1,XIE Jun2(1.College of Electrical and Power Engineering, Taiyuan University of Technology, Taiyuan 030024, China;2.College of Information and Computer, Taiyuan University of Technology, Jinzhong 030600, China)Abstract: 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 3D 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 3D 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- 收稿日期:2021-04-07 基金项目:山西省自然科学基金资助项目(201801D121144;201801D221190) 第一作者:刘蕊(1995-),女,河北衡水人,太原理工大学在读硕士研究生,主要从事计算机视觉和医学图像处理研究.E-mail: 137730427@qq.com 通信作者:谢珺(1979-),女,山西太原人,太原理工大学副教授,博士,主要从事粒计算、数据挖掘和智能信息处理研究.E-mail: xiejun@tyut.edu.cn第4期刘蕊等:基于多维度特征提取网络的肝脏图像分割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.965 2 on the liver segmentation results of LiTS dataset, which achieves a higher accuracy level compared with other methods.

Key words: 3D liver image segmentation, residual connection, mixed loss function, depth-wise separable dilated convolution

CLC Number: