提出高效双路解码网络,兼顾分割精度与计算效率。
An Efficient Dual-Line Decoder Network with Multi-Scale Convolutional Attention for Multi-organ Segmentation
- 双路解码结构:训练时加入噪声提升鲁棒性,推理时仅用干净解码器降开销。
- 在Synapse数据集上达84.00%的Dice分数,比UNet高13.89%,MACs减少89.7%。
- 适用于需高精度与低延迟的医疗影像分割场景,如放疗规划与手术导航。
器官风险区的准确分割对放射治疗、手术规划和诊断决策至关重要。尽管基于深度学习的分割架构取得显著进展,但往往难以平衡分割精度与计算效率。现有先进方法要么以高复杂度换取性能,要么牺牲精度以求效率。本文提出一种高效的双路解码分割网络(EDLDNet),其核心为带噪声的解码器,在训练时引入结构化扰动以增强模型鲁棒性,推理时仅使用无噪声解码器,从而降低计算开销。进一步结合多尺度卷积注意力模块(MSCAM)、注意力门(AG)和上采样卷积块(UCB)优化特征表示并提升分割性能。通过融合双解码器生成的多尺度分割掩码,采用基于突变的损失函数增强模型泛化能力。在四个公开医学图像数据集上,EDLDNet超越当前最优模型:在Synapse数据集上达到84.00%的Dice分数,较基线UNet提升13.89%,且乘累加操作(MACs)降低89.7%。相较于EMCAD等近期方法,不仅精度更高,计算效率也相当。跨多样数据集的优异表现证明了EDLDNet在泛化性、计算效率与鲁棒性方面的优势。源代码、预处理数据及预训练权重将开源于https://github.com/riadhassan/EDLDNet。
原文摘要 · Abstract (English)
Proper segmentation of organs-at-risk is important for radiation therapy, surgical planning, and diagnostic decision-making in medical image analysis. While deep learning-based segmentation architectures have made significant progress, they often fail to balance segmentation accuracy with computational efficiency. Most of the current state-of-the-art methods either prioritize performance at the cost of high computational complexity or compromise accuracy for efficiency. This paper addresses this gap by introducing an efficient dual-line decoder segmentation network (EDLDNet). The proposed method features a noisy decoder, which learns to incorporate structured perturbation at training time for better model robustness, yet at inference time only the noise-free decoder is executed, leading to lower computational cost. Multi-Scale convolutional Attention Modules (MSCAMs), Attention Gates (AGs), and Up-Convolution Blocks (UCBs) are further utilized to optimize feature representation and boost segmentation performance. By leveraging multi-scale segmentation masks from both decoders, we also utilize a mutation-based loss function to enhance the model's generalization. Our approach outperforms SOTA segmentation architectures on four publicly available medical imaging datasets. EDLDNet achieves SOTA performance with an 84.00% Dice score on the Synapse dataset, surpassing baseline model like UNet by 13.89% in Dice score while significantly reducing Multiply-Accumulate Operations (MACs) by 89.7%. Compared to recent approaches like EMCAD, our EDLDNet not only achieves higher Dice score but also maintains comparable computational efficiency. The outstanding performance across diverse datasets establishes EDLDNet's strong generalization, computational efficiency, and robustness. The source code, pre-processed data, and pre-trained weights will be available at https://github.com/riadhassan/EDLDNet .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。