Latente Diffusionsmodelle (LDMs) haben in den letzten Jahren die Bildgenerierung revolutioniert. Indem sie Bilder in einem komprimierten latenten Raum repräsentieren, ermöglichen LDMs effizientes Training und eine hohe Bildqualität. Trotz ihres Erfolgs zeigen LDMs Schwächen bei der Generierung hochfrequenter Details und komplexer Kompositionen. Ein Forschungsteam hat nun eine Methode entwickelt, um die Detailgenauigkeit von LDMs durch zusätzliches Training im Pixelraum zu verbessern.
LDMs bilden Bilder durch einen iterativen Prozess, bei dem nach und nach Rauschen aus einem zufälligen Rauschen entfernt wird. Dieser Prozess findet im latenten Raum statt, der typischerweise eine deutlich geringere Auflösung als der Pixelraum aufweist. Die Kompression im latenten Raum, so die Hypothese des Forschungsteams, könnte der Grund für die Schwierigkeiten bei der Generierung feiner Details sein.
Um diese Hürde zu überwinden, schlagen die Forschenden vor, LDMs nach dem Training im latenten Raum einem zusätzlichen Training im Pixelraum zu unterziehen. Durch die Einführung eines pixelbasierten Ziels während des Nachtrainings soll die Detailgenauigkeit der generierten Bilder verbessert werden.
In ihren Experimenten demonstriert das Team, dass die zusätzliche Überwachung im Pixelraum sowohl die Qualität der Feinabstimmung als auch das präferenzbasierte Nachtraining signifikant verbessert. Sowohl die visuelle Qualität als auch die Metriken für visuelle Fehler zeigten bei State-of-the-Art-Modellen wie DiT-Transformer und U-Net-Diffusionsmodellen deutliche Verbesserungen.
Besonders hervorzuheben ist, dass die Verbesserung der Detailgenauigkeit nicht zu Lasten der Textausrichtung geht. Die Fähigkeit der Modelle, Bilder basierend auf Texteingaben zu generieren, blieb unverändert, was für Anwendungen im Bereich der Text-zu-Bild-Synthese von großer Bedeutung ist.
Das Nachtraining von LDMs im Pixelraum bietet mehrere Vorteile:
- Verbesserte Wiedergabe hochfrequenter Details - Präzisere Darstellung komplexer Bildkompositionen - Keine Beeinträchtigung der TextausrichtungsfähigkeitDie Forschungsarbeit zeigt, dass die zusätzliche Einbeziehung von Pixelraumdaten während des Nachtrainings ein vielversprechender Ansatz ist, um die Detailgenauigkeit und Qualität von LDMs zu verbessern. Diese Technik könnte zu robusteren und leistungsfähigeren Bildgenerierungsmodellen führen, die in der Lage sind, noch realistischere und detailliertere Bilder zu erzeugen.