arXiv:2511.04334cs.CVcs.LG2025-11被引 1

用稀疏卷积实现高精度肾及肿瘤3D分割,速度快内存少

Submanifold Sparse Convolutional Networks for Automated 3D Segmentation of Kidneys and Kidney Tumours in Computed Tomography

  • 先稀疏定位兴趣区,再高分辨率精细分割,避免全图计算
  • 肾+肿瘤骰子系数达95.8%,肿瘤单独分割达80.3%,性能领先
  • 适合医学影像高分辨率处理,尤其对算力有限的场景友好

CT中精准勾画肾肿瘤对定量分析和精准肿瘤学至关重要,但人工分割耗时且难扩展。自动化3D分割面临体积大、计算成本高的挑战,传统密集卷积网络常需下采样或分块推理。本文提出基于体素稀疏化与子流形稀疏卷积网络(SSCNs)的两阶段3D分割方法:第一阶段用低分辨率稀疏网络定位感兴趣区域(ROI);第二阶段在裁剪后的ROI内使用高分辨率稀疏网络进行精细化分割。该方法支持原生高分辨率处理,显著降低内存占用和推理时间。在KiTS23肾癌CT数据集上5折交叉验证,方法取得肾+肿块的骰子系数95.8%,肿瘤+囊肿85.7%,肿瘤单独80.3%,性能媲美顶尖方案。与同一交叉验证集上的分块nnU-Net基线相比,本方法在肿瘤+囊肿和肿瘤单独任务上表现相当或更优,且始终占用更少显存、推理更快。在测试的各类GPU上,本方法明显快于nnU-Net和零样本缩放模型SegVol(虽能准定位肾,但对小而异质病变表现差)。相较于同架构的稠密实现,本稀疏方法在测试的CPU与GPU配置下,推理时间最多减少60%,显存使用最多降低75%。

原文摘要 · Abstract (English)

Accurate delineation of kidney tumours in Computed Tomography (CT) is essential for downstream quantitative analysis and precision oncology, but manual segmentation is a specialised task, time-consuming and difficult to scale. Automated 3D segmentation remains challenging because CT scans are large volumetric images, making high-resolution dense convolutional networks computationally expensive and often dependent on downsampling or patch-based inference. We propose a two-stage 3D segmentation methodology based on voxel sparsification and submanifold sparse convolutional networks (SSCNs). Stage 1 uses a low-resolution sparse network to identify a region of interest (ROI); Stage 2 applies a high-resolution sparse network for refined segmentation within the cropped ROI. This enables native high-resolution 3D processing while reducing memory use and inference time. We evaluate the method on the KiTS23 renal cancer CT dataset using 5-fold cross-validation. Our method achieved Dice similarity coefficients of 95.8% for kidneys + masses, 85.7% for tumours + cysts, and 80.3% for tumours alone, competitive with top KiTS23 approaches. In direct comparisons on the same cross-validation folds, the proposed sparse method achieves tumour + cyst and tumour-only Dice scores comparable to, and slightly higher than, a patch-based nnU-Net baseline, while consistently requiring less VRAM and shorter inference time across the tested hardware. Across the tested GPUs, our sparse model is markedly faster than both nnU-Net and the zero-shot zoom-out/zoom-in foundation model SegVol, which localises kidneys well but underperforms on small heterogeneous lesions. Compared to an equivalent dense implementation of the same architecture, the proposed sparse approach achieves up to a 60% reduction in inference time and up to a 75% reduction in VRAM usage across both CPU and the GPU configurations tested.

3D分割稀疏卷积医学影像肾肿瘤

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