无需适配奖励模型,一键优化文本生成图像的提示词。
Reward-Agnostic Prompt Optimization for Text-to-Image Diffusion Models
- 用大模型迭代生成提示词,不依赖特定奖励任务描述
- 测试时搜索效率提升4.8倍,且性能接近微调方法
- 适合多场景评估,如美学、物体关系、用户偏好等
我们研究了一种通用方法,通过在测试阶段寻找最大化奖励函数的提示词来改进文本到图像扩散模型的用户输入。尽管有多种奖励模型用于评估图像生成质量,但现有自动化提示工程方法通常针对特定奖励配置设计,导致在新场景下表现不佳。为此,我们提出RATTPO(Reward-Agnostic Test-Time Prompt Optimization),一种无需修改即可适应多种奖励场景的灵活测试时优化方法。RATTPO通过查询大语言模型迭代搜索最优提示词,无需奖励特定的任务描述;而是利用优化轨迹和一种新颖的奖励感知反馈信号(称为“提示”)作为上下文。实验表明,RATTPO在多种评估设置下均能有效提升提示质量,涵盖美学、人类偏好、对象空间关系等方面。其搜索效率比朴素的无奖励依赖基线平均快4.8倍。在足够推理预算下,性能可媲美需奖励特定微调的学习型基线。代码已开源:https://github.com/seminkim/RATTPO。
原文摘要 · Abstract (English)
We investigate a general approach for improving user prompts in text-to-image (T2I) diffusion models by finding prompts that maximize a reward function specified at test-time. Although diverse reward models are used for evaluating image generation, existing automated prompt engineering methods typically target specific reward configurations. Consequently, these specialized designs exhibit suboptimal performance when applied to new prompt engineering scenarios involving different reward models. To address this limitation, we introduce RATTPO (Reward-Agnostic Test-Time Prompt Optimization), a flexible test-time optimization method applicable across various reward scenarios without modification. RATTPO iteratively searches for optimized prompts by querying large language models (LLMs) \textit{without} requiring reward-specific task descriptions. Instead, it uses the optimization trajectory and a novel reward-aware feedback signal (termed a "hint") as context. Empirical results demonstrate the versatility of RATTPO, effectively enhancing user prompts across diverse reward setups that assess various generation aspects, such as aesthetics, general human preference, or spatial relationships between objects. RATTPO surpasses other test-time search baselines in search efficiency, running 4.8 times faster than naive reward-agnostic test-time search baseline on average. Furthermore, with sufficient inference budget, it can achieve comparable performance to learning-based baselines that require reward-specific fine-tuning. The code is available at https://github.com/seminkim/RATTPO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。