ECM-TransUNet: Edge-enhanced multi-scale attention and convolutional Mamba for medical image segmentation

被引:0
|
作者
Lv, Chunjie [1 ]
Li, Biyuan [1 ,2 ]
Wang, Xiuwei [1 ]
Cai, Pengfei [1 ]
Yang, Bo [1 ]
Sun, Gaowei [1 ]
Yan, Jun [3 ]
机构
[1] Tianjin Univ Technol & Educ, Sch Elect Engn, Tianjin 300222, Peoples R China
[2] Tianjin Dev Zone Jingnuohanhai Data Technol Co Ltd, Tianjin, Peoples R China
[3] Tianjin Univ, Sch Math, Tianjin 300072, Peoples R China
基金
中国国家自然科学基金;
关键词
Convolutional Mamba; Novel Attention Mechanism; Edge Feature Extraction; Spatial Consistency; Medical Image Segmentation; NET; NETWORK;
D O I
10.1016/j.bspc.2025.107845
中图分类号
R318 [生物医学工程];
学科分类号
0831 ;
摘要
The segmentation of CT and MRI images faces challenges such as detail loss and the inability to ensure consistency in physiological tissue representation. To address these issues, we propose a Edge-enhanced multi-scale attention and Convolutional Mamba Transformer UNet (ECM-TransUNet). ECM-TransUNet integrates the ECMBlock into the skip connections, incorporating the Edge-Enhanced Multi-Scale Transposed Attention (E-MTA) and the Multi-Scale Convolutional State-Space Module (MS-CSM) to improve feature extraction and spatial consistency modeling. Specifically, E-MTA enhances sensitivity to subtle grayscale variations, enabling accurate modeling of both local and global structural details in complex regions. Unlike traditional attention mechanisms, E-MTA integrates multi-scale depthwise convolutions to strengthen local feature representation, while the Sobel edge detection module further refines the extraction of critical edges and local detail features. MS-CSM combines state-space modeling with multi-scale feature extraction to improve the accuracy of local detail representation and global feature integration, while significantly reducing computational complexity. Compared to traditional convolution-based methods and earlier state-space models, it demonstrates superior performance and efficiency. Additionally, to achieve end-to-end feature balance within skip connections, we introduce the Cross-Region Multi-Scale Attention (CR-MSA) mechanism into the Transformer-based encoder architecture. CR-MSA effectively harmonizes multi-scale and spatial feature fusion, establishes cross-regional feature relationships, and enhances the model's ability to capture both local and global information, thereby further improving segmentation accuracy and stability. Our method effectively addresses the limitations of existing medical image segmentation techniques. Experimental results on large-scale annotated CT and MRI datasets demonstrate that our approach achieves an optimal balance between segmentation accuracy and computational efficiency. Specifically, on the Synapse dataset, ECM-TransUNet achieved a DSC of 84.68 %, with a computational cost of 50.68G FLOPs and a parameter count of 66.47 M. These findings underscore the reliability and efficiency of our method, offering a robust solution for complex medical image segmentation tasks. is available at: https://github.com/ lvchunjie/ECM-TransUNet.
引用
收藏
页数:17
相关论文
共 50 条
  • [1] Selective and multi-scale fusion Mamba for medical image segmentation
    Li, Guangju
    Huang, Qinghua
    Wang, Wei
    Liu, Longzhong
    EXPERT SYSTEMS WITH APPLICATIONS, 2025, 261
  • [2] Multi-scale convolutional attention frequency-enhanced transformer network for medical image segmentation
    Yan, Shun
    Yang, Benquan
    Chen, Aihua
    Zhao, Xiaoming
    Zhang, Shiqing
    INFORMATION FUSION, 2025, 119
  • [3] MAGNet: A Convolutional Neural Network with Multi-Scale and Global Attention Modules for Medical Image Segmentation
    Bharati, Subrato
    Ahmad, M. Omair
    Swamy, M. N. S.
    2024 IEEE INTERNATIONAL SYMPOSIUM ON CIRCUITS AND SYSTEMS, ISCAS 2024, 2024,
  • [4] MAXFormer: Enhanced transformer for medical image segmentation with multi-attention and multi-scale features fusion
    Liang, Zhiwei
    Zhao, Kui
    Liang, Gang
    Li, Siyu
    Wu, Yifei
    Zhou, Yiping
    KNOWLEDGE-BASED SYSTEMS, 2023, 280
  • [5] MedSegNet: A Lightweight Convolutional Network Combining Dual Self-Attention and Multi-Scale Attention for Medical Image Segmentation
    Bharati, Subrato
    Ahmad, M. Omair
    Swamy, M. N. S.
    2024 IEEE 67TH INTERNATIONAL MIDWEST SYMPOSIUM ON CIRCUITS AND SYSTEMS, MWSCAS 2024, 2024, : 965 - 969
  • [6] Medical Image Segmentation Use Convolutional Attention Augmentation TransUNet with Skip Connection Enhancement
    Zou, Qingzhi
    Zhao, Jing
    Li, Ming
    Chen, Ling
    Yuan, Lin
    Zhang, Ronghuan
    Hu, Yushuai
    PROCEEDINGS OF THE 2024 27 TH INTERNATIONAL CONFERENCE ON COMPUTER SUPPORTED COOPERATIVE WORK IN DESIGN, CSCWD 2024, 2024, : 1710 - 1715
  • [7] MSDANet: A multi-scale dilation attention network for medical image segmentation
    Zhang, Jinquan
    Luan, Zhuang
    Ni, Lina
    Qi, Liang
    Gong, Xu
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2024, 90
  • [8] Multi-Scale Self-Guided Attention for Medical Image Segmentation
    Sinha, Ashish
    Dolz, Jose
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2021, 25 (01) : 121 - 130
  • [9] MSAANet: Multi-scale Axial Attention Network for medical image segmentation
    Zeng, Hao
    Shan, Xinxin
    Feng, Yu
    Wen, Ying
    2023 IEEE INTERNATIONAL CONFERENCE ON MULTIMEDIA AND EXPO, ICME, 2023, : 2291 - 2296
  • [10] A Multi-Scale Context Aware Attention Model for Medical Image Segmentation
    Alam, Md. Shariful
    Wang, Dadong
    Liao, Qiyu
    Sowmya, Arcot
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2023, 27 (08) : 3731 - 3739