🤖 AI Summary
This work addresses the limitations of convolutional neural networks in capturing long-range dependencies and the high computational complexity of Transformers in 3D brain tumor segmentation by proposing a U-Net-based multi-head State Space Model (Mamba) approach. The method parallelizes SSM heads along the channel dimension and integrates them via residual aggregation, augmented with a channel-spatial calibration module and an adaptive skip connection fusion mechanism. This design maintains linear computational complexity while enhancing multi-modal training stability, boundary consistency, and sensitivity to small lesions. Evaluated on the BraTS2021 and BraTS2023 datasets, the proposed model achieves substantial improvements in overall segmentation accuracy, particularly excelling in delineating the tumor core and detecting subtle enhancing regions.
📝 Abstract
Brain tumors exhibit high heterogeneity in morphology and multimodal contrast, making manual slice-by-slice de lineation time-consuming and experience-dependent, thus necessitating efficient and stable automated segmentation methods. To address the limitations of CNNs in modeling long-range dependencies, and the heavy computational and memory overhead and inter-block contextual in coherence of Transformers in 3D MRI, this paper proposes Multi-Head Mamba (MHMamba). This method combines a U-shaped architecture with a multi-head state-space model (Mamba), splitting the channel dimension into parallel SSM heads and aggregating them with residuals. This enhances long-range representation and improves the stability of multimodal training while maintaining linear complexity. To further align statistics and enhance lesion response, we designed a channel-space calibration module for multi-head outputs and introduced an adaptive fusion mechanism at skip connections to dynamically connect global semantics with local details, thereby improving boundary consistency and the detection of small-volume lesions. We conducted experiments and ablations on BraTS2021 and BraTS2023. The results showed that MHMamba achieved stable and significant improvements in overall accuracy, boundary smoothness, and sensitivity to tumor core and small-volume enhancement areas, while preserving the linear-complexity advantage of Mamba-based modeling, thus verifying the effectiveness and versatility of the method.