HyperAIHyperAI

Command Palette

Search for a command to run...

SA-Solver : Solveur Adams stochastique pour un échantillonnage rapide des modèles de diffusion

Shuchen Xue Mingyang Yi Weijian Luo Shifeng Zhang Jiacheng Sun Zhenguo Li Zhi-Ming Ma

Résumé

Les Modèles de diffusion probabilistes (DPMs) ont connu un succès considérable dans les tâches de génération. Étant donné que l’échantillonnage à partir des DPMs revient à résoudre une équation différentielle stochastique (SDE) ou une équation différentielle ordinaire (ODE) de diffusion, une tâche coûteuse en temps, de nombreuses méthodes rapides d’échantillonnage fondées sur des solveurs améliorés d’équations différentielles ont été proposées. La majorité de ces techniques se concentrent sur la résolution de l’ODE de diffusion, en raison de son efficacité supérieure. Toutefois, l’échantillonnage stochastique peut offrir des avantages supplémentaires en matière de diversité et de qualité des données générées. Dans ce travail, nous menons une analyse approfondie de l’échantillonnage stochastique sous deux angles : la SDE de diffusion à variance contrôlée et le solveur SDE à plusieurs étapes linéaires. À partir de cette analyse, nous proposons \textit{SA-Solver}, une méthode améliorée et efficace de type Adams stochastique pour résoudre les SDE de diffusion, permettant ainsi de générer des données de haute qualité. Nos expérimentations montrent que \textit{SA-Solver} atteint : 1) des performances améliorées ou comparables aux méthodes d’échantillonnage d’avant-garde (SOTA) existantes pour un petit nombre d’étapes d’échantillonnage ; 2) un score FID d’avant-garde (SOTA) sur de nombreux jeux de données standards, pour un nombre adéquat d’évaluations de fonction (NFE). Le code est disponible à l’adresse suivante : https://github.com/scxue/SA-Solver.


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