Conditional Generation of Medical Images via Disentangled Adversarial Inference.

Mohammad Havaei,Ximeng Mao,Yipping Wang,Qicheng Lao
DOI: https://doi.org/10.1007/978-3-030-88210-5_5
2021-01-01
Abstract:We propose DRAI-a dual adversarial inference framework with augmented disentanglement constraints-to learn from the image itself, disentangled representations of style and content, and use this information to impose control over conditional generation process. We undergo two novel regularization steps to ensure content-style disentanglement. First, we minimize the shared information between content and style by introducing a novel application of the gradient reverse layer (GRL); second, we introduce a self-supervised regularization method to further separate information in the content and style variables. We conduct extensive qualitative and quantitative assessments on two publicly available medical imaging datasets (LIDC and HAM10000) and test for conditional image generation and style-content disentanglement. We also show that our proposed model (DRAI) achieves the best disentanglement score and has the best overall performance.
What problem does this paper attempt to address?