CycleGAN: Both losses from discriminator and generator drop fast, after 100 epochs outputs blurred original image

I'm trying to train a 3D Cycle-GAN on medical image synthesis, more specifically CT to MR.

Currently I'm using a 3-Layer Discriminator and a 6 layer UNetGenerator borrowed from the official CycleGAN codes. Same lambda A, B of 10 and .5 of identity.

The discriminator loss drops to around 0 in the first few epochs, and the total loss for generator drops to around 1 as well. The generator continues to ouput blurred original input image.

During my debugging I noticed that the discriminator can distinguish the fake and real image very well. Should I increase the complexity of the generator? Can someone help to explain the problem with my model?

Topic cyclegan gan deep-learning

Category Data Science

About

Geeks Mental is a community that publishes articles and tutorials about Web, Android, Data Science, new techniques and Linux security.