用少量反馈实现医学图像分割的高效优化,降低标注成本。
Resource-efficient Automatic Refinement of Segmentations via Weak Supervision from Light Feedback
- 基于区域质量评分和过/欠分割标签设计弱监督损失函数
- 在肱骨CT数据上提升TotalSegmentator初始分割效果,性能相当
- 仅需轻量级用户反馈,大幅减少标注时间和监督需求
医学图像中解剖结构的精确勾画是关键任务。人工分割虽准确但耗时且易变异,促使自动化方法发展。近年来,各类基础模型已实现多种解剖结构与成像模态的自动分割,但未必满足临床精度标准。现有分割优化方法依赖大量交互或全监督标注。本文提出SCORE(基于区域评估的分割修正),一种仅需训练时轻量反馈的弱监督框架。SCORE引入新型损失函数,利用区域质量评分及过/欠分割错误标签,而非密集标注。我们在肱骨CT数据上验证其效果,显著改进TotalSegmentator的初始预测,性能达到现有方法水平,同时极大降低监督要求与标注时间。代码已开源:https://gitlab.inria.fr/adelangl/SCORE。
原文摘要 · Abstract (English)
Delineating anatomical regions is a key task in medical image analysis. Manual segmentation achieves high accuracy but is labor-intensive and prone to variability, thus prompting the development of automated approaches. Recently, a breadth of foundation models has enabled automated segmentations across diverse anatomies and imaging modalities, but these may not always meet the clinical accuracy standards. While segmentation refinement strategies can improve performance, current methods depend on heavy user interactions or require fully supervised segmentations for training. Here, we present SCORE (Segmentation COrrection from Regional Evaluations), a weakly supervised framework that learns to refine mask predictions only using light feedback during training. Specifically, instead of relying on dense training image annotations, SCORE introduces a novel loss that leverages region-wise quality scores and over/under-segmentation error labels. We demonstrate SCORE on humerus CT scans, where it considerably improves initial predictions from TotalSegmentator, and achieves performance on par with existing refinement methods, while greatly reducing their supervision requirements and annotation time. Our code is available at: https://gitlab.inria.fr/adelangl/SCORE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。