Qwen-Image-PE:面向文生图提示词增强的视觉设计学习 Qwen-Image-PE: Learning Visual Design for Text-to-Image Prompt Enhancement
反向蒸馏加组内设计奖励的两阶段训练,让提示词增强器显式学会光照、构图、版式与细节设计。
前置知识
提示词增强(Prompt Enhancement, PE)
PE 是文生图流水线中位于用户请求与生成模型之间的文本扩写环节:把“一罐桃子气泡水的产品照,1:1”这类简短请求,扩写成包含光照方向、构图层级、版式安排、材质细节的完整视觉规格,同时不得篡改用户的显式要求(品牌文字、比例、内容主体)。生成模型对长提示词的响应能力越强,PE 的设计空间与责任越大。
本文的研究对象就是 PE:两阶段训练(CRD 与 GDR)、评测基准、缩放分析全部围绕“如何训练一个会做视觉设计的增强器模型”展开。
GRPO(Group Relative Policy Optimization)
组相对策略优化,一种免价值网络的在线强化学习算法:对同一请求采样一组(group)响应,把每个响应的奖励在组内做标准化(减均值、除标准差)得到相对优势,直接用策略梯度更新模型。相比 PPO 省去了单独训练的价值函数,是目前 LLM RLHF 的主流选择之一。
论文第二阶段直接用 GRPO 优化增强器:GDR 打出的设计分与对齐分先合成为总奖励 $R_i$,再在组内归一化计算优势更新参数,不理解的读者无法读懂 2.3 节。
MLLM 裁判与奖励模型
用多模态大模型(MLLM)按评分细则(rubric)给生成图像打分,充当 RL 的奖励函数。传统偏好模型如 PickScore、HPSv3 只输出一个整体偏好分数;MLLM 裁判可按维度细评,但存在位置偏差(图片在网格中的顺序影响分数)和评分噪声,通常靠多次随机化采样取平均来缓解。
GDR 的核心就是把 MLLM 裁判的“组内联合打分”当作设计奖励。理解裁判的偏差来源,才能读懂论文为什么要做 $K$ 次洗牌网格平均(式 7-8)以及为什么组内互照能提升与人类判断的一致率。
KL 正则化(KL regularization)
RL 微调语言模型时,在目标函数中加入当前策略与参考策略之间的 KL 散度惩罚项:$\mathcal{L}_{total} = \mathcal{L}_{RL} + \beta\, D_{KL}(\pi_\theta \| \pi_{ref})$。作用是防止策略为了刷高奖励而偏离合理的语言分布(reward hacking),保持输出格式和基础能力的稳定。
论文在 GRPO 中加了拉向冻结 SFT 初始策略的 KL 正则,保证 RL 阶段在提升设计分的同时不破坏第一阶段学到的输出格式与语言质量。
GPG(Grounded Perplexity Gain,接地困惑度增益)
衡量文本“图像可验证信息量”的指标:把文本和配对图像一起喂给冻结的视觉语言模型,计算观察到图像前后该文本 token 对数似然的提升之和并取平均。图像能解释的文本细节越多,GPG 越高;空洞套话的 GPG 接近零。论文用 Qwen3-VL-8B-Instruct 计算。
论文用 GPG 定义信息密度这一缩放轴:L1–L4 四档的 GPG 从 14.42 升到 170.36,对应 Overall 从 44.65 升到 57.31,并得出“信息密度足够是增强有效的前提”这一关键结论。
MAUVE 与 JS 散度(文本分布对齐度量)
MAUVE 基于两个文本语料在嵌入空间中的聚类分布相似度,取值 0 到 1、越高越接近;Jensen–Shannon(JS)散度基于 corpus 级 token(unigram/bigram)频率分布,越低越接近;FD 是嵌入表示上的 Fréchet 距离,同样越低越好。三者分别从 token、分布、嵌入三个层面刻画文本分布差距。
论文用这组指标证明 CRD 缩小了“增强提示词分布 vs T2I 训练 caption 分布”的鸿沟:unigram JS 从 0.229 降到 0.105,MAUVE 从 0.032 升到 0.414,这是反向蒸馏动机成立的定量证据。
研究动机
现代文生图模型(Qwen-Image-2.1、FLUX 3、HunyuanImage 3.0 等)已能遵循精细的视觉规格,对光照、构图、版式和细节都有可控性,但用户的请求往往只有一句话,大量视觉决策处于留白状态。提示词增强(PE)负责补全这些决策,而现有方法分两类:一类追求语义对齐,如 HunyuanImage-2.1-PE 用 CoT 保留并澄清用户意图、ERNIE-Image-PE 补充结构化版式描述、FLUX 3 Klein PE 做常规扩写;另一类用审美或人类偏好奖励(PickScore、HPSv3)提升整体“好看程度”。问题在于:两者都只优化整体观感,没有显式地逐维度优化精细视觉设计。数据上,Qwen-Image-Bench 中原始请求的 Lighting 只有 64.12、Composition 54.68、Layout 54.88、Detail 55.15,部分训练型基线甚至在 Detail 上倒退(HunyuanImage-2.1-PE-32B 为 54.65)。如何把“光源方向是否物理自洽、视觉焦点层级是否清晰”这类设计判断转化为可监督的目标和可优化的反馈,是该领域缺失的一环。
本文的目标是本文要训练一个显式学习视觉设计的提示词增强器:给定用户请求 $x$,输出思维链理由 $z$、增强提示词 $y$ 和宽高比 $\rho$,交给冻结的 T2I 生成器出图。具体目标有四层:其一,在光照、构图、版式、细节四个精细维度上系统性地超过所有现役 PE(包括 GPT-5.5、Gemini-3.1-Pro、Qwen-3.8-Max 等闭源 API 与 Hunyuan/ERNIE/FLUX 的训练型增强器);其二,在整体生成质量上把 Qwen-Image-Bench 的 Overall 从原始请求的 54.36 显著拉高,同时保证对用户显式要求的对齐不掉分;其三,同步提升美学(UniPercept IAA)、图像质量(IQA)、结构纹理(ISTA)与人类偏好(HPSv3、GSB 人工评估);其四,学到可迁移的通用设计原则而非绑定某个生成器的 caption 模式,能直接用于 FLUX.2-dev、Krea 2 Turbo、HunyuanImage-2.1、FLUX 3 Image Klein 四个未参与训练的 backbone,并开源一个 9B 可用版本。
与已有工作不同的是,本文的独特切入是两个方向上的重构。第一是监督信号的方向:不做“正向”扩写(让 LLM 凭空脑补详细提示词),而是反向蒸馏(CRD)——先拿到设计精良的图片,用多模态模型把它在光照/构图/版式/细节上的选择密集地反标成 caption,再把这个 caption 当作增强提示词的学习目标。直觉是:好图本身就是设计答案,caption 只是把答案翻译成文字;这同时天然把 PE 输出拉回 T2I 训练 caption 的分布内(MAUVE 从 0.032 提到 0.414),而 forward 式增强的输出分布与训练 caption 几乎不重叠。第二是奖励信号的结构:现有 reward model 只能给整体偏好分,无法区分四个设计维度;pointwise 单图打分缺乏参照系、pairwise 两两比较调用次数平方级爆炸($D\cdot G(G-1)/2$ 次)。GDR 把同一请求的全部 $G$ 张候选图拼成一张网格,让 MLLM 裁判在一次调用中互相对照打分,用共享参照校准相对设计质量,只需 $D\cdot K$ 次调用($K\ll G$),两头兼顾。
核心方法
整体思路是“先模仿、再比较”的两阶段训练。形式化定义:增强器采样 $o=(z,y,\rho)\sim\pi_\theta(\cdot|x)$,冻结的生成器输出 $I\sim G(\cdot|y,\rho)$。第一阶段是带 Caption-Based Reverse Distillation(CRD)的监督微调:先构造 request–image 对(Request-to-Image 路径用强 T2I 系统按真实与合成请求出图再按设计质量和一致性筛选;Image-to-Request 路径把高质量图的短 caption 退化成含噪简短请求),用多模态 captioner 生成密集描述 $y^*_i=\mathrm{Captioner}(I_i)$ 与宽高比 $\rho^*_i$,再用 LLM 合成设计理由 $z^*_i=\mathrm{LLM}(x_i,y^*_i)$,最后以 SFT 损失 $\mathcal{L}_{SFT}=-\frac{1}{N}\sum_i\big[\log\pi_\theta(z^*_i|x_i)+\log\pi_\theta(y^*_i,\rho^*_i|x_i,z^*_i)\big]$ 训练模型先说理由再写提示词。第二阶段是带 Groupwise Design Reward(GDR)的强化学习:每个请求采样 $G$ 个响应、共享随机种子渲染 $G$ 张图拼成网格,MLLM 裁判按光照/构图/版式/细节四套 rubric 联合打分,做 $K$ 次洗牌取平均消除位置偏差,再与对齐奖励合成 $R_i=\sum_d\lambda_d r^{design}_{i,d}+\lambda_a r^{align}_i$,用 GRPO 优化并加 KL 正则拉向 SFT 策略。两个变体分别初始化自 Qwen3.5-9B 与 Qwen3.5-397B-A17B。
核心创新有两点,且都能被消融直接验证。第一,Caption-Based Reverse Distillation 把监督学习的方向倒过来:不问“LLM 该怎么扩写”,而问“好图是怎么被设计出来的”。论文给了两条证据:一是 Figure 3a 显示好图的密集 caption 能引导 Qwen-Image-2.1 再现关键的光照与构图选择,说明 caption 确实编码了可执行的设计知识;二是 t-SNE 投影中 PE-397B 的提示词嵌入与 T2I 训练 caption 嵌入的重叠度远高于 forward 式增强。消融表 3(a) 证实:同样用这个 MLLM 当目标来源,CRD(当 captioner)比 Forward-Based Supervision(当零样本扩写器)Overall 高 1.66 分(61.52 对 59.86),Aesthetics 高 3.00 分。第二,Groupwise Design Reward 的组内联合评分:裁判一次看到全部 $G$ 张候选,形成共享视觉参照,能分辨单图打分分不出的细微设计差异——与人类盲评的一致率从 pointwise 的 50.9% 升到 59.1%(+8.2 个百分点)。效率上每次组只需 $D\cdot K$ 次裁判调用,远低于 pointwise 的 $D\cdot G$ 和 exhaustive pairwise 的 $D\cdot G(G-1)/2$,后者在 $G$ 较大时对在线 RL 完全不可行。
方法步骤详情
完整流程如下。阶段一(CRD 监督微调):(1) 构建图像锚定目标——Request-to-Image 路径:用强 T2I 系统对真实与合成请求渲染图片,按设计质量与请求一致性双重筛选;Image-to-Request 路径:把质量过滤后的图片的短 caption 退化成噪声更大、设计信息丢失但保留核心意图的用户请求。(2) 对每张选中图片 $I_i$,多模态 captioner 输出覆盖光照/构图/版式/细节的密集描述 $y^*_i$(式 2),同时记录 $\rho^*_i=\mathrm{width}(I_i)/\mathrm{height}(I_i)$。(3) 用 LLM 以 $(x_i,y^*_i)$ 为输入合成 CoT 设计理由 $z^*_i$(式 3),说明哪些要求需要保留、目标 caption 中的设计选择为何合理。(4) 组成 $\mathcal{D}_{SFT}=\{(x_i,z^*_i,y^*_i,\rho^*_i)\}$(式 4),按式 5 训练增强器依次生成理由、提示词与比例。阶段二(GDR 强化学习):(1) 从 SFT 模型的失败案例中挖掘弱设计请求,再合成补充请求,构建 RL 提示词池 $\mathcal{D}_{RL}$。(2) 每个请求采 $G$ 个响应,冻结生成器用共享种子渲染 $G$ 张候选图(式 6)。(3) 按维度 rubric 把 $K$ 个随机洗牌网格交给 MLLM 裁判联合打分,归一化到 $[0,1]$ 后按式 8 求平均 $r^{design}_{i,d}=\frac{1}{K}\sum_k r^{design,(k)}_{i,d}$。(4) 按式 9 合成总奖励 $R_i$,其中对齐项保住用户显式要求。(5) GRPO 组内归一化计算相对优势更新增强器,KL 拉向 SFT 初始策略,T2I 生成器全程冻结。基准评测请求与训练请求不重叠,评测指标也与训练奖励独立。
技术新颖性
技术新颖性体现在四处。一,CRD 的“逆向”构造在 PE 文献中是新的:以往增强器都用 LLM 正向扩写或人类改写做 SFT 目标,本文证明把“图像→密集 caption”的反向管线当目标,既注入设计知识又顺带解决了提示词与训练 caption 的分布错位问题(这一错位此前很少有人量化,本文给出 JS/FD/MAUVE 三层证据)。二,GDR 的“组内网格互照”评分机制是新的奖励结构:它在 pointwise(无参照、$D\cdot G$ 次调用)与 exhaustive pairwise(有参照、平方级调用)之间开辟了第三条路,用空间网格提供参照、用 $K$ 次洗牌控制偏差、用 $D\cdot K$ 次调用控制成本,并实证其排序信号与人类一致率显著更高。三,首次把 PE 的缩放性质系统化为三条轴:训练数据(12.5K→100K)、模型规模(2B→397B)、信息密度(GPG 14.42→170.36),并发现信息密度是“前提条件”而非普通增益——低密度 caption 的增强器甚至比不增强更差。四,给出跨生成器泛化的系统证据:在不接触目标模型训练 caption 的条件下反超其原生增强器,支持“学到的是通用设计原则”这一论断。同期工作 WanPE 虽然也用 caption 构造监督,但面向视频的跨镜头时序规划,与本文的单图精细设计正交。
实验结果
核心结果分五块。一,主基准(Table 1,Qwen-Image-Bench + Qwen-Image-2.1):PE-397B Overall 62.17,比原始请求 54.36 高 7.81 分且五个维度全部第一(Quality 57.95、Aesthetics 64.50、Alignment 62.63、Real-world Fidelity 55.57、Creative Generation 70.89);PE-9B 得 60.28,超过全部评测基线(GPT-5.5 60.10、FLUX 3 Klein PE 59.90、ERNIE-Image-PE 58.70、HunyuanImage-2.1-PE-32B 57.13)。二,精细设计(Table 2):397B 在 Lighting 76.59、Composition 66.66、Detail 58.91 上居首,Layout 61.11 以 0.04 分微逊 GPT-5.5 的 61.15;九个指标中八个第一,UniPercept IAA/IQA/ISTA 达 73.73/76.65/55.12,HPSv3 9.9553、PickScore 23.7541 全面领先;9B 在九个指标上全部超过训练型基线。三,人类评估(Figure 4,500 例 GSB):397B 对各基线的净胜率为对原始请求 +35.48%、对 Hunyuan-PE-32B +32.30%、对 ERNIE +22.49%、对 GPT-5.5 +17.09%、对 FLUX Klein PE +13.60%,全为正。四,缩放(Figure 7):数据 12.5K→25K→50K→100K 使 Overall 55.87→56.68→57.31→57.61,增益 0.81/0.63/0.30 递减;模型 2B→4B→9B→35B→397B 使 Overall 49.56→59.86(+10.30,Creative Generation +19.66、Alignment +14.80);信息密度 L1→L4(GPG 14.42/41.16/87.25/170.36)使 Overall 44.65→51.36→55.39→57.31(+12.66,三轴中最大),且 L1/L2 低于不增强的 54.36 而 L3/L4 反超 1.03/2.95 分——信息密度足够是有效增强的前提。五,泛化(Table 5):四个未见 backbone 上 9B 提升 2.53–6.74 分、397B 提升 3.86–8.03 分,并在 HunyuanImage-2.1(51.66 对原生 PE-32B 的 48.59,+3.07)和 FLUX 3 Klein(59.10 对原生 PE 56.36,+2.74)上反超原生增强器。消融上 CRD 贡献 +1.66、GDR 比 pointwise 高 +0.81(Aesthetics +1.70),RL 过程中 Lighting 从 72.14 稳步升到 77.64。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 综合生成质量(Qwen-Image-Bench,Qwen-Image-2.1 骨干) | Overall | PE-397B 62.17;PE-9B 60.28 | 原始请求 54.36;最佳外部基线 GPT-5.5 60.10 | 对原始请求 +7.81 分;对最佳基线 +0.18 分 |
| 精细设计——光照 | QIB Lighting | 76.59(397B) | 原始请求 64.12;GPT-5.5 71.50 | 对原始请求 +12.47 分,对最强基线 +5.09 分 |
| 精细设计——构图 | QIB Composition | 66.66(397B) | 原始请求 54.68;GPT-5.5 63.88 | 对原始请求 +11.98 分,对最强基线 +2.78 分 |
| 人类偏好(GSB 盲评,500 例) | 净胜率 (G−B)/(G+S+B) | +35.48% | 对原始请求 | 对全部五条基线净胜率均为正(+13.60% 至 +35.48%) |
| 提示词与训练 caption 分布对齐 | MAUVE(越高越好) | 0.414(397B) | Qwen-3.8-Max 0.032 | unigram JS 从 0.229 降至 0.105,bigram JS 从 0.539 降至 0.329 |
| 跨 backbone 泛化(HunyuanImage-2.1) | Overall | PE-397B 51.66 | 原生 HunyuanImage-2.1-PE-32B 48.59;原始请求 44.48 | 对原生增强器 +3.07 分,对原始请求 +7.18 分 |
局限与改进
作者承认或可从数据中读出的局限有:一,Layout 是四个设计维度中最弱的一环——RL 训练曲线(Figure 6)显示 Layout 只小幅改善且波动剧烈,397B 的 Layout 61.11 仍略低于 GPT-5.5 的 61.15,说明版式类目标(对齐、留白、文字层级)的 rubric 或 MLLM 裁判能力仍不足;二,Composition 与 Detail 在 RL 中趋势为正但噪声大,GDR 奖励的方差会传导进训练;三,缩放数据有边际递减迹象,50K→100K 的增益从 0.63 降到 0.30,更大规模数据的收益未验证;四,GDR 与人类盲评的一致率只有 59.1%,仍有约四成分歧,奖励模型偏差客观存在;五,397B 变体是内部模型不开源,开源的只有 9B,公开可复现的最强结果有限;六,泛化实验虽然覆盖四个 backbone,但请求分布以英文场景/商业设计类为主,对插画、漫画等风格化设计的迁移未测试;七,RL 只与 pointwise 比较,exhaustive pairwise 因算力不可行而缺席,也没有测其他 reward 组合或不同 $G$、$K$ 的敏感性。
独立分析的弱点
独立分析出的弱点:一,裁判噪声直接进入策略梯度——$K$ 次洗牌平均只缓解而非消除 MLLM 的位置偏差与随机性,小 $G$ 组内这种噪声会放大 GRPO 的优势估计方差,改进方向是用裁判集成、分数校准,或对组内分数拟合 Bradley–Terry 排序模型替代算术平均。二,RL 提示词池来自“SFT 失败案例挖掘”,覆盖面受挖掘策略偏置影响;论文展示的样例集中在产品摄影、海报排版等商业设计,对艺术风格类(水彩、概念艺术)的设计 rubric 是否同样有效存疑,改进方向是按风格分域构造 rubric 与提示池。三,宽高比预测 $\rho$ 与提示词耦合输出,但论文没有单独报告比例预测准确率及其对 Overall 的贡献消融,无法判断这一输出头的真实价值。四,各维度权重 $\lambda_d$ 与对齐权重 $\lambda_a$ 未公开取值和敏感性分析,复现者只能盲调;对齐项如何在“保留用户要求”和“允许设计自由”之间取舍也缺少定量刻画。五,35B 中间规模只做了 SFT 缩放测试而未做 RL,GDR 的增益随模型规模的演化曲线缺失,无法判断小模型是否也能从组内奖励中获益。六,Layout 停滞提示纯 MLLM 评分对可验证目标不够敏感,可引入 OCR 文本准确性、网格对齐检测等可计算指标做成 hybrid reward。
未来方向
作者指出的方向:把视觉设计确立为 PE 的中心目标,推动社区从整体美学优化转向维度化设计优化(结论章展望);与同期 WanPE 所代表的视频提示词增强合流,把单图设计扩展到跨镜头、时序一致的电影级规划;继续探明信息密度的下限与最优区间,实现 caption 密度的自动调控。基于本文成果可延伸的方向:一,把 GDR 组内比较框架与可验证指标结合(OCR 文字准确性、透视一致性检测、构图黄金比例),构造设计维度上可计算的 hybrid 奖励;二,对数据、模型、信息密度三轴做联合 scaling law 拟合,回答“给定算力预算时三者的最优配比”;三,把 397B 的设计能力蒸馏进 2B–4B 端侧模型(CoT/推理蒸馏),让消费级显卡也能跑设计级增强;四,建立在线学习闭环,把用户对生成图的实际采纳、编辑、重生成行为当作偏好信号继续做 RL,替代人工 GSB 标注;五,把 rubric 扩展到多语言与多文化设计规范(中文海报的字距与竖排、日式留白、阿拉伯文排版),检验“通用设计原则”的边界;六,GDR 的组内互照结构本身可以反哺评估器训练——用组内相对排序做弱监督训练更便宜的设计评估器。
复现评估
开源情况:Qwen-Image-PE-9B 权重已发布在 HuggingFace(Qwen/Qwen-Image-2.1-PE-T2I),可直接加载使用,这是最容易的复现路径。但复现训练的另一切都缺失:397B 权重不开源;SFT 数据集(10 万级 request–caption–rationale 三元组)未发布;RL 提示词池、四个维度的 rubric 全文、裁判所用 MLLM 的身份、$\lambda_d$ 与 $\lambda_a$ 权重、$G$ 与 $K$ 的取值均未完整给出。算力门槛:SFT 阶段需要强 T2I 系统(Qwen-Image-2.1)批量渲染并筛选约 10 万张图再做密集 caption,仅 Request-to-Image 路径的渲染成本就很高;RL 阶段每步要在线生成 $G$ 张 2K 图并调用 MLLM 裁判 $D\cdot K$ 次,397B-A17B 级别的 GRPO 训练需要多机多卡集群,一般实验室难以承担。低成本的验证路径:研究者可以用开源 9B 模型直接复测 Qwen-Image-Bench 式的人工对比;或用公开高密度 caption 数据集加开源小模型复现 CRD 的 SFT 部分;GDR 的核心结论(组内比较比 pointwise 与人类一致率高 8.2 个百分点)可以用 $G=8$、开源 VLM 裁判在小规模上独立检验,这是本文最值得也最容易验证的声明。总体评价:使用容易、训练复现困难、部分结论可低成本抽查。
论文图表
四组前后对比图:每组左边是原始简短请求在 Qwen-Image-2.1 上的输出,右边是经 Qwen-Image-PE 增强后的提示词的输出,分别对应光照、构图、版式、细节四个维度的改进示例。
一图给出论文的问题与承诺:增强前后的直观差距就是全文要解决和量化的东西,读者应先建立对'视觉设计增强'的直觉再进入方法。
(a) 源图→密集 caption→再生图的链条,展示 caption 能传递光照/构图/版式/细节设计;(b) t-SNE 散点图显示 PE-397B 生成提示词的嵌入与 T2I 训练 caption 嵌入重叠度,明显高于 forward 式增强提示词。
这是 CRD 的动机实验:证明'密集 caption 编码了可执行的设计知识'且'反向蒸馏能弥合提示词与训练 caption 的分布鸿沟',是方法合理性的根基。