HyperAIHyperAI

Command Palette

Search for a command to run...

Modèle de diffusion binaire probabiliste

author1 author2

Résumé

Nous présentons le modèle de diffusion probabiliste binaire (Binary Diffusion Probabilistic Model, BDPM), une nouvelle approche générative optimisée pour les représentations de données binaires. Bien que les modèles de diffusion probabiliste débruitants (Denoising Diffusion Probabilistic Models, DDPMs) aient montré des succès notables dans des tâches telles que la synthèse et la restauration d'images, les DDPMs traditionnels s'appuient sur des représentations de données continues et utilisent une perte quadratique moyenne (Mean Squared Error, MSE) pour l'entraînement, en appliquant des modèles de bruit gaussiens qui peuvent ne pas être optimaux pour les structures de données discrètes ou binaires. Le BDPM remédie à cela en décomposant les images en plans de bits et en utilisant des transformations de bruit basées sur XOR, avec un modèle débruiteur entraîné à l'aide d'une perte de divergence croisée binaire (binary cross-entropy). Cette méthode permet un contrôle précis du bruit et une inférence computationnellement efficace, réduisant considérablement les coûts informatiques et améliorant la convergence du modèle. Lorsqu'il est évalué sur des tâches de restauration d'images telles que la super-résolution, l'inpainting et la restauration aveugle d'images, le BDPM surpasses les méthodes de pointe sur les jeux de données FFHQ, CelebA et CelebA-HQ. Il convient de souligner que le BDPM nécessite moins d'étapes d'inférence que les modèles DDPM traditionnels pour atteindre des résultats optimaux, mettant ainsi en évidence son efficacité accrue en matière d'inférence.


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