MedFormer用分层结构与智能稀疏注意力,提升医疗影像识别效率和泛化能力。
MedFormer: Hierarchical Medical Vision Transformer with Content-Aware Dual Sparse Selection Attention
- 采用分层金字塔结构,适配多种医疗图像任务。
- 引入内容感知双稀疏注意力,计算量降低且抗噪性强。
- 在分类、分割、病灶检测上均表现优异,适合临床部署。
医学图像识别是辅助临床诊断的关键手段,有助于更准确及时地发现疾病与异常。基于视觉变换器的方法在各类医学识别任务中表现良好,但面临两大挑战:一是通常任务特定、架构定制,通用性差;二是要么采用全注意力建模长程依赖,导致计算开销大,要么依赖人工设计的稀疏注意力,可能性能不佳。为此,我们提出MedFormer,一种高效且通用的医学视觉变换器。其核心思想包括:第一,采用金字塔缩放结构作为通用骨干网络,适用于图像分类及语义分割、病灶检测等密集预测任务,实现分层特征表示的同时降低特征图计算负担;第二,提出新型内容感知双稀疏选择注意力(DSSA),在保持高性能的同时提升计算效率与抗噪能力。理论分析表明,MedFormer在通用性与效率方面优于现有医学视觉变换器。在多种成像模态数据集上的大量实验显示,其在上述三类医学图像识别任务中持续提升性能。MedFormer为医学图像识别提供了高效且通用的解决方案,具备良好的临床应用潜力。
原文摘要 · Abstract (English)
Medical image recognition serves as a key way to aid in clinical diagnosis, enabling more accurate and timely identification of diseases and abnormalities. Vision transformer-based approaches have proven effective in handling various medical recognition tasks. However, these methods encounter two primary challenges. First, they are often task-specific and architecture-tailored, limiting their general applicability. Second, they usually either adopt full attention to model long-range dependencies, resulting in high computational costs, or rely on handcrafted sparse attention, potentially leading to suboptimal performance. To tackle these issues, we present MedFormer, an efficient medical vision transformer with two key ideas. First, it employs a pyramid scaling structure as a versatile backbone for various medical image recognition tasks, including image classification and dense prediction tasks such as semantic segmentation and lesion detection. This structure facilitates hierarchical feature representation while reducing the computation load of feature maps, highly beneficial for boosting performance. Second, it introduces a novel Dual Sparse Selection Attention (DSSA) with content awareness to improve computational efficiency and robustness against noise while maintaining high performance. As the core building technique of MedFormer, DSSA is designed to explicitly attend to the most relevant content. Theoretical analysis demonstrates that MedFormer outperforms existing medical vision transformers in terms of generality and efficiency. Extensive experiments across various imaging modality datasets show that MedFormer consistently enhances performance in all three medical image recognition tasks mentioned above. MedFormer provides an efficient and versatile solution for medical image recognition, with strong potential for clinical application.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。