A Novel Domain Adaptation Framework for Medical Image Segmentation
This work addresses domain adaptation for medical image segmentation, particularly in brain tumor analysis, but is incremental as it builds on existing neural network methods.
The paper tackled the problem of medical image segmentation by proposing a domain adaptation framework that uses a biophysical tumor growth model and generative adversarial networks to create synthetic multimodal MR images, achieving better results than existing state-of-the-art GAN models on the BraTS'18 challenge.
We propose a segmentation framework that uses deep neural networks and introduce two innovations. First, we describe a biophysics-based domain adaptation method. Second, we propose an automatic method to segment white and gray matter, and cerebrospinal fluid, in addition to tumorous tissue. Regarding our first innovation, we use a domain adaptation framework that combines a novel multispecies biophysical tumor growth model with a generative adversarial model to create realistic looking synthetic multimodal MR images with known segmentation. Regarding our second innovation, we propose an automatic approach to enrich available segmentation data by computing the segmentation for healthy tissues. This segmentation, which is done using diffeomorphic image registration between the BraTS training data and a set of prelabeled atlases, provides more information for training and reduces the class imbalance problem. Our overall approach is not specific to any particular neural network and can be used in conjunction with existing solutions. We demonstrate the performance improvement using a 2D U-Net for the BraTS'18 segmentation challenge. Our biophysics based domain adaptation achieves better results, as compared to the existing state-of-the-art GAN model used to create synthetic data for training.