A data pre-processing with mass-preserving optimal mass transportation for brain tumor segmentation

TSUNG-MING HUANG,JIA-WEI LIAO,WEN-WEI LIN,HAO-REN YAO
DOI: https://doi.org/10.69829/jdmh-024-0101-ta01
2024-06-30
Abstract:This article aims to build a framework of brain tumor segmentation for 3D MRI brain images by using the UNet-based deep learning method with optimal mass transportation (OMT) for data preprocessing. For this purpose, we develop a novel 2-phase UNet-based OMT to increase the ratio of brain tumors in the input OMT tensors. Moreover, due to the scarcity of training data, we change the density function by different parameters to increase the data diversity. For the post-processing, we propose an adaptive ensemble procedure by computing the eigenvector of the Dice similarity matrix corresponding to the largest eigenvalue and then using it to aggregate the probability as the predicted label. Using SegResUNet with OMT input tensors to train the data for the adult glioma (Task 1), Sub-Sahara-Africa adult glioma (Task 2), and meningioma (Task 3) in the international Brain Tumor Segmentation Cluster of Challenges 2023, the Dice scores of (whole tumor, tumor core, enhanced tumor) for online validations of Tasks 1, 2, and 3 are (0.9214, 0.8823, 0.8411), (0.8747, 0.8344, 0.8267), and (0.8316, 0.8395, 0.8401), respectively. Compared with random crop pre-processing, OMT is far superior.
What problem does this paper attempt to address?