Discriminative Class Tokens for Text-to-Image Diffusion Models

Publikation: Bidrag til bog/antologi/rapportKonferencebidrag i proceedingsForskningfagfællebedømt

Recent advances in text-to-image diffusion models have enabled the generation of diverse and high-quality images. While impressive, the images often fall short of depicting subtle details and are susceptible to errors due to ambiguity in the input text. One way of alleviating these issues is to train diffusion models on class-labeled datasets. This approach has two disadvantages: (i) supervised datasets are generally small compared to large-scale scraped text-image datasets on which text-to-image models are trained, affecting the quality and diversity of the generated images, or (ii) the input is a hard-coded label, as opposed to free-form text, limiting the control over the generated images.In this work, we propose a non-invasive fine-tuning technique that capitalizes on the expressive potential of freeform text while achieving high accuracy through discriminative signals from a pretrained classifier. This is done by iteratively modifying the embedding of an added input token of a text-to-image diffusion model, by steering generated images toward a given target class according to a classifier. Our method is fast compared to prior fine-tuning methods and does not require a collection of in-class images or retraining of a noise-tolerant classifier. We evaluate our method extensively, showing that the generated images are: (i) more accurate and of higher quality than standard diffusion models, (ii) can be used to augment training data in a low-resource setting, and (iii) reveal information about the data used to train the guiding classifier. The code is available at https://github.com/idansc/discriminative-class-tokens.

OriginalsprogEngelsk
TitelProceedings - 2023 IEEE/CVF International Conference on Computer Vision, ICCV 2023
Antal sider11
ForlagInstitute of Electrical and Electronics Engineers Inc.
Publikationsdato2023
Sider22668-22678
ISBN (Elektronisk)9798350307184
DOI
StatusUdgivet - 2023
Begivenhed2023 IEEE/CVF International Conference on Computer Vision, ICCV 2023 - Paris, Frankrig
Varighed: 2 okt. 20236 okt. 2023

Konference

Konference2023 IEEE/CVF International Conference on Computer Vision, ICCV 2023
LandFrankrig
ByParis
Periode02/10/202306/10/2023
NavnProceedings of the IEEE International Conference on Computer Vision
ISSN1550-5499

Bibliografisk note

Publisher Copyright:
© 2023 IEEE.

ID: 389306742