HyperAIHyperAI

Command Palette

Search for a command to run...

ProbGAN : Vers un GAN probabiliste avec des garanties théoriques

Guang-He Lee Hao Wang Yonglong Tian Hao He

Résumé

La modélisation probabiliste constitue un cadre rigoureux pour effectuer une agrégation de modèles, qui s’est imposée comme un mécanisme principal pour lutter contre le phénomène de « mode collapse » dans le contexte des réseaux antagonistes génératifs (GAN). Dans cet article, nous proposons un nouveau cadre probabiliste pour les GAN, appelé ProbGAN, qui apprend itérativement une distribution sur les générateurs à l’aide d’un prior soigneusement conçu. L’apprentissage est efficacement déclenché par un algorithme de Monte Carlo hamiltonien à gradient stochastique adapté, accompagné d’une nouvelle approximation de gradient permettant d’effectuer une inférence bayésienne. Notre analyse théorique révèle par ailleurs que notre approche constitue le premier cadre probabiliste à conduire à un équilibre où les distributions des générateurs sont fidèles à la distribution des données. Des expérimentations menées sur des données synthétiques à haute dimension et à multiples modes, ainsi que sur des bases d’images (CIFAR-10, STL-10 et ImageNet), démontrent la supériorité de notre méthode par rapport aux GAN à plusieurs générateurs de pointe ainsi qu’à d’autres traitements probabilistes pour les GAN.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp