近日，阿里云人工智能平台 PAI 与华南理工大学朱金辉教授团队合作在自然语言处理顶级会议 EMNLP2023 上发表了 BeautifulPrompt 的深度生成模型，可以从简单的图片描述中生成高质量的提示词，从而使文生图模型能够生成更美观的图像。BeautifulPrompt 通过对低质量和高质量的提示进行微调，并进一步提出了一种基于强化学习和视觉信号反馈的技术，以最大化生成提示的奖励值。

论文：

Tingfeng Cao, Chengyu Wang, Bingyan Liu, Ziheng Wu, Jinhui Zhu, Jun Huang. BeautifulPrompt: Towards Automatic Prompt Engineering for Text-to-Image Synthesis. EMNLP 2023 (Industry Track)

背景

文生图是 AIGC 中最引人注目和广泛应用的技术之一，旨在通过文本输入创建逼真的图像。最近，随着大型模型建模能力的提升，文生图模型得到快速的发展。大规模的 TIS 模型，如 DALLE-2、Imagen 和 stable diffusion，显著提高了最先进的性能，并允许没有艺术专业知识的用户通过个人想象力创建前所未有的图像。

然而，文成图模型要求用户在模型推理之前编写文本提示（例如“一艘雄伟的帆船”）。编写满足设计师或艺术工作者需求的这些提示充满了不确定性，就像开盲盒一样。这是由于训练数据的质量问题，导致需要详细的描述才能生成高质量的图像。在现实场景中，非专家往往很难手工编写这些提示，并且需要通过试错的迭代修改来重新生成图像，从而导致时间和计算资源的严重浪费。

提示工程（prompt engineering）是一个新兴的研究领域，旨在探索如何为深度生成模型提供提示，并提高人与 AI 之间直接交互的效率。因此，我们关注于大语言模型（LLM）自动地生成高质量的提示词，下图展示了使用简单的图片描述和 BeautifulPrompt 之后生产的图片。

算法概述

数据收集

我们提出一个自动化收集 prompt 优化的数据集方案：

原始数据源是 DiffusionDB，它只包含未配对的提示。启发式地，我们根据提示的长度、提示中包含的某些标签等将提示分为低质量提示和高质量提示。

接下来，我们

i)使用 BLIP 对与高质量提示相关的图像进行 caption，并将结果视为相应的低质量提示，因为说明文字较短且缺乏细节;

ii)使用 ChatGPT 对高质量的提示进行总结，并将总结视为低质量的提示;

iii)使用 ChatGPT 从低质量的提示生成更好的提示;结果被认为是高质量的提示。

通过以上三种方法，我们获得了大量的提示对；然而，这些提示对的质量无法保证。因此，我们需要进行进一步的数据清理和过滤。我们清洗了包含色情、政治敏感等不适合工作场景的数据，并对图片的美观值进行筛选。得到最终的数据集。与 InstructGPT 类似，我们采用了三阶段的训练，整体训练架构图如下：

Step 1. SFT

给定提示对的数据集$D=(\mathbf{x},\mathbf{y})$，其中包含低质量提示对$\mathbf{x}$和高质量提示对$\mathbf{y}$，我们对仅解码器语言模模型（BLOOM）进行微调，以输出具有给定指令和低质量提示$\mathbf{x}$的高质量令牌 token$\mathbf{y}=y_1,...,y_n$。

我们使用自回归语言建模目标来微调语言模型:

$\mathcal{L}{sft} = - \sum_i \log P(y_i\mid \mathbf{x}, y_1, ..., y{i-1})$

Step 2. RM

我们基于 PickScore 和 Aesthetic Score 来训练奖励模型。

简单地说，PickScore 是一个基于文本到图像提示和真实用户偏好的大型数据集训练的偏好模型，它在预测人类对图像的偏好方面表现出超人的表现。我们计算低质量提示和相应高质量提示生成的图像的 PickScore。为了减少随机种子对 TIS 模型生成的图像质量的影响，我们使用 8 种不同的随机种子生成图像并对结果进行平均。计算的平均 PickScore $\mathbb{P}\mathbb{S}$被用作训练奖励模型的基础真相。损失函数为:

$\mathcal{L}{ps} = - \frac{1}{N}\sum_i^N \text{MSE}(r{ps}(\mathbf{x}, \mathbf{y}), \mathbb{PS}), $

其中$r_{ps}(\mathbf{x},\mathbf{y})$是提示对$(\mathbf{x},\mathbf{y})$的奖励模型的标量输出。MSE 是均方误差。N 为样本总数。

类似地，奖励模型还被训练以将图像中的相应提示与美学分数匹配$\mathbb{A}\mathbb{E}\mathbb{S}$:

$\mathcal{L}{aes} = - \frac{1}{N}\sum_i^N \text{MSE}(r{aes}(\mathbf{y}), \mathbb{AES}),$

其中$r_{aes}(\mathbf{y})$是奖励模型的标量输出。

最后，我们使用$\alpha$作为平衡因子，将两个奖励模型的得分组合为最终奖励:

Step 3. PPO（RL）

由于收集到的数据集不可避免地包含一些噪声，例如，低质量提示与相应的高质量提示之间的一致性相对较低，因此监督训练模型 $\rho$ 的性能可能不令人满意。为了进一步提高模型性能，我们初始化一个策略$\pi =\rho$ , 然后微调$\pi$来使用强化学习执行任务。我们利用近端策略优化(PPO) 算法直接优化预期奖励:

其中$\beta$为 Kullback-Leibler (KL)罚系数。它可以防止政策偏离 $\rho$ 。我们在这里采用自适应 KL 惩罚。

算法评测

为了验证 BeautifulPrompt 的有效性，我们在一些基于模型打分的客观指标和人类主观评估上做了实验：

我们也对算法的模块进行了详细有效性分析，我们可以算法的各个流程都是有效的。

为了更好地服务开源社区，BeautifulPrompt 算法的源代码即将贡献在自然语言处理算法框架 EasyNLP 中，欢迎 NLP 从业人员和研究者使用。

EasyNLP 开源框架：https://github.com/alibaba/EasyNLP

参考文献

• Chengyu Wang, Minghui Qiu, Taolin Zhang, Tingting Liu, Lei Li, Jianing Wang, Ming Wang, Jun Huang, Wei Lin. EasyNLP: A Comprehensive and Easy-to-use Toolkit for Natural Language Processing. EMNLP 2022

• Stiennon, Nisan, et al. "Learning to summarize with human feedback." Advances in Neural Information Processing Systems 33 (2020): 3008-3021

• Rombach, Robin, et al. "High-resolution image synthesis with latent diffusion models." Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2022

• Kirstain, Yuval, et al. "Pick-a-pic: An open dataset of user preferences for text-to-image generation." arXiv preprint arXiv:2305.01569 (2023)

论文信息

论文标题：BeautifulPrompt: Towards Automatic Prompt Engineering for Text-to-Image Synthesis

论文作者：曹庭锋、汪诚愚、刘冰雁、吴梓恒、朱金辉、黄俊

论文 pdf 链接：https://arxiv.org/abs/2311.06752