计算机工程与应用››2024,第60卷››发行(8): 258-266.内政部:10.3778/j.issn.1002-8331.2302-0002

图形图像处理 • 上一篇   下一篇

卷曲:基于卷积胶囊网络的医学图像分割模型

邓希泉,陈刚  

  1. 1武汉大学 空天信息安全与可信计算教育部重点实验室,武汉 430072
    2.武汉大学 国家网络安全学院,武汉 430072
  • 出版日期:2024-04-15 发布日期:2024-04-15

卷积:基于卷积胶囊网络的医学图像分割模型

邓锡泉、陈刚  

  1. 1.武汉大学航空航天信息安全与可信计算教育部重点实验室,武汉430072
    2.武汉大学网络科学与工程学院,中国武汉430072
  • 在线:2024-04-15 出版:2024-04-15

摘要:在医学影像分割领域,掌中宽带网络是目前最成功和最受关注的方法之一,但是掌中宽带本质上是一种经过改造的全卷积神经网络模型,要获得更为全面和准确的局部-整体关系,不但需要增加网络层次从而加大计算量,而且效果也并不明显。胶囊网络提供了一种有效的建模图像的局部与整体关系的方法,可以用更少的参数取得好的性能。但原始的胶囊网络并没有充分考虑图像局部特征的粒度问题,将其应用在医学图像分割领域还需进一步改造。因此,提出一种将掌中宽带和胶囊网络相结合的医学图像分割模型卷曲该模型对掌中宽带的编码器部分进行改进,使用卷积模块学习不同尺度的局部特征,然后通过胶囊模块学习高层特征,并建模局部与整体之间的关系。实验结果表明,相比U-Net、UNet++、SegCaps、Matwo-CapsNet、ConvUCaps提高了分割精度和收敛速度,同时,与单纯基于胶囊网络的分割模型相比,显著减少了推理时间。

关键词: 医学图像分割, 卷积神经网络, U-Net网络, 胶囊网络

摘要:在医学图像分割领域,U-Net网络是目前最成功和最受关注的方法之一。然而,U-Net本质上是一种改进的完全卷积神经网络模型。为了获得更全面、更准确的局部-局部关系,不仅需要提高网络级别,增加计算量,而且效果并不明显。胶囊网络为图像的局部和整体关系建模提供了一种有效的方法,可以用较少的参数获得良好的性能。然而,原始胶囊网络并没有充分考虑图像局部特征的粒度,其在医学图像分割领域的应用需要进一步改进。因此,本文提出了一种将U-Net和胶囊网络相结合的医学图像分割模型ConvUCaps。该模型改进了U-Net的编码部分,使用卷积模块学习不同尺度的局部特征,然后使用胶囊模块学习高层特征并建模局部-整体关系。实验结果表明,与U-Net、UNet++、SegCaps和Matwo-CapsNet网络相比,ConvUCaps网络提高了分割精度和收敛速度。同时,与单纯基于胶囊网络的分割模型相比,推理时间大大缩短。

关键词: 医学图像分割, 卷积神经网络, U-Net网络, 膜盒网络