Semi-supervised Learning Approach to Generate Neuroimaging Modalities with Adversarial Training

Harrison Nguyen,Simon Luo,Fabio Ramos
DOI: https://doi.org/10.48550/arXiv.1912.04391
2019-12-10
Abstract:Magnetic Resonance Imaging (MRI) of the brain can come in the form of different modalities such as T1-weighted and Fluid Attenuated Inversion Recovery (FLAIR) which has been used to investigate a wide range of neurological disorders. Current state-of-the-art models for brain tissue segmentation and disease classification require multiple modalities for training and inference. However, the acquisition of all of these modalities are expensive, time-consuming, inconvenient and the required modalities are often not available. As a result, these datasets contain large amounts of \emph{unpaired} data, where examples in the dataset do not contain all modalities. On the other hand, there is smaller fraction of examples that contain all modalities (\emph{paired} data) and furthermore each modality is high dimensional when compared to number of datapoints. In this work, we develop a method to address these issues with semi-supervised learning in translating between two neuroimaging modalities. Our proposed model, Semi-Supervised Adversarial CycleGAN (SSA-CGAN), uses an adversarial loss to learn from \emph{unpaired} data points, cycle loss to enforce consistent reconstructions of the mappings and another adversarial loss to take advantage of \emph{paired} data points. Our experiments demonstrate that our proposed framework produces an improvement in reconstruction error and reduced variance for the pairwise translation of multiple modalities and is more robust to thermal noise when compared to existing methods.
Machine Learning,Image and Video Processing
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the conversion problem between different modalities in brain magnetic resonance imaging (MRI). Specifically, the current state - of - the - art brain tissue segmentation and disease classification models require data of multiple modalities during training and inference. However, obtaining all data of these modalities is both expensive and time - consuming, and is often unavailable in clinical settings. Therefore, these datasets contain a large amount of unpaired data, that is, the examples in the dataset do not contain all modalities. On the other hand, only a small number of examples contain all modalities (paired data), and compared with the number of data points, the dimension of each modality is relatively high. These problems make the training of data - dependent models more difficult or reduce their applicability during inference. To solve these problems, the paper proposes a method to convert between two neuroimaging modalities through semi - supervised learning. The proposed model is called Semi - Supervised Adversarial CycleGAN (SSA - CGAN), which uses adversarial loss to learn from unpaired data points, uses cycle loss to enforce consistent reconstruction of mappings, and uses another adversarial loss to utilize paired data points. Experimental results show that the proposed framework reduces reconstruction error and variance in multi - modal pairwise translation, and is more robust in terms of thermal noise compared with existing methods.