论文标题

UNET#:用于医疗图像细分的UNET般的重新设计跳过连接

UNet#: A UNet-like Redesigning Skip Connections for Medical Image Segmentation

论文作者

Qian, Ledan, Zhou, Xiao, Li, Yi, Hu, Zhongyi

论文摘要

作为开发医学智能助理系统的基本先决条件,医疗图像细分已获得神经网络社区的广泛研究和集中度。一系列具有编码器架构架构的UNET式网络已取得了非凡的成功,其中UNET2+和UNET3+重新设计跳过连接分别提出了密集的跳过连接以及与Medical Image分割中的UNET相比,与UNET相比,具有密集的跳过连接,并大大改善。但是,UNET2+缺乏从整个规模探索的足​​够信息,这将影响器官的位置和边界的学习。尽管UNET3+可以获得完整的聚合特征图,这是由于结构中的少量神经元,但是当样本数量较小时,它不能满足微小对象的分割。本文提出了一种新颖的网络结构,结合了密集的跳过连接和全尺度跳过连接,称为Unet-sharp(UNET \#),其形状类似于符号\#。所提出的UNET \#可以在解码器子网络中汇总不同尺度的特征图,并捕获完整量表的细颗粒细节和粗粒语义,这使学习确切的位置并准确地分割了器官或病变的边界。我们对模型修剪进行深入的监督,以加快测试加快测试,并使模型在移动设备上运行;此外,设计两个分类引导的模块以减少假阳性可实现更准确的分割结果。关于不同模态(EM,CT,MRI)和尺寸(2D,3D)数据集的语义分割和实例分割的各种实验,包括细胞核,脑肿瘤,肝脏和肺部,表明所提出的方法优于最先进的模型。

As an essential prerequisite for developing a medical intelligent assistant system, medical image segmentation has received extensive research and concentration from the neural network community. A series of UNet-like networks with encoder-decoder architecture has achieved extraordinary success, in which UNet2+ and UNet3+ redesign skip connections, respectively proposing dense skip connection and full-scale skip connection and dramatically improving compared with UNet in medical image segmentation. However, UNet2+ lacks sufficient information explored from the full scale, which will affect the learning of organs' location and boundary. Although UNet3+ can obtain the full-scale aggregation feature map, owing to the small number of neurons in the structure, it does not satisfy the segmentation of tiny objects when the number of samples is small. This paper proposes a novel network structure combining dense skip connections and full-scale skip connections, named UNet-sharp (UNet\#) for its shape similar to symbol \#. The proposed UNet\# can aggregate feature maps of different scales in the decoder sub-network and capture fine-grained details and coarse-grained semantics from the full scale, which benefits learning the exact location and accurately segmenting the boundary of organs or lesions. We perform deep supervision for model pruning to speed up testing and make it possible for the model to run on mobile devices; furthermore, designing two classification-guided modules to reduce false positives achieves more accurate segmentation results. Various experiments of semantic segmentation and instance segmentation on different modalities (EM, CT, MRI) and dimensions (2D, 3D) datasets, including the nuclei, brain tumor, liver, and lung, demonstrate that the proposed method outperforms state-of-the-art models.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源