用临床知识优化标签平滑,提升小肝肿瘤分割精度。
SmoothSegNet: A Global-Local Framework for Liver Tumor Segmentation with Clinical KnowledgeInformed Label Smoothing
- 结合临床数据设计标签平滑,缓解过拟合。
- 全局-局部框架分解任务,提升小肿瘤分割效果。
- 专为肝肿瘤设计前后处理流程,边界更清晰。
肝癌是全球主要致死病因之一,基于CT的肿瘤精准分割对诊疗至关重要。人工勾画耗时且易变,亟需可靠自动化方法。尽管深度学习在肝脏分割中表现良好,但肿瘤异质性强、边界模糊、标注数据有限,导致精确分割仍具挑战。本文提出SmoothSegNet,通过三大设计应对:(1)基于临床数据的知识引导标签平滑,生成平滑标签以正则化训练,降低过拟合风险;(2)全局-局部分割框架,将主任务拆解为两个子任务,分别优化预处理与训练;(3)针对各子任务定制前后处理流程,增强肿瘤可见性并精修边界。在具有挑战性的HCC-TACE-Seg数据集上验证,SmoothSegNet在小肿瘤(<10cm)分割上优于多种基准模型。消融实验证明三者协同提升性能。代码已开源:https://github.com/lingchm/medassist-liver-cancer。
原文摘要 · Abstract (English)
Liver cancer is a leading cause of mortality worldwide, and accurate Computed Tomography (CT)-based tumor segmentation is essential for diagnosis and treatment. Manual delineation is time-intensive, prone to variability, and highlights the need for reliable automation. While deep learning has shown promise for automated liver segmentation, precise liver tumor segmentation remains challenging due to the heterogeneous nature of tumors, imprecise tumor margins, and limited labeled data. We present SmoothSegNet, a novel deep learning framework that addresses these challenges with the three key designs: (1) A novel knowledge-informed label smoothing technique that distills knowledge from clinical data to generate smooth labels, which are used to regularize model training, reducing the overfitting risk and enhancing model performance; (2) A global and local segmentation framework that breaks down the main task into two simpler sub-tasks, allowing optimized preprocessing and training for each; and (3) Pre- and post-processing pipelines customized to the challenges of each subtask aimed to enhance tumor visibility and refines tumor boundaries. We apply the proposed model on a challenging HCC-TACE-Seg dataset and show that SmoothSegNet outperformed various benchmarks in segmentation performance, particularly at smaller tumors (<10cm). Our ablation studies show that the three design components complementarily contribute to the model improved performance. Code for the proposed method are available at https://github.com/lingchm/medassist-liver-cancer.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。