arXiv:2505.06217cs.CV2025-05被引 1

用分割模型提升医学图像分类,兼顾精度与数据效率

Adapting a Segmentation Foundation Model for Medical Image Classification

  • 冻结分割模型编码器提取空间特征,避免训练开销
  • 设计局部注意力机制,增强模型对关键区域的关注
  • 在3个公开数据集上验证有效,适合小样本医学场景

近年来,以通用分割模型(如SAM)为代表的基础模型在图像分割等视觉任务中表现出色,尤其具备出色的零样本分割能力。然而,将此类模型有效适配于医学图像分类仍属研究空白。本文提出一种新框架,利用SAM的图像编码器作为特征提取器,捕获蕴含重要空间与上下文信息的分割特征,并冻结其权重以减少训练开销。进一步提出一种新型空间局部通道注意力机制(SLCA),对特征图计算空间局部注意力权重,将其融入深度学习分类模型,提升模型对图像中关键区域的关注度,从而改善分类性能。在三个公开医学图像分类数据集上的实验表明,该方法在提升准确率的同时具有良好的数据效率。

原文摘要 · Abstract (English)

Recent advancements in foundation models, such as the Segment Anything Model (SAM), have shown strong performance in various vision tasks, particularly image segmentation, due to their impressive zero-shot segmentation capabilities. However, effectively adapting such models for medical image classification is still a less explored topic. In this paper, we introduce a new framework to adapt SAM for medical image classification. First, we utilize the SAM image encoder as a feature extractor to capture segmentation-based features that convey important spatial and contextual details of the image, while freezing its weights to avoid unnecessary overhead during training. Next, we propose a novel Spatially Localized Channel Attention (SLCA) mechanism to compute spatially localized attention weights for the feature maps. The features extracted from SAM's image encoder are processed through SLCA to compute attention weights, which are then integrated into deep learning classification models to enhance their focus on spatially relevant or meaningful regions of the image, thus improving classification performance. Experimental results on three public medical image classification datasets demonstrate the effectiveness and data-efficiency of our approach.

医学图像分割模型注意力机制小样本学习

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