DragonDiffusion est une solution d'édition d'images sophistiquée basée sur un modèle de diffusion. Elle prend en charge plusieurs modes d'édition, notamment le déplacement d'objets, le redimensionnement d'objets, le remplacement de l'apparence des objets et le glisser-déposer de contenu. Elle convertit les signaux d'édition en gradients grâce à une perte de correspondance de caractéristiques, modifiant ainsi la représentation intermédiaire du modèle de diffusion. Cette perte de correspondance de caractéristiques prend en compte plusieurs échelles d'alignement sémantique et géométrique, et intègre un mécanisme d'auto-attention inter-branches pour maintenir la cohérence entre l'image originale et le résultat édité.