提出PMR-Net框架,提升医学图像分割精度与灵活性。
PMR-Net: Parallel Multi-Resolution Encoder-Decoder Network Framework for Medical Image Segmentation
- 并行多分辨率编码器融合多尺度局部特征
- 解码器同步补充全局上下文信息,减少上采样损失
- 可灵活调整结构,适配不同场景需求
近年来,编码器-解码器网络致力于扩大感受野并引入多尺度上下文以捕捉不同尺寸物体的全局特征。然而,随着网络加深,细粒度空间细节常被丢失,影响精确定位;传统解码器采用插值上采样,导致全局上下文丢失,降低边缘分割精度。为此,本文提出一种新型并行多分辨率编码器-解码器网络PMR-Net。首先设计并行多分辨率编码器与多分辨率上下文编码器:前者并行提取和融合不同分辨率输入的细粒度局部特征;后者融合各编码分支来自不同感受野的全局语义特征,有效保持全局信息完整性。其次设计对称的并行多分辨率解码器:可将低分辨率分支的全局上下文特征持续补充至高分辨率分支特征图,有效缓解解码过程中上采样带来的上下文信息损失。大量实验表明,所提PMR-Net在五个公开数据集上均优于现有先进方法。此外,该框架具有高度灵活性,通过调节网络层数与并行分支数,可满足不同应用场景需求。
原文摘要 · Abstract (English)
In recent years, encoder-decoder networks have focused on expanding receptive fields and incorporating multi-scale context to capture global features for objects of varying sizes. However, as networks deepen, they often discard fine spatial details, impairing precise object localization. Additionally, conventional decoders' use of interpolation for upsampling leads to a loss of global context, diminishing edge segmentation accuracy. To address the above problems, we propose a novel parallel multi-resolution encoder-decoder network, namely PMR-Net for short. First, we design a parallel multi-resolution encoder and a multi-resolution context encoder. The parallel multi-resolution encoder can extract and fuse multi-scale fine-grained local features in parallel for input images with different resolutions. The multi-resolution context encoder fuses the global context semantic features of different receptive fields from different encoder branches to maintain effectively the integrity of global information. Secondly, we design a parallel multi-resolution decoder symmetrical to the structure of parallel multi-resolution encoder. The decoder can continuously supplement the global context features of low-resolution branches to the feature maps of high-resolution branches, and effectively solve the problem of global context feature loss caused by upsampling operation in the decoding process. Extensive experiment results demonstrate that our proposed PMR-Net can achieve more accurate segmentation results than state-of-the-art methods on five public available datasets. Moreover, PMR-Net is also a flexible network framework, which can meet the requirements of different scenarios by adjusting the number of network layers and the number of parallel encoder-decoder branches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。