Ensembling CNN, Transformer, and Mamba with Stacking for Brain Tumor Segmentation
摘要
In this study, we explore the effectiveness of a stacking ensemble approach for brain tumor segmentation by combining three distinct deep neural network architectures: nnU-Net (based on convolutional neural network, SwinUNETR (based on transformer), and LoG-VMamba (based on VMamba). Each model offers unique representational strengths that, when integrated through ensemble learning, enhance overall segmentation performance. To enable robust fusion, our model utilizes two core components: (1) label noise modeling through signed soft labels and (2) stacking-based model fusion. All models, including ensembles, are evaluated on two tasks of the BraTS 2025 challenge: Task 1 (pre- and post-treatment glioma in adults) and Task 5 (glioma segmentation within a Sub-Saharan African patient population). Our results show that SwinUNETR consistently performs below nnU-Net and LoG-VMamba on both tasks. The ensemble method surpasses all individual models according to both regional and boundary-aware metrics, demonstrating improved accuracy and robustness. Our findings underscore the value of architectural diversity and ensemble learning in advancing medical image segmentation.