Inpainting Croppped Diffusion MRI using Deep Generative Models

Code used for the analysis in Ayub et al. 2020, "Inpainting Cropped Diffusion MRI using Deep Generative Models"

Cropping artifacts can occur in MRI scans and neuroimaging data when the subject is poorly positioned in the scanner or the field-of-view does not contain entire head. This tends to occur in developmental studies where head size changes with age in the same subjects and the field-of-view from baseline scans becomes suboptimal. Having missing parts of the skull, even worse the brain itself, can be problematic for subsequent preprocessing steps such as aligning subjectt images within the same group or multiple scans of the same subject. Also, missing data is detrimental for group analyses.

Here, we reconstruct the missing data using generative models, specifically variational autoencoders and GANs. The vqvae.ipynb Jupyter Notebook contains code for training the variational autoencoder models. The vaegan.ipynb Jupyter Notebook contains code for training the GAN models.