A User-Friendly Framework for Generating Model-Preferred Prompts in Text-to-Image Synthesis

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Hei, Nailei, Guo, Qianyu, Wang, Zihao, Wang, Yan, Wang, Haofen, Zhang, Wenqiang
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912982345187328
author Hei, Nailei
Guo, Qianyu
Wang, Zihao
Wang, Yan
Wang, Haofen
Zhang, Wenqiang
author_facet Hei, Nailei
Guo, Qianyu
Wang, Zihao
Wang, Yan
Wang, Haofen
Zhang, Wenqiang
contents Well-designed prompts have demonstrated the potential to guide text-to-image models in generating amazing images. Although existing prompt engineering methods can provide high-level guidance, it is challenging for novice users to achieve the desired results by manually entering prompts due to a discrepancy between novice-user-input prompts and the model-preferred prompts. To bridge the distribution gap between user input behavior and model training datasets, we first construct a novel Coarse-Fine Granularity Prompts dataset (CFP) and propose a novel User-Friendly Fine-Grained Text Generation framework (UF-FGTG) for automated prompt optimization. For CFP, we construct a novel dataset for text-to-image tasks that combines coarse and fine-grained prompts to facilitate the development of automated prompt generation methods. For UF-FGTG, we propose a novel framework that automatically translates user-input prompts into model-preferred prompts. Specifically, we propose a prompt refiner that continually rewrites prompts to empower users to select results that align with their unique needs. Meanwhile, we integrate image-related loss functions from the text-to-image model into the training process of text generation to generate model-preferred prompts. Additionally, we propose an adaptive feature extraction module to ensure diversity in the generated results. Experiments demonstrate that our approach is capable of generating more visually appealing and diverse images than previous state-of-the-art methods, achieving an average improvement of 5% across six quality and aesthetic metrics.
format Preprint
id arxiv_https___arxiv_org_abs_2402_12760
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A User-Friendly Framework for Generating Model-Preferred Prompts in Text-to-Image Synthesis
Hei, Nailei
Guo, Qianyu
Wang, Zihao
Wang, Yan
Wang, Haofen
Zhang, Wenqiang
Multimedia
Artificial Intelligence
Computer Vision and Pattern Recognition
Well-designed prompts have demonstrated the potential to guide text-to-image models in generating amazing images. Although existing prompt engineering methods can provide high-level guidance, it is challenging for novice users to achieve the desired results by manually entering prompts due to a discrepancy between novice-user-input prompts and the model-preferred prompts. To bridge the distribution gap between user input behavior and model training datasets, we first construct a novel Coarse-Fine Granularity Prompts dataset (CFP) and propose a novel User-Friendly Fine-Grained Text Generation framework (UF-FGTG) for automated prompt optimization. For CFP, we construct a novel dataset for text-to-image tasks that combines coarse and fine-grained prompts to facilitate the development of automated prompt generation methods. For UF-FGTG, we propose a novel framework that automatically translates user-input prompts into model-preferred prompts. Specifically, we propose a prompt refiner that continually rewrites prompts to empower users to select results that align with their unique needs. Meanwhile, we integrate image-related loss functions from the text-to-image model into the training process of text generation to generate model-preferred prompts. Additionally, we propose an adaptive feature extraction module to ensure diversity in the generated results. Experiments demonstrate that our approach is capable of generating more visually appealing and diverse images than previous state-of-the-art methods, achieving an average improvement of 5% across six quality and aesthetic metrics.
title A User-Friendly Framework for Generating Model-Preferred Prompts in Text-to-Image Synthesis
topic Multimedia
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2402.12760