Ensemble Outperforms Single Models in Brain Tumor Segmentation

Jianxun Ren,Wei Zhang,Ning An,Qingyu Hu,Youjia Zhang,Ying Zhou
DOI: https://doi.org/10.1007/978-3-031-08999-2_39
2021-01-01
Abstract:Brain Tumor Segmentation (BraTS) and Federated Tumor Segmentation (FeTS) are open and popular challenges, for which countless medical image segmentation models have been proposed. Based on the platform that BraTS challenge 2021 provided for researchers, we implemented a battery of cutting-edge deep neural networks, such as nnU-Net, UNet++, CoTr, HRNet, and Swin-Unet to compare performances amongst distinct models directly. To improve segmentation accuracy, we first tried several modification techniques (e.g., data augmentation, region-based training, batch-dice loss function, etc.). Next, the outputs from the five best models were averaged using a final ensemble model, of which four models in the committee were organized in different architectures. As a result, the strengths of every single model were amplified by aggregation. Our model took one of the best performing places in the BraTS 2021 competition amongst over 1200 excellent researchers from all over the world, and robustly produced outstanding segmentation results across different unseen datasets from various institutions in the FeTS 2022 Challenge, which achieved Dice score of 0.9256, 0.8774, 0.8576 and Hausdorff Distances (95%) of 4.36, 14.80, 14.49 for whole tumor, tumor core, and enhancing tumor respectively. Since our model is not successfully evaluated on the test data due to a bug, all the scores stated in this manuscript are based on the validation dataset.
What problem does this paper attempt to address?