arXiv:2412.08345eess.IVcs.CV2024-12中稿 · AAAI被引 78

提升医学图像分割精度,解决模糊边界与共现干扰问题

ConDSeg: A General Medical Image Segmentation Framework via Contrast-Driven Feature Enhancement

  • 通过对比增强策略提升模型在低对比度环境下的特征提取能力
  • 分离前景、背景与不确定区域,逐步降低分割不确定性
  • 适合处理复杂医学图像,尤其对小目标和弱边界场景有效

医学图像分割在临床决策、治疗规划和疾病追踪中至关重要,但仍面临两大挑战:一是图像中前景与背景常存在‘软边界’,光照不足和低对比度进一步降低二者可区分性;二是医学图像中普遍存在共现现象,学习此类特征会误导模型判断。为此,我们提出通用框架ConDSeg。首先设计一致性强化对比训练策略,提升编码器在不同光照与对比度下的鲁棒性,确保在恶劣环境下仍能提取高质量特征。其次引入语义信息解耦模块,将编码器特征解耦为前景、背景与不确定区域,训练过程中逐步降低不确定性。对比驱动特征聚合模块则对比前景与背景特征,引导多层级特征融合与关键特征增强,进一步区分待分割实体。此外,提出尺寸感知解码器,解决解码器尺度单一问题,精准定位不同大小的目标,避免错误学习共现特征。在五个医学图像数据集、三种场景下的大量实验表明,该方法达到当前最优性能,验证了其先进性与广泛适用性。代码已开源:https://github.com/Mengqi-Lei/ConDSeg。

原文摘要 · Abstract (English)

Medical image segmentation plays an important role in clinical decision making, treatment planning, and disease tracking. However, it still faces two major challenges. On the one hand, there is often a ``soft boundary'' between foreground and background in medical images, with poor illumination and low contrast further reducing the distinguishability of foreground and background within the image. On the other hand, co-occurrence phenomena are widespread in medical images, and learning these features is misleading to the model's judgment. To address these challenges, we propose a general framework called Contrast-Driven Medical Image Segmentation (ConDSeg). First, we develop a contrastive training strategy called Consistency Reinforcement. It is designed to improve the encoder's robustness in various illumination and contrast scenarios, enabling the model to extract high-quality features even in adverse environments. Second, we introduce a Semantic Information Decoupling module, which is able to decouple features from the encoder into foreground, background, and uncertainty regions, gradually acquiring the ability to reduce uncertainty during training. The Contrast-Driven Feature Aggregation module then contrasts the foreground and background features to guide multi-level feature fusion and key feature enhancement, further distinguishing the entities to be segmented. We also propose a Size-Aware Decoder to solve the scale singularity of the decoder. It accurately locate entities of different sizes in the image, thus avoiding erroneous learning of co-occurrence features. Extensive experiments on five medical image datasets across three scenarios demonstrate the state-of-the-art performance of our method, proving its advanced nature and general applicability to various medical image segmentation scenarios. Our released code is available at \url{https://github.com/Mengqi-Lei/ConDSeg}.

医学图像分割对比学习特征增强

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