Wasserstein Loss for Semantic Editing in the Latent Space of GANs

Perla Doubinsky (1, 2) , Nicolas Audebert (1, 2) , Michel Crucianu (1, 2) , Hervé Le Borgne (3)

1 CEDRIC - VERTIGO - CEDRIC. Données complexes, apprentissage et représentations
2 CNAM - Conservatoire National des Arts et Métiers [CNAM]
3 LIST (CEA) - Laboratoire d'Intégration des Systèmes et des Technologies

The latent space of GANs contains rich semantics reflecting the training data. Different methods propose to learn edits in latent space corresponding to semantic attributes, thus allowing to modify generated images. Most supervised methods rely on the guidance of classifiers to produce such edits. However, classifiers can lead to out-of-distribution regions and be fooled by adversarial samples. We propose an alternative formulation based on the Wasserstein loss that avoids such problems, while maintaining performance on-par with classifier-based approaches. We demonstrate the effectiveness of our method on two datasets (digits and faces) using StyleGAN2.

Mots clés: GAN; Wasserstein distance; image edition; GAN Wasserstein distance image edition

Domaines: Intelligence artificielle [cs.AI]

Perla Doubinsky, Nicolas Audebert, Michel Crucianu, Hervé Le Borgne. Wasserstein Loss for Semantic Editing in the Latent Space of GANs. 20th International Conference on Content-based Multimedia Indexing, Sep 2023, Orléans, France. ⟨hal-04036414⟩