HyperAIHyperAI

Command Palette

Search for a command to run...

ContraGAN : Apprentissage par contraste pour la génération d’images conditionnelles

Minguk Kang Jaesik Park

Résumé

La génération d’images conditionnelles consiste à produire des images diversifiées en utilisant des informations de label de classe. Bien que de nombreux réseaux adversariaires génératifs (GAN) conditionnels aient montré des résultats réalistes, ces méthodes considèrent uniquement les relations appariées entre l’embedding d’une image et l’embedding du label correspondant (relations données-classes) comme pertes de conditionnement. Dans cet article, nous proposons ContraGAN, une méthode qui prend en compte à la fois les relations entre plusieurs embeddings d’images au sein d’un même lot (relations données- données) et les relations données-classes, en utilisant une perte contrastive conditionnelle. Le discriminateur de ContraGAN évalue l’authenticité des échantillons donnés tout en minimisant un objectif contrastif afin d’apprendre les relations entre les images d’entraînement. Parallèlement, le générateur cherche à produire des images réalistes capables de tromper l’authenticité tout en présentant une faible perte contrastive. Les résultats expérimentaux montrent que ContraGAN surpasser les modèles de pointe de 7,3 % et 7,7 % respectivement sur les jeux de données Tiny ImageNet et ImageNet. En outre, nous démontrons expérimentalement que l’apprentissage contrastif contribue à atténuer le surajustement du discriminateur. Pour assurer une comparaison équitable, nous avons réimplémenté douze GAN de pointe à l’aide de la bibliothèque PyTorch. Le package logiciel est disponible à l’adresse suivante : https://github.com/POSTECH-CVLab/PyTorch-StudioGAN.


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
ContraGAN : Apprentissage par contraste pour la génération d’images conditionnelles | Articles | HyperAI