Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

DualGAN: Unsupervised Dual Learning for Image-to-Image Translation

About

Conditional Generative Adversarial Networks (GANs) for cross-domain image-to-image translation have made much progress recently. Depending on the task complexity, thousands to millions of labeled image pairs are needed to train a conditional GAN. However, human labeling is expensive, even impractical, and large quantities of data may not always be available. Inspired by dual learning from natural language translation, we develop a novel dual-GAN mechanism, which enables image translators to be trained from two sets of unlabeled images from two domains. In our architecture, the primal GAN learns to translate images from domain U to those in domain V, while the dual GAN learns to invert the task. The closed loop made by the primal and dual tasks allows images from either domain to be translated and then reconstructed. Hence a loss function that accounts for the reconstruction error of images can be used to train the translators. Experiments on multiple image translation tasks with unlabeled data show considerable performance gain of DualGAN over a single GAN. For some tasks, DualGAN can even achieve comparable or slightly better results than conditional GAN trained on fully labeled data.

Zili Yi, Hao Zhang, Ping Tan, Minglun Gong• 2017

Related benchmarks

TaskDatasetResultRank
SegmentationCardiac MRI
Accuracy95.4
10
SegmentationRetinal OCT
Accuracy83.9
10
SegmentationChest MRI to CT
Accuracy81.8
10
Image SynthesisChest MRI to CT (test)
FID256.3
9
Image SynthesisRetinal OCT (test)
FID257.5
9
Image SynthesisCardiac MRI (test)
FID295.8
9
Unpaired Image-to-Image Translation (Horse to Zebra)Horse2Zebra (test)
KID (x100)10.38
8
Unpaired Image-to-Image Translation (Orange to Apple)Apple2Orange (test)
KID0.1242
8
Unpaired Image-to-Image Translation (Zebra to Horse)Horse2Zebra (test)
KID (x100)12.86
8
Unpaired Image-to-Image Translation (Apple to Orange)Apple2Orange (test)
KID0.1304
8
Showing 10 of 22 rows

Other info

Code

Follow for update