arXiv:2412.02314cs.CV2024-12被引 1

通过增强低对比度特征,提升内窥镜图像分割精度。

Low-Contrast-Enhanced Contrastive Learning for Semi-Supervised Endoscopic Image Segmentation

  • 设计低对比度增强对比学习,突出病灶与正常组织差异。
  • 在两个公开数据集和三年积累的私有数据上均达领先性能。
  • 适合标注稀缺、病灶对比度低的医疗图像分割任务。

内窥镜图像分割在计算机辅助诊断与治疗中至关重要。尽管深度学习推动了多种模型在内窥镜肿瘤分割中的应用并取得良好效果,但受限于标注数据少和图像对比度低,精准分割仍具挑战。为此,本文提出一种新颖的半监督分割框架LoCo,基于低对比度增强对比学习(LCC),有效利用大量未标注数据,提升分割准确性和鲁棒性。LCC引入两种策略:类间对比增强(ICE)和边界对比增强(BCE),强化恶性、良性肿瘤与正常组织间低对比度像素的区分能力。此外,设计置信度动态滤波器(CDF)进行伪标签选择,特别关注少数类别,提升伪标签利用率。在两个公开数据集及三年积累的大规模私有数据集上进行的大量实验表明,LoCo性能显著优于现有方法。代码已开源:https://github.com/AnoK3111/LoCo。

原文摘要 · Abstract (English)

The segmentation of endoscopic images plays a vital role in computer-aided diagnosis and treatment. The advancements in deep learning have led to the employment of numerous models for endoscopic tumor segmentation, achieving promising segmentation performance. Despite recent advancements, precise segmentation remains challenging due to limited annotations and the issue of low contrast. To address these issues, we propose a novel semi-supervised segmentation framework termed LoCo via low-contrast-enhanced contrastive learning (LCC). This innovative approach effectively harnesses the vast amounts of unlabeled data available for endoscopic image segmentation, improving both accuracy and robustness in the segmentation process. Specifically, LCC incorporates two advanced strategies to enhance the distinctiveness of low-contrast pixels: inter-class contrast enhancement (ICE) and boundary contrast enhancement (BCE), enabling models to segment low-contrast pixels among malignant tumors, benign tumors, and normal tissues. Additionally, a confidence-based dynamic filter (CDF) is designed for pseudo-label selection, enhancing the utilization of generated pseudo-labels for unlabeled data with a specific focus on minority classes. Extensive experiments conducted on two public datasets, as well as a large proprietary dataset collected over three years, demonstrate that LoCo achieves state-of-the-art results, significantly outperforming previous methods. The source code of LoCo is available at the URL of \href{https://github.com/AnoK3111/LoCo}{https://github.com/AnoK3111/LoCo}.

医学图像半监督分割对比学习

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