Medical image segmentation is critical for computer-aided medical diagnosis, both convolutional neural networks and the Transformer model have received widespread attention. However, convolutional Neural Networks and Transformers are each limited by global information and computational complexity. Recent studies have shown that the state space model (SSM) represented by Vision Mamba is able to maintain linear computational complexity and to simulate long-range interactions. Inspired by this, we propose the High-order Morphological Method Vision Mamba (HMM-VMamba) in this paper to further enhance the focus on lesion regions. Among its components, the proposed High-order Morphological Method Selective-Scan-2D (HMM-SS2D) exploits high-order information channel interactions to explore morphological knowledge. Moreover, the combination of Local Selective-Scan-2D, External Multi-scale Attention Mechanism, and Convolutional Neural Network improves the capability of local boundary feature learning. We evaluated comparison and ablation experiments on three open-source medical image datasets, and the results demonstrate that HMM-VMamba exhibits strong learning capabilities in medical image segmentation. Github

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

HMM-VMamba: High-Order Morphological Method Vision Mamba for Medical Image Segmentation

  • Yifeng Yao,
  • Bei He,
  • Minsheng Tan,
  • Xiang Li,
  • Zhenzhen Hu,
  • Xingxing Duan,
  • Lingna Chen

摘要

Medical image segmentation is critical for computer-aided medical diagnosis, both convolutional neural networks and the Transformer model have received widespread attention. However, convolutional Neural Networks and Transformers are each limited by global information and computational complexity. Recent studies have shown that the state space model (SSM) represented by Vision Mamba is able to maintain linear computational complexity and to simulate long-range interactions. Inspired by this, we propose the High-order Morphological Method Vision Mamba (HMM-VMamba) in this paper to further enhance the focus on lesion regions. Among its components, the proposed High-order Morphological Method Selective-Scan-2D (HMM-SS2D) exploits high-order information channel interactions to explore morphological knowledge. Moreover, the combination of Local Selective-Scan-2D, External Multi-scale Attention Mechanism, and Convolutional Neural Network improves the capability of local boundary feature learning. We evaluated comparison and ablation experiments on three open-source medical image datasets, and the results demonstrate that HMM-VMamba exhibits strong learning capabilities in medical image segmentation. Github