Non-Conditional Anatomically-Accurate 2D Synthetic Mask Generation

Although the use of AI models in medicine reveals great potential, the use of medical images for the training of models understandingly raises ethical and privacy concerns. This study aims to implement a WGAN-GP model that uses a set of lung CT scans for cancer-suffering patients to generate accurate 2D synthetic semantic segmentation masks, by segmenting each CT scan into semantic masks. To compare model’s performance, different sample resolutions and hyperparameters were experimented with. Results obtained demonstrate the model’s capability to correctly map lung anatomy and segment its different components, thus producing realistic and feasible semantic segmentation masks. While current findings are limited to 2D and sensitive to sample resolution, prospects envision the branching out into 3D medical-grade and more complex samples. Said results highlight the potential for such architectures to be used in tandem with mask-conditioned generative models and two-step data augmentation.