提出HarmonySeg框架,提升医学图像中管状结构分割精度与鲁棒性。
HarmonySeg: Tubular Structure Segmentation with Deep-Shallow Feature Fusion and Growth-Suppression Balanced Loss
- 采用深浅特征融合的解码器,适应不同尺度的管状结构。
- 引入血管度图辅助,显著提升小结构召回率且保持高精度。
- 设计拓扑保持损失函数,有效处理标注不完整和低质量数据。
医学图像中管状结构(如血管、气道树)的精准分割对计算机辅助诊断、放疗规划和手术设计至关重要。然而,面对多样的尺寸、复杂的拓扑结构以及常存在的不完整标注数据,算法设计面临挑战。本文提出HarmonySeg框架:首先设计深层到浅层的解码网络,采用可变感受野的灵活卷积块,使模型能适应不同尺度的管状结构;其次,引入血管度图作为辅助信息,通过浅-深特征融合模块,强化潜在解剖区域并剔除不合理候选,提升小结构召回率同时维持高精度;最后,提出一种保留拓扑结构的损失函数,利用上下文与形状先验平衡管状结构的生长与抑制,增强对低质量及不完整标注数据的适应能力。在四个公开数据集上进行大量定量实验,结果表明该模型能准确分割2D/3D管状结构,优于现有最先进方法。私有数据集上的外部验证也证实了其良好的泛化能力。
原文摘要 · Abstract (English)
Accurate segmentation of tubular structures in medical images, such as vessels and airway trees, is crucial for computer-aided diagnosis, radiotherapy, and surgical planning. However, significant challenges exist in algorithm design when faced with diverse sizes, complex topologies, and (often) incomplete data annotation of these structures. We address these difficulties by proposing a new tubular structure segmentation framework named HarmonySeg. First, we design a deep-to-shallow decoder network featuring flexible convolution blocks with varying receptive fields, which enables the model to effectively adapt to tubular structures of different scales. Second, to highlight potential anatomical regions and improve the recall of small tubular structures, we incorporate vesselness maps as auxiliary information. These maps are aligned with image features through a shallow-and-deep fusion module, which simultaneously eliminates unreasonable candidates to maintain high precision. Finally, we introduce a topology-preserving loss function that leverages contextual and shape priors to balance the growth and suppression of tubular structures, which also allows the model to handle low-quality and incomplete annotations. Extensive quantitative experiments are conducted on four public datasets. The results show that our model can accurately segment 2D and 3D tubular structures and outperform existing state-of-the-art methods. External validation on a private dataset also demonstrates good generalizability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。