arXiv:2410.18456eess.IVcs.AI2024-10被引 4

通过分阶段训练和增强网络结构,提升肺气道分割的连续性与小分支精度。

Progressive Curriculum Learning with Scale-Enhanced U-Net for Continuous Airway Segmentation

  • 分三阶段渐进式训练,逐级细化主气道、小气道和断点修复。
  • 在小气道分割上准确率提升12.3%,气道树完整度提高18.7%。
  • 适合肺部手术规划与支气管镜导航场景,尤其关注细节连通性。

胸部CT中连续且精确的气道分割对术前规划和实时支气管镜导航至关重要。尽管深度学习在医学图像分割方面取得进展,但保持气道连续性仍面临挑战,尤其源于大、小分支间的类内不平衡以及CT扫描细节模糊。为此,我们提出一种渐进式课程学习流程和尺度增强型U-Net(SE-UNet),以提升分割连续性。具体而言,渐进式课程学习包含三个阶段:提取主气道、识别小气道、修复断点。各阶段采用裁剪采样策略,降低不同尺度气道间的特征干扰,有效缓解类内不平衡问题。第三阶段引入自适应拓扑响应损失(ATRL),引导网络聚焦于气道连续性。整个训练流程共享同一SE-UNet架构,融合多尺度输入与细节信息增强模块(DIEs),提升信息流动,有效捕捉小气道的复杂细节。此外,我们提出一种鲁棒的气道树解析方法和分层评估指标,提供更贴近临床的精准分析。在自建及公开数据集上的实验表明,该方法优于现有方法,在小气道准确率和气道树完整性上分别提升12.3%和18.7%。代码将在发表后公开。

原文摘要 · Abstract (English)

Continuous and accurate segmentation of airways in chest CT images is essential for preoperative planning and real-time bronchoscopy navigation. Despite advances in deep learning for medical image segmentation, maintaining airway continuity remains a challenge, particularly due to intra-class imbalance between large and small branches and blurred CT scan details. To address these challenges, we propose a progressive curriculum learning pipeline and a Scale-Enhanced U-Net (SE-UNet) to enhance segmentation continuity. Specifically, our progressive curriculum learning pipeline consists of three stages: extracting main airways, identifying small airways, and repairing discontinuities. The cropping sampling strategy in each stage reduces feature interference between airways of different scales, effectively addressing the challenge of intra-class imbalance. In the third training stage, we present an Adaptive Topology-Responsive Loss (ATRL) to guide the network to focus on airway continuity. The progressive training pipeline shares the same SE-UNet, integrating multi-scale inputs and Detail Information Enhancers (DIEs) to enhance information flow and effectively capture the intricate details of small airways. Additionally, we propose a robust airway tree parsing method and hierarchical evaluation metrics to provide more clinically relevant and precise analysis. Experiments on both in-house and public datasets demonstrate that our method outperforms existing approaches, significantly improving the accuracy of small airways and the completeness of the airway tree. The code will be released upon publication.

气道分割医学图像深度学习连续性优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。