用蒙特卡洛一致性增强SAM,少标注也能精准分割前列腺区域。
MCICSAM: Monte Carlo-guided Interpolation Consistency Segment Anything Model for Semi-Supervised Prostate Zone Segmentation
- 通过蒙特卡洛插值一致性约束,提升未标注数据利用效率。
- 在前列腺过渡区分割上达到89.95% Dice和2.27mm HD95。
- 适合标注数据稀缺的医学影像分割任务,通用性强。
准确分割前列腺各区域对诊断和治疗前列腺疾病至关重要。然而,专业医学领域如前列腺影像中标签数据稀缺,构成重大挑战。尽管分割一切模型(SAM)在自然图像分割中表现优异,但在医学影像中仍存在应用难题。为更好利用SAM强大的特征提取能力并缓解医学图像标注数据量不足的问题,本文采用低秩适配(LoRA)与蒙特卡洛引导插值一致性(MCIC)的半监督学习方法,对SAM进行优化,提出蒙特卡洛引导插值一致性分割一切模型(MCICSAM),用于半监督前列腺区域分割。在无标签数据部分,MCIC对输入数据施加两种不同插值变换,并在输出中引入蒙特卡洛不确定性分析,强制模型预测保持一致性。这种一致性约束使模型更贴合无标签数据分布,从而提升半监督场景下的性能。采用Dice系数和95%分位数豪斯多夫距离(HD95)评估模型表现。实验结果显示,MCICSAM在过渡区与外周区分割上的Dice分别达到79.38%和89.95%,HD95分别为3.12和2.27。同时,该方法展现出优异的泛化能力。本方法有望为前列腺影像分割领域带来新可能。
原文摘要 · Abstract (English)
Accurate segmentation of various regions within the prostate is pivotal for diagnosing and treating prostate-related diseases. However, the scarcity of labeled data, particularly in specialized medical fields like prostate imaging, poses a significant challenge. Segment Anything Model (SAM) is a new large model for natural image segmentation, but there are some challenges in medical imaging. In order to better utilize the powerful feature extraction capability of SAM as well as to address the problem of low data volume for medical image annotation, we use Low-Rank Adaptation (LoRA) and semi-supervised learning methods of Monte Carlo guided interpolation consistency (MCIC) to enhance the fine-tuned SAM. We propose Monte Carlo-guided Interpolation Consistency Segment Anything Model (MCICSAM) for application to semi-supervised learning based prostate region segmentation. In the unlabeled data section, MCIC performs two different interpolation transformations on the input data and incorporates Monte Carlo uncertainty analysis in the output, forcing the model to be consistent in its predictions. The consistency constraints imposed on these interpolated samples allow the model to fit the distribution of unlabeled data better, ultimately improving its performance in semi-supervised scenarios. We use Dice and Hausdorff Distance at 95th percentile (HD95) to validate model performance. MCICSAM yieldes Dice with 79.38% and 89.95%, along with improves HD95 values of 3.12 and 2.27 for transition zone and transition zone. At the same time MCICSAM demonstrates strong generalizability. This method is expected to bring new possibilities in the field of prostate image segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。