aMUSEd: An Open MUSE Reproduction

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Patil, Suraj, Berman, William, Rombach, Robin, von Platen, Patrick
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916080447913984
author Patil, Suraj
Berman, William
Rombach, Robin
von Platen, Patrick
author_facet Patil, Suraj
Berman, William
Rombach, Robin
von Platen, Patrick
contents We present aMUSEd, an open-source, lightweight masked image model (MIM) for text-to-image generation based on MUSE. With 10 percent of MUSE's parameters, aMUSEd is focused on fast image generation. We believe MIM is under-explored compared to latent diffusion, the prevailing approach for text-to-image generation. Compared to latent diffusion, MIM requires fewer inference steps and is more interpretable. Additionally, MIM can be fine-tuned to learn additional styles with only a single image. We hope to encourage further exploration of MIM by demonstrating its effectiveness on large-scale text-to-image generation and releasing reproducible training code. We also release checkpoints for two models which directly produce images at 256x256 and 512x512 resolutions.
format Preprint
id arxiv_https___arxiv_org_abs_2401_01808
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle aMUSEd: An Open MUSE Reproduction
Patil, Suraj
Berman, William
Rombach, Robin
von Platen, Patrick
Computer Vision and Pattern Recognition
We present aMUSEd, an open-source, lightweight masked image model (MIM) for text-to-image generation based on MUSE. With 10 percent of MUSE's parameters, aMUSEd is focused on fast image generation. We believe MIM is under-explored compared to latent diffusion, the prevailing approach for text-to-image generation. Compared to latent diffusion, MIM requires fewer inference steps and is more interpretable. Additionally, MIM can be fine-tuned to learn additional styles with only a single image. We hope to encourage further exploration of MIM by demonstrating its effectiveness on large-scale text-to-image generation and releasing reproducible training code. We also release checkpoints for two models which directly produce images at 256x256 and 512x512 resolutions.
title aMUSEd: An Open MUSE Reproduction
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2401.01808