arXiv:2410.12419eess.IVcs.CV2024-10

用注意力引导的扰动策略提升医学图像分割的弱-强一致性效果

Mind the Context: Attention-Guided Weak-to-Strong Consistency for Enhanced Semi-Supervised Medical Image Segmentation

  • 通过图像和特征级注意力引导扰动,增强模型对复杂语义的捕捉能力
  • 在ACDC数据集7病例场景下达到90.4%的Dice分数,优于现有方法
  • 适合临床医学图像分割任务,尤其适用于标注数据稀缺场景

医学图像分割是诊断与治疗的关键步骤,但高质量标注数据获取困难且成本高。半监督学习可通过利用未标注数据提升模型性能。尽管弱-强一致性是主流方法,针对医学图像分割的扰动策略研究仍不足。本文提出一种名为注意力引导弱-强一致性匹配(AIGCMatch)的高效半监督框架,在图像和特征层面引入注意力引导的扰动策略,实现弱-强一致性正则化。该方法既保留医学图像结构信息,又增强模型处理复杂语义的能力。在ACDC和ISIC-2017数据集上的实验验证了其有效性。AIGCMatch在ACDC数据集7病例场景下达到90.4%的Dice分数,超越当前最优方法,展现出在临床应用中的潜力。

原文摘要 · Abstract (English)

Medical image segmentation is a pivotal step in diagnostic and therapeutic processes, relying on high-quality annotated data that is often challenging and costly to obtain. Semi-supervised learning offers a promising approach to enhance model performance by leveraging unlabeled data. Although weak-to-strong consistency is a prevalent method in semi-supervised image segmentation, there is a scarcity of research on perturbation strategies specifically tailored for semi-supervised medical image segmentation tasks. To address this challenge, this paper introduces a simple yet efficient semi-supervised learning framework named Attention-Guided weak-to-strong Consistency Match (AIGCMatch). The AIGCMatch framework incorporates attention-guided perturbation strategies at both the image and feature levels to achieve weak-to-strong consistency regularization. This method not only preserves the structural information of medical images but also enhances the model's ability to process complex semantic information. Extensive experiments conducted on the ACDC and ISIC-2017 datasets have validated the effectiveness of AIGCMatch. Our method achieved a 90.4\% Dice score in the 7-case scenario on the ACDC dataset, surpassing the state-of-the-art methods and demonstrating its potential and efficacy in clinical settings.

医学图像分割半监督学习注意力机制弱-强一致性

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