This model and its implementation is partially based on "Generation of 3D Brain MRI Using Auto-Encoding Generative Adversarial Networks" by Kwon et al. paper: (https://arxiv.org/abs/1908.02498) implementation: (https://github.com/cyclomon/3dbraingen)