Skip to main content
QUICK REVIEW

[论文解读] Cascaded Volumetric Convolutional Network for Kidney Tumor Segmentation from CT volumes

Yao Zhang, Yixin Wang|arXiv (Cornell University)|Oct 5, 2019
Advanced Neural Network Applications参考文献 5被引用 7
一句话总结

本文提出一种两级级联的3D全卷积网络,用于在CT影像中精确分割肾脏和肿瘤,通过粗略的肾脏定位来降低计算成本和类别不平衡,随后在空间先验和残差连接的辅助下进行细粒度分割。该方法在MICCAI 2019 KiTS挑战赛中取得了90.24%的综合Dice分数,排名第四。

ABSTRACT

Automated segmentation of kidney and tumor from 3D CT scans is necessary for the diagnosis, monitoring, and treatment planning of the disease. In this paper, we describe a two-stage framework for kidney and tumor segmentation based on 3D fully convolutional network (FCN). The first stage preliminarily locate the kidney and cut off the irrelevant background to reduce class imbalance and computation cost. Then the second stage precisely segment the kidney and tumor on the cropped patch. The proposed method ranks the 4th place out of 105 competitive teams in MICCAI 2019 KiTS Challenge with a Composite Dice of 90.24%.

研究动机与目标

  • 解决从CT影像中进行3D肾脏和肿瘤分割时面临的高计算成本和类别不平衡问题。
  • 通过基于初步肾脏定位的感兴趣区域(ROI)裁剪,减少背景干扰,提升分割准确率。
  • 在第二阶段分割网络中引入残差连接、深度监督和空间先验,以提升模型性能。
  • 通过聚焦于相关解剖区域,实现在有限GPU内存下的有效训练。
  • 在MICCAI 2019年肾脏肿瘤分割挑战赛(KiTS 2019)中实现最先进性能。

提出的方法

  • 采用两级级联框架:第一阶段使用带有实例归一化和Leaky ReLU激活函数的3D U-Net进行粗略肾脏定位。
  • 利用第一阶段的输出提取以肾脏为中心的裁剪3D块(80×160×160),以减少内存消耗和类别不平衡。
  • 第二阶段采用多任务3D U-Net,结合残差块、深度监督和空间先验(肾脏掩码),在裁剪块上联合分割肾脏和肿瘤。
  • 通过百分位数裁剪(0.05%–99.5%)和按病例标准化对输入数据进行归一化,以稳定训练并提升泛化能力。
  • 模型使用转置卷积进行上采样,并通过跳跃连接保留解码过程中的高分辨率特征。
  • 对第一阶段输入应用各向异性重采样,以平衡各轴上的分辨率,提升上下文特征学习能力。

实验结果

研究问题

  • RQ1两级级联的3D全卷积网络框架是否能在降低计算成本的同时提升肾脏和肿瘤在CT影像中的分割准确率?
  • RQ2在第二阶段肿瘤分割中,引入第一阶段肾脏掩码的空间先验在多大程度上提升了模型性能?
  • RQ3在GPU内存有限的条件下,残差连接和深度监督在3D体积分割中对模型学习的增强作用有多大?
  • RQ4基于粗略定位的ROI裁剪是否能有效缓解类别不平衡并提升训练效率?
  • RQ5与KiTS 2019挑战赛中的最先进2D和3D全卷积网络方法相比,所提方法表现如何?

主要发现

  • 所提方法在KiTS 2019在线测试集上取得了90.24%的综合Dice分数,105支参赛队伍中排名第4。
  • 第二阶段模型在验证集上,肾脏Dice达到98.05%,肿瘤Dice达到83.70%,所有增强组件(残差块、深度监督、空间先验)均被采用。
  • 与基线3D U-Net(带深度监督)相比,使用第一阶段肾脏掩码的空间先验使肿瘤分割Dice提高了3.08个百分点。
  • 通过将输入裁剪至中位大小40×160×160,模型将内存消耗降低,相比原始138×512×512体素。
  • 第一阶段定位模型在测试集上肾脏Dice达到97.42%,证明了ROI提取的有效性。
  • 该方法对解剖变异性和类别不平衡具有鲁棒性,通过连通组件分析后处理有效去除了假阳性结果。

更好的研究,从现在开始

从阅读论文到最终审阅,大幅缩短您的研究时间。

无需绑定信用卡

本解读由 AI 生成,并经人工编辑审核。