通过空间先验引导双编码器网络,提升多器官分割精度。
SPG-CDENet: Spatial Prior-Guided Cross Dual Encoder Network for Multi-Organ Segmentation
- 用空间先验生成粗略定位图,指导双编码器精分割
- 对称交叉注意力融合全局与局部特征,提升细节捕捉能力
- 适合医学图像中形状差异大的多器官分割任务
多器官分割是计算机辅助诊断中的关键任务。尽管深度学习在图像分割中取得显著进展,但器官大小和形状的巨大差异仍制约其在多器官分割中的表现。为此,我们提出一种两阶段分割框架SPG-CDENet,包含空间先验网络和交叉双编码器网络。先验网络生成粗略定位图,作为双编码器的空間引导。交叉双编码器包含全局编码器、局部编码器、对称交叉注意力模块和基于流的解码器。全局编码器捕获全图语义特征,局部编码器聚焦于先验图区域。通过跨所有层的对称交叉注意力模块,实现全局与局部特征的融合与优化。基于流的解码器将高层语义特征直接传递至各解码层,最大化特征保留与利用。在两个公开数据集上的定性和定量实验表明,SPG-CDENet优于现有方法。消融实验进一步验证了各模块的有效性。
原文摘要 · Abstract (English)
Multi-organ segmentation is a critical task in computer-aided diagnosis. While recent deep learning methods have achieved remarkable success in image segmentation, huge variations in organ size and shape challenge their effectiveness in multi-organ segmentation. To address these challenges, we propose a Spatial Prior-Guided Cross Dual Encoder Network (SPG-CDENet), a novel two-stage segmentation paradigm designed to improve multi-organ segmentation accuracy. Our SPG-CDENet consists of two key components: a spatial prior network and a cross dual encoder network. The prior network generates coarse localization maps that delineate the approximate ROI, serving as spatial guidance for the dual encoder network. The cross dual encoder network comprises four essential components: a global encoder, a local encoder, a symmetric cross-attention module, and a flow-based decoder. The global encoder captures global semantic features from the entire image, while the local encoder focuses on features from the prior network. To enhance the interaction between the global and local encoders, a symmetric cross-attention module is proposed across all layers of the encoders to fuse and refine features. Furthermore, the flow-based decoder directly propagates high-level semantic features from the final encoder layer to all decoder layers, maximizing feature preservation and utilization. Extensive qualitative and quantitative experiments on two public datasets demonstrate the superior performance of SPG-CDENet compared to existing segmentation methods. Furthermore, ablation studies further validate the effectiveness of the proposed modules in improving segmentation accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。