Cascaded mutual enhancing networks for brain tumor subregion segmentation in multiparametric MRI

Abstract

Accurate segmentation of glioma and its subregions plays an important role in radiotherapy treatment planning. Due to a very populated multiparameter magnetic resonance imaging image, manual segmentation tasks can be very time-consuming, meticulous, and prone to subjective errors. Here, we propose a novel deep learning framework based on mutual enhancing networks to automatically segment brain tumor subregions. The proposed framework is suitable for the segmentation of brain tumor subregions owing to the contribution of Retina U-Net followed by the implementation of a mutual enhancing strategy between the classification localization map (CLM) module and segmentation module. Retina U-Net is trained to accurately identify view-of-interest and feature maps of the whole tumor (WT), which are then transferred to the CLM module and segmentation module. Subsequently, CLM generated by the CLM module is integrated with the segmentation module to bring forth a mutual enhancing strategy. In this way, our proposed framework first focuses on WT through Retina U-Net, and since WT consists of subregions, a mutual enhancing strategy then further aims to classify and segment subregions embedded within WT. We implemented and evaluated our proposed framework on the BraTS 2020 dataset consisting of 369 cases. We performed a 5-fold cross-validation on 200 datasets and a hold-out test on the remaining 169 cases. To demonstrate the effectiveness of our network design, we compared our method against the networks without Retina U-Net, mutual enhancing strategy, and a recently published Cascaded U-Net architecture. Results of all four methods were compared to the ground truth for segmentation and localization accuracies. Our method yielded significantly (P < 0.01) better values of dice-similarity-coefficient, center-of-mass-distance, and volume difference compared to all three competing methods across all tumor labels (necrosis and non-enhancing, edema, enhancing tumor, WT, tumor core) on both validation and hold-out dataset. Overall quantitative and statistical results of this work demonstrate the ability of our method to both accurately and automatically segment brain tumor subregions.