针对医学图像小结构分割难,提出轻量级网络与轮廓加权损失,提升准确率与鲁棒性。
Partial Decoder Attention Network with Contour-weighted Loss Function for Data-Imbalance Medical Image Segmentation
- 设计部分解码器网络,聚焦小器官区域,增强对稀疏结构的建模能力。
- 在三个数据集上平均提升Dice分数2.32%~3.60%,显著改善小结构分割效果。
- 策略可嵌入多种框架,适合需高精度分割小病灶的临床研究与系统开发。
医学图像分割对临床诊断、治疗规划和疾病评估至关重要。深度学习虽显著提升了复杂结构与细粒度解剖区域的建模能力,但医学图像常存在器官体积差异大、样本分布不均等问题,导致模型偏向大器官而忽略小或低频结构,影响分割精度与鲁棒性。为此,本文提出一种新的轮廓加权分割方法,旨在提升模型对小结构和低频结构的表征能力。我们构建了基于部分解码器机制的轻量高效网络PDANet。在三个公开数据集上的实验表明,该方法在腹部多器官、脑肿瘤及骨盆骨折碎片分割任务中均优于九种前沿方法。同时,所提轮廓加权策略使其他对比方法在三组数据集上平均Dice得分分别提升2.32%、1.67%和3.60%。结果表明,该方法在精度与鲁棒性上均超越现有主流方案。作为模型无关策略,其可无缝适配各类分割框架,显著提升性能,具备广泛实际应用潜力。
原文摘要 · Abstract (English)
Image segmentation is pivotal in medical image analysis, facilitating clinical diagnosis, treatment planning, and disease evaluation. Deep learning has significantly advanced automatic segmentation methodologies by providing superior modeling capability for complex structures and fine-grained anatomical regions. However, medical images often suffer from data imbalance issues, such as large volume disparities among organs or tissues, and uneven sample distributions across different anatomical structures. This imbalance tends to bias the model toward larger organs or more frequently represented structures, while overlooking smaller or less represented structures, thereby affecting the segmentation accuracy and robustness. To address these challenges, we proposed a novel contour-weighted segmentation approach, which improves the model's capability to represent small and underrepresented structures. We developed PDANet, a lightweight and efficient segmentation network based on a partial decoder mechanism. We evaluated our method using three prominent public datasets. The experimental results show that our methodology excelled in three distinct tasks: segmenting multiple abdominal organs, brain tumors, and pelvic bone fragments with injuries. It consistently outperformed nine state-of-the-art methods. Moreover, the proposed contour-weighted strategy improved segmentation for other comparison methods across the three datasets, yielding average enhancements in Dice scores of 2.32%, 1.67%, and 3.60%, respectively. These results demonstrate that our contour-weighted segmentation method surpassed current leading approaches in both accuracy and robustness. As a model-independent strategy, it can seamlessly fit various segmentation frameworks, enhancing their performance. This flexibility highlighted its practical importance and potential for broad use in medical image analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。