基于Mamba的多尺度网络提升心脏超声图像分割精度
MSV-Mamba: A Multiscale Vision Mamba Network for Echocardiography Segmentation
- 引入大窗口Mamba模块捕捉全局结构依赖
- 多层注意力融合提升复杂解剖结构分割准确率
- 在两个数据集上显著优于现有方法,适合医学影像分析
超声成像常受噪声高、时空分辨率低及解剖结构复杂影响,导致模型难以准确捕捉心脏各区域的结构关系与动态模式。本文提出一种U型深度学习模型MSV-Mamba,结合大窗口Mamba模块(LMS)与分层特征融合策略,用于心脏超声图像分割。首先,级联残差块作为编码器,逐步提取多尺度细节特征;其次,在解码器中引入大窗口多尺度Mamba模块,增强对跨区域全局依赖的建模能力;同时,在每层解码器添加辅助损失,并采用双注意力机制实现空间与通道维度的多层特征融合。实验在EchoNet-Dynamic和CAMUS数据集上验证,对左心室心内膜(LV_endo)分割分别达到95.01和93.36的评分,对左心室外膜(LV_epi)分别为87.35和87.80,相较最优基线提升0.54至1.11,展现出更高精度与鲁棒性。
原文摘要 · Abstract (English)
Ultrasound imaging frequently encounters challenges, such as those related to elevated noise levels, diminished spatiotemporal resolution, and the complexity of anatomical structures. These factors significantly hinder the model's ability to accurately capture and analyze structural relationships and dynamic patterns across various regions of the heart. Mamba, an emerging model, is one of the most cutting-edge approaches that is widely applied to diverse vision and language tasks. To this end, this paper introduces a U-shaped deep learning model incorporating a large-window Mamba scale (LMS) module and a hierarchical feature fusion approach for echocardiographic segmentation. First, a cascaded residual block serves as an encoder and is employed to incrementally extract multiscale detailed features. Second, a large-window multiscale mamba module is integrated into the decoder to capture global dependencies across regions and enhance the segmentation capability for complex anatomical structures. Furthermore, our model introduces auxiliary losses at each decoder layer and employs a dual attention mechanism to fuse multilayer features both spatially and across channels. This approach enhances segmentation performance and accuracy in delineating complex anatomical structures. Finally, the experimental results using the EchoNet-Dynamic and CAMUS datasets demonstrate that the model outperforms other methods in terms of both accuracy and robustness. For the segmentation of the left ventricular endocardium (${LV}_{endo}$), the model achieved optimal values of 95.01 and 93.36, respectively, while for the left ventricular epicardium (${LV}_{epi}$), values of 87.35 and 87.80, respectively, were achieved. This represents an improvement ranging between 0.54 and 1.11 compared with the best-performing model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。