吉林大学学报(理学版) ›› 2022, Vol. 60 ›› Issue (5): 1153-1160.

• • 上一篇    下一篇

基于改进U-Net的肝脏MRI分割方法

汪慎文, 周瑶   

  1. 河北地质大学 信息工程学院, 石家庄 050031; 河北地质大学 人工智能与机器学习研究室, 石家庄 050031; 河北地质大学 新零售联合研究院, 石家庄 050031
  • 收稿日期:2021-07-29 出版日期:2022-09-26 发布日期:2022-09-26
  • 通讯作者: 汪慎文 E-mail:wangshenwen@hgu.edu.cn

Liver MRI Segmentation Method Based on Improved U-NET

WANG Shenwen, ZHOU Yao   

  1. College of Information Engineering, Hebei GEO University, Shijiazhuang 050031,  China;
     Laboratory of Artificial Intelligence and Machine Learning, Hebei GEO University, Shijiazhuang 050031,  China;
     New Retail Joint Research Institute, Hebei GEO University, Shijiazhuang 050031,  China
  • Received:2021-07-29 Online:2022-09-26 Published:2022-09-26

摘要: 针对U型结构网络在特征提取过程中存在语义信息丢失的情况, 从而影响肝脏图像分割精度的问题, 提出一种基于多尺度特征融合, 并引入内嵌空间注意力和改进通道注意力的肝脏图像分割方法. 首先, 在网络的编码阶段使用多尺度卷积模块提取不同尺度的语义信息并拼接在一起; 然后在整个网络模型中引入空间注意力和改进的通道注意力, 从空间域和通道域的角度加强有意义语义信息的权重; 最后通过一个卷积层输出肝脏的分割结果. 实验结果表明, 该分割方法在肝脏图像数据集上分割效果较好, 提高了肝脏的分割准确率, 有效改善了小面积肝脏分割困难的问题.

关键词: 深度学习, 图像分割, 多尺度卷积, 注意力机制

Abstract: Aiming at the problem of the loss of semantic information in the feature extraction process of U-shaped network, which affected the accuracy of liver image segmentation, we  proposed a liver image segmentation method based on multi-scale feature fusion, introducing embedded spatial attention and improved channel attention. Firstly,  multi-scale convolution modules were used  to extract semantic information of different scales and stitch them together in the encoding stage of the network. Secondly,  spatial attention and improved channel attention were introduced into the whole network model to strengthen the weight of meaningful semantic information from the perspectives of spatial and channel domains. Finally, the liver segmentation results were output through a convolutional layer. The experimental results show that the segmentation method has a better segmentation effect on the liver image data set, improves the liver segmentation accuracy, and effectively improves the difficulty of small-area liver segmentation.

Key words: deep learning, image segmentation, multi-scale convolution, attention mechanism

中图分类号: 

  • TP391