SupAdversarialGeneratorLoss#

class deepinv.loss.adversarial.SupAdversarialGeneratorLoss(weight_adv=0.01, D=None, device='cpu', **kwargs)[source]#

Bases: GeneratorLoss

Supervised adversarial consistency loss for generator.

This loss was as used in conditional GANs such as Kupyn et al.[1] and generative models such as Bora et al.[2].

Constructs adversarial loss between reconstructed image and the ground truth, to be minimized by generator.

\(\mathcal{L}_\text{adv}(x,\hat x;D)=\mathbb{E}_{x\sim p_x}\left[q(D(x))\right]+\mathbb{E}_{\hat x\sim p_{\hat x}}\left[q(1-D(\hat x))\right]\)

See Imaging inverse problems with adversarial networks for examples of training generator and discriminator models.

Simple example (assuming a pretrained discriminator):

from deepinv.models import DCGANDiscriminator
D = DCGANDiscriminator() # assume pretrained discriminator

loss = SupAdversarialGeneratorLoss(D=D)

l = loss(x, x_net)

l.backward()
Parameters:

  • weight_adv (float) – weight for adversarial loss, defaults to 0.01 (from original paper)

  • D (torch.nn.Module) – discriminator network. If not specified, D must be provided in forward(), defaults to None.

  • device (str) – torch device, defaults to “cpu”

References:

forward(x, x_net, D=None, **kwargs)[source]#

Forward pass for supervised adversarial generator loss.

Parameters:

Examples using SupAdversarialGeneratorLoss:#

Imaging inverse problems with adversarial networks

Imaging inverse problems with adversarial networks