CycleGAN for style transfer in X-ray angiography

Int J Comput Assist Radiol Surg. 2019 Oct;14(10):1785-1794. doi: 10.1007/s11548-019-02022-z. Epub 2019 Jul 8.

Abstract

Purpose: We aim to perform generation of angiograms for various vascular structures as a mean of data augmentation in learning tasks. The task is to enhance the realism of vessels images generated from an anatomically realistic cardiorespiratory simulator to make them look like real angiographies.

Methods: The enhancement is performed by applying the CycleGAN deep network for transferring the style of real angiograms acquired during percutaneous interventions into a data set composed of realistically simulated arteries.

Results: The cycle consistency was evaluated by comparing an input simulated image with the one obtained after two cycles of image translation. An average structural similarity (SSIM) of 0.948 on our data sets has been obtained. The vessel preservation was measured by comparing segmentations of an input image and its corresponding enhanced image using Dice coefficient.

Conclusions: We proposed an application of the CycleGAN deep network for enhancing the artificial data as an alternative to classical data augmentation techniques for medical applications, particularly focused on angiogram generation. We discussed success and failure cases, explaining conditions for the realistic data augmentation which respects both the complex physiology of arteries and the various patterns and textures generated by X-ray angiography.

Keywords: Data augmentation; Deep learning; Image translation; Vascular angiograms.

MeSH terms

  • Angiography / methods*
  • Diagnostic Techniques, Cardiovascular*
  • Humans
  • Image Processing, Computer-Assisted / methods*