用轻量结构提升显微图像分割精度,兼顾速度与细节
InceptionMamba: Efficient Multi-Stage Feature Enhancement with Selective State Space Model for Microscopic Medical Image Segmentation
- 融合多阶段特征与选择性状态空间模型,增强细胞边界感知
- 在4个数据集上达顶尖性能,计算量降低约5倍
- 适合医疗图像分割场景,尤其适用于小样本与复杂背景
精准的显微医学图像分割对癌症细胞诊断和肿瘤识别至关重要。尽管深度学习推动了卷积神经网络(CNN)与基于Transformer的模型发展,使其具备更广的感受野,但在背景杂乱、目标重叠等挑战性场景下仍难以捕捉复杂的细胞与组织结构。此外,这些方法通常依赖大规模数据集以提升性能,且计算成本高,限制了实际应用。为此,我们提出一种名为InceptionMamba的高效分割框架,通过编码多阶段丰富特征,在保持高性能的同时实现计算效率提升。具体地,利用语义线索捕获低频与高频区域,增强多阶段特征以应对模糊边界(如细胞边界)。这些增强特征输入由扩张深度卷积与Mamba块组成的混合模型,有效保留高效性并捕捉兴趣区域在尺度与形状上的内在变化。最终,将增强特征与低分辨率特征融合生成分割掩码。该模型在两个挑战性显微分割数据集(SegPC21和GlaS)以及两个皮肤病变分割数据集(ISIC2017和ISIC2018)上均达到当前最优性能,同时相比此前最佳方法减少约5倍的计算成本。
原文摘要 · Abstract (English)
Accurate microscopic medical image segmentation plays a crucial role in diagnosing various cancerous cells and identifying tumors. Driven by advancements in deep learning, convolutional neural networks (CNNs) and transformer-based models have been extensively studied to enhance receptive fields and improve medical image segmentation task. However, they often struggle to capture complex cellular and tissue structures in challenging scenarios such as background clutter and object overlap. Moreover, their reliance on the availability of large datasets for improved performance, along with the high computational cost, limit their practicality. To address these issues, we propose an efficient framework for the segmentation task, named InceptionMamba, which encodes multi-stage rich features and offers both performance and computational efficiency. Specifically, we exploit semantic cues to capture both low-frequency and high-frequency regions to enrich the multi-stage features to handle the blurred region boundaries (e.g., cell boundaries). These enriched features are input to a hybrid model that combines an Inception depth-wise convolution with a Mamba block, to maintain high efficiency and capture inherent variations in the scales and shapes of the regions of interest. These enriched features along with low-resolution features are fused to get the final segmentation mask. Our model achieves state-of-the-art performance on two challenging microscopic segmentation datasets (SegPC21 and GlaS) and two skin lesion segmentation datasets (ISIC2017 and ISIC2018), while reducing computational cost by about 5 times compared to the previous best performing method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。