MD-GAN: Multi-Discriminator Generative Adversarial Networks for Distributed Datasets - Inria - Institut national de recherche en sciences et technologies du numérique
Pré-Publication, Document De Travail Année : 2018

MD-GAN: Multi-Discriminator Generative Adversarial Networks for Distributed Datasets

Résumé

A recent technical breakthrough in the domain of machine learning is the discovery and the multiple applications of Generative Adversarial Networks (GANs). Those generative models are computationally demanding, as a GAN is composed of two deep neural networks, and because it trains on large datasets. A GAN is generally trained on a single server. In this paper, we address the problem of distributing GANs so that they are able to train over datasets that are spread on multiple workers. MD-GAN is exposed as the first solution for this problem: we propose a novel learning procedure for GANs so that they fit this distributed setup. We then compare the performance of MD-GAN to an adapted version of Federated Learning to GANs, using the MNIST and CIFAR10 datasets. MD-GAN exhibits a reduction by a factor of two of the learning complexity on each worker node, while providing better performances than federated learning on both datasets. We finally discuss the practical implications of distributing GANs.
Fichier principal
Vignette du fichier
1811.03850.pdf (1.9 Mo) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-01946665 , version 1 (06-12-2018)
hal-01946665 , version 2 (09-05-2019)

Identifiants

  • HAL Id : hal-01946665 , version 1

Citer

Corentin Hardy, Erwan Le Merrer, Bruno Sericola. MD-GAN: Multi-Discriminator Generative Adversarial Networks for Distributed Datasets. 2018. ⟨hal-01946665v1⟩
425 Consultations
915 Téléchargements

Partager

More