← 返回 2026-03-13

WeEdit:面向文本中心图像编辑的数据集、基准与字形引导框架 WeEdit: A Dataset, Benchmark and Glyph-Guided Framework for Text-centric Image Editing

Hui Zhang, Juntao Liu, Zongkai Liu, Liqiang Niu, Fandong Meng, Zuxuan Wu, Yu-Gang Jiang 📅 2026-03-12 👍 25 2026-07-13 08:36
图像编辑 多语言 强化学习 扩散模型 数据集构建 文本渲染

用字形引导SFT+多目标RL解决图像内文字编辑难题,330K数据+15语言基准

前置知识

扩散模型与Flow Matching

扩散模型通过逐步向干净数据 $x_0$ 添加高斯噪声得到 $x_t = \alpha_t x_0 + \sigma_t \epsilon$,然后训练神经网络 $v_\theta(x_t, t)$ 来预测速度场(velocity),从而逆转噪声过程生成图像。Rectified Flow 进一步简化为 $\alpha_t = 1-t, \sigma_t = t$ 的直线插值路径,目标速度 $v = \epsilon - x_0$。本文基于 flow-based MM-DiT(即 Qwen-Image-Edit)构建。

本文的整个生成框架建立在 flow matching 之上,SFT 和 RL 阶段都直接操作 velocity 预测,理解这一基础才能看懂训练目标的设计。

LoRA(Low-Rank Adaptation)

参数高效微调方法,通过在预训练权重矩阵旁添加低秩分解矩阵 $\Delta W = BA$(其中 $B \in \mathbb{R}^{d \times r}, A \in \mathbb{R}^{r \times k}, r \ll \min(d,k)$)来实现任务适配,只训练 $A, B$ 而冻结原始权重。本文使用 rank=256 的 LoRA 仅作用于 MM-DiT 的多模态注意力线性层。

理解 LoRA 机制才能理解本文为何能在不大幅增加参数的情况下注入字形先验,以及 SFT 和 RL 阶段的训练效率。

扩散模型的强化学习后训练(RL post-training)

借鉴 LLM 的 RLHF 思路,用奖励模型对扩散模型生成的样本打分,通过策略梯度方法优化生成质量。本文采用 DiffusionNFT 框架,在 flow matching 的正向过程中进行在线 RL,通过隐式正负策略的对比学习来提升生成质量。关键创新在于设计了面向文本编辑的多维奖励函数。

这是本文的核心算法贡献之一,理解 DiffusionNFT 的正负策略对比机制才能看懂 RL 阶段如何将像素级监督提升到语义级质量优化。

字形引导(Glyph Guidance)

将目标文本的渲染图像(白色文字黑色背景)作为显式空间先验注入扩散模型。通过 VLM 检测源图中的文字区域、规划目标文本的位置和内容,然后在空白画布上渲染字形图像。该图像与源图一起编码后拼接输入 MM-DiT,为模型提供精确的字符级空间约束。

这是本文的核心创新点,区别于以往仅在生成阶段使用字形先验的方法,本文通过 SFT 让模型学会利用字形信息,实现了质的飞跃。

VLM-as-Judge 评估

使用大型视觉语言模型(如 Gemini-3-Pro)作为自动评估器,对编辑结果在指令遵循度、文本清晰度、背景保持度三个维度上进行 0-9 分的打分。评估时先生成 chain-of-thought 推理过程再给分,以提高评分的可靠性和可解释性。

本文的基准评估完全依赖这种范式,理解其设计才能正确解读实验结果的含义和局限性。

研究动机

当前领先的图像编辑模型(如 Gemini-3-Pro-Image、GPT-Image-1.5、FLUX.2-dev、Qwen-Image-Edit)在通用物体编辑和风格迁移上表现优异,但在文本中心图像编辑这一关键维度上存在严重短板。具体而言,现有模型在处理图像内文字的修改、翻译、删除、重排等操作时,经常产生模糊、错位甚至幻觉的字符,尤其在非拉丁文字(如阿拉伯语、泰语、印地语)上表现更差。从定量数据来看,在本文提出的基准上,最强的开源基线模型在 Bilingual 基准的整体指令遵循度(IA)仅为 4.16,而在 Rearrange 和 Translate 任务上所有开源基线的 IA 均低于 1.8,几乎完全失败。即使是 GPT-Image-1.5 和 Seedream4.5 等专有模型在这些任务上也出现明显质量下降,说明文本编辑面临的挑战与传统物体/风格编辑有本质区别。

本文的目标是本文旨在构建一个系统性解决方案,从算法、数据、评估三个维度全面攻克文本中心图像编辑的难题。具体目标包括:(1) 设计专门的训练范式,让扩散模型具备精确的文字渲染能力;(2) 构建大规模、高质量、多语言的训练数据集;(3) 建立标准化的评估基准,覆盖多种编辑操作类型和语言,实现闭环的训练-评估体系。

与已有工作不同的是,本文的独特切入角度在于将问题分解为三个相互关联的缺口并逐一填补。在算法层面,现有编辑方法缺乏针对文本内容修改的专门训练范式,本文提出字形引导 SFT + 多目标 RL 的两阶段训练策略,前者注入显式空间先验,后者对齐人类感知质量。在数据层面,现有数据集和基准主要面向通用编辑场景,缺乏针对文本编辑的大规模数据,本文提出基于 HTML 的自动化数据构建流水线,天然支持多语言扩展,构建了 330K 训练样本。在评估层面,本文建立了首个覆盖 8 种编辑操作、15 种语言、三个评估维度的综合基准,填补了该领域缺乏标准化评估的空白。

核心方法

WeEdit 的整体技术路线可以概括为「数据驱动 + 先验注入 + 奖励对齐」的三管齐下策略。直觉上,要让模型学会精确编辑图像中的文字,首先需要大量高质量的训练样本(数据构建流水线),其次需要给模型一个「脚手架」告诉它目标文字应该放在哪里、长什么样(字形引导 SFT),最后需要让模型的输出与人类对「好编辑」的判断对齐(多目标 RL)。整个框架基于 Qwen-Image-Edit(flow-based MM-DiT)构建,通过 LoRA 进行参数高效适配,分两个阶段训练:先用 330K 数据做字形引导的 SFT 8000 步,再用多目标 RL 训练 140 个 epoch。

本文的核心创新在于字形引导机制与以往文本生成方法的本质区别。以往方法(如 GlyphControl、TextDiffuser)主要在图像生成阶段引入字形信息,但没有专门训练模型如何利用这些信息。本文的关键洞察是:仅在推理时提供字形图像是不够的(消融实验证明直接给基模型加字形引导几乎没有提升),必须通过 SFT 让模型「学会」使用字形先验。当 SFT 与字形引导结合时,IA 从 3.58 跃升至 6.99,提升幅度达 95%。另一个核心创新是将 RL 的奖励设计从单一质量分数分解为四个独立评估维度(指令遵循、文本清晰度、背景保持、相对质量),并使用独立的奖励模型分别评估,避免了指标纠缠导致的次优优化。

方法步骤详情

方法分为两大阶段。第一阶段是字形引导监督微调(SFT):给定源图 $I_{src}$ 和编辑指令 $p$,首先使用 Qwen3-VL-235B-A22B-Instruct 进行两步检测-规划流程。检测步骤识别源图中所有需要编辑的文字区域,输出边界框和内容 $\{(b_{orig}^i, t_{orig}^i)\}_{i=1}^m$;规划步骤根据指令确定目标文字内容和空间位置 $\{(b_{tgt}^j, t_{tgt}^j)\}_{j=1}^n$。然后在与源图同尺寸的黑色画布上用白色 Arial 字体渲染字形图像 $x_{glyph}$。训练时,编辑指令、源图、字形图分别通过 VLM 获取语义 token,源图和字形图经 VAE 编码后在 token 维度拼接,与噪声隐变量一起输入 MM-DiT,仅对注意力层的线性层施加 LoRA(rank=256),用 AdamW 优化器学习率 5e-5 训练 8000 步。第二阶段是多目标强化学习:基于 DiffusionNFT 框架,对每个条件输入 $c$(含源图、字形图、指令)采样 $K$ 个候选图像,通过四个独立的 VLM 奖励模型分别评估指令遵循度 $R_{Adherence}^{task}$、文本清晰度 $R_{Clarity}^{task}$、背景保持度 $R_{Preservation}^{task}$ 和相对质量 $R_{Quality}^{task}$。每个维度的奖励通过 logit 加权连续评分机制计算,即对 0-9 分的 logit 做 softmax 后取期望值归一化到 $[0,1]$,综合奖励 $R^{task} = \lambda_{acc}R_{Adherence}^{task} + \lambda_{cla}R_{Clarity}^{task} + \lambda_{pre}R_{Preservation}^{task} + \lambda_{qua}R_{Quality}^{task}$。奖励经组内归一化后得到优化概率 $r$,用于加权正负策略的对比学习目标。

技术新颖性

本文的技术新颖性体现在多个层面。首先,基于 HTML 的数据构建流水线是一个巧妙的设计:将图像转换为 HTML 后,文本编辑变成了简单的字符串替换,再通过 Playwright 确定性渲染回图像,保证非编辑区域像素级一致。这个流水线天然支持多语言——只需先将所有文本翻译为目标语言,再走同样的编辑流程。其次,字形引导的「检测-规划-渲染」流程充分利用了 VLM 的理解和规划能力,不需要额外的 OCR 或布局检测模型。第三,logit 加权连续评分机制避免了单一整数奖励的稀疏性问题,通过利用 VLM 在 0-9 分上的完整 logit 分布获得更平滑的奖励信号。第四,DiffusionNFT 框架中隐式正负策略的设计 $v_{\theta}^+ = (1-\beta)v_{old} + \beta v_{\theta}$ 和 $v_{\theta}^- = (1+\beta)v_{old} - \beta v_{\theta}$,建立了对比改进方向:正分支拉向高奖励生成,负分支推离低奖励生成。

字形引导监督微调阶段概览
Figure 2: 字形引导监督微调阶段概览
强化学习阶段概览
Figure 3: 强化学习阶段概览
数据构建流水线概览
Figure 4: 数据构建流水线概览
WeEdit 数据集统计
Figure 5: WeEdit 数据集统计

实验结果

实验结果表明 WeEdit 在文本中心图像编辑任务上取得了显著突破。在 Bilingual 基准上,WeEdit-RL 的整体得分达到 IA=7.47、TC=8.19、BP=9.01,相比基模型 Qwen-Image-Edit-2509 分别提升 +3.98、+2.35、+2.21,在所有开源模型中排名第一,仅次于专有模型 Gemini-3-Pro-Image(8.58/9.10/8.85)。在更具挑战性的 Multilingual 基准上,WeEdit-RL 达到 6.70/7.10/8.49,同样大幅领先所有开源基线。值得注意的是,文本编辑对现有模型是一个根本性弱点:最强开源基线在 Bilingual 上的 IA 仅为 4.16,而在 Rearrange 和 Translate 任务上所有开源基线的 IA 均低于 1.8,几乎完全失败。WeEdit 在这些困难任务上的 IA 分别达到 7.82 和 6.36,提升幅度巨大。SFT 和 RL 两个阶段的贡献清晰可分:SFT 阶段将 IA 从 3.49 提升到 6.99(+100%),RL 阶段在此基础上进一步将 IA 提升到 7.47、TC 从 7.33 提升到 8.19。消融实验验证了每个组件的必要性:直接给基模型加字形引导几乎无效(IA 从 3.49 到 3.58),但 SFT+字形引导的组合产生巨大协同效应(IA 从 3.58 到 6.99),RL 阶段中移除参考图像会导致评估锚点缺失、使用单一奖励模型会导致指标纠缠,均造成性能下降。用户研究进一步证实了自动评估的结论:WeEdit 在指令遵循度上以 82.0% 的胜率击败 Qwen-Image-Edit-2509,以 80.4% 的胜率击败 FLUX.2-dev,以 45.6% 的胜率击败 GPT-Image-1.5。

Bilingual 基准定量结果
Table 1: Bilingual 基准定量结果
Multilingual 基准定量结果
Table 2: Multilingual 基准定量结果
消融实验
Table 3: 消融实验
WeEdit 编辑效果展示与模型性能对比
Figure 1: WeEdit 编辑效果展示与模型性能对比
WeEdit 基准概览
Figure 6: WeEdit 基准概览
用户研究结果
Figure 7: 用户研究结果
Add 操作定性对比
Figure 8: Add 操作定性对比
Replace 操作定性对比
Figure 9: Replace 操作定性对比
Delete 和 Rearrange 操作定性对比
Figure 10: Delete 和 Rearrange 操作定性对比
Translate 操作定性对比
Figure 11: Translate 操作定性对比
Change Style 操作定性对比
Figure 12: Change Style 操作定性对比
Combined 操作定性对比
Figure 13: Combined 操作定性对比
Reasoning 操作定性对比
Figure 14: Reasoning 操作定性对比
查看结构化数据
任务指标本文基线提升
Bilingual Benchmark - Overall IA / TC / BP (0-9 scale) 7.47 / 8.19 / 9.01 Qwen-Image-Edit-2509: 3.49 / 5.84 / 6.80 +3.98 / +2.35 / +2.21
Bilingual - Rearrange IA / TC / BP 7.82 / 7.94 / 9.00 Best open-source (FLUX.2-dev): 1.44 / 5.73 / 5.84 +6.38 / +2.21 / +3.16
Bilingual - Translate IA / TC / BP 6.36 / 7.11 / 9.44 Best open-source (FLUX.2-dev): 1.24 / 3.59 / 8.78 +5.12 / +3.52 / +0.66
Multilingual Benchmark - Overall IA / TC / BP 6.70 / 7.10 / 8.49 Qwen-Image-Edit-2509: 3.63 / 5.82 / 6.67 +3.07 / +1.28 / +1.82
Bilingual - Add IA / TC / BP 8.12 / 8.99 / 9.40 Best open-source (HY-image-3): 6.04 / 7.55 / 7.68 +2.08 / +1.44 / +1.72
Bilingual - Replace IA / TC / BP 6.90 / 7.77 / 8.63 Best open-source (FireRed-Image-Edit): 4.58 / 6.30 / 6.26 +2.32 / +1.47 / +2.37
User Study - IA Win Rate vs Qwen-Image-Edit-2509 Win / Tie / Lose 82.0% / 15.0% / 3.0% Qwen-Image-Edit-2509 压倒性优势

局限与改进

尽管 WeEdit 取得了显著进展,但仍存在以下局限性。首先,在 Reasoning 任务上表现相对薄弱,Bilingual 基准上 IA 仅为 2.57、TC 为 5.24,远低于其他操作类型,说明模型在需要外部知识推理的编辑场景中能力有限(例如根据事实知识替换体育赛事信息)。其次,依赖 Gemini-3-Pro 作为评估器引入了潜在的评估偏差,VLM-as-Judge 的可靠性在文本编辑这种需要精确字符级判断的任务上可能不够稳定。第三,字形引导机制依赖 Qwen3-VL-235B 进行检测和规划,这增加了推理时的计算开销和延迟,且 VLM 的检测精度直接影响最终编辑质量。第四,数据构建流水线中结构化数据依赖 HTML 渲染,对于复杂视觉风格(如手写体、艺术字)的覆盖有限;非结构化数据则依赖现有编辑模型生成候选,存在质量上限。第五,训练数据中英语(39.1%)和中文(11.3%)占主导,其余 13 种语言各占约 3.5%-7.1%,这种不均衡可能影响低资源语言的编辑质量。

独立分析的弱点

本文有几个值得深入讨论的弱点。第一,Reasoning 任务的性能瓶颈(IA=2.57)表明模型缺乏世界知识推理能力,改进方向可以是将 LLM 的知识检索能力与编辑模型结合,例如先用 LLM 推理出目标文本再执行编辑。第二,推理效率问题:当前流程需要 VLM 做检测规划 + 扩散模型做生成 + RL 阶段的多奖励评估,推理链较长,未来可以探索将检测规划能力内化到编辑模型中,或使用蒸馏方法压缩 VLM 组件。第三,字形渲染使用固定的 Arial 字体和白字黑底格式,无法处理与源图风格匹配的复杂排版需求,改进方向可以是让 VLM 同时预测字体样式参数,或引入字体风格迁移模块。第四,数据构建中非结构化数据依赖现有编辑模型生成候选并通过 VLM 验证,形成了一种「自举」式的数据飞轮,但其质量受限于编辑模型和验证器的上限。第五,RL 阶段的四个奖励维度权重 $\lambda_{acc}, \lambda_{cla}, \lambda_{pre}, \lambda_{qua}$ 是手动设定的超参数,缺乏自适应调节机制。

未来方向

基于本文的成果,有几个有前景的研究方向。首先是扩展到视频文本编辑,将字形引导机制适配到视频扩散模型中,处理视频中文字的添加、替换和翻译。其次是结合 OCR 和字体识别技术,实现更精确的字体风格匹配,让编辑后的文字在字体、颜色、大小上与原图保持一致。第三是探索更强的推理编辑能力,将 LLM 的 Chain-of-Thought 推理能力与图像编辑结合,处理需要世界知识的复杂编辑场景。第四是将 HTML 流水线扩展到更多文档类型(如 PDF、Word),覆盖更广泛的办公场景。第五是研究交互式文本编辑,允许用户通过点击、框选等更自然的方式指定编辑区域和内容。作者也提到该能力在信息图更新、海报修改、多语言 UI 本地化和文档编辑等实际应用中有重要价值,未来可以针对这些垂直场景做深度优化。

复现评估

从复现角度来看,本文的可复现性存在一定的挑战和优势。优势方面:数据构建流水线基于开源工具(BeautifulSoup4、Playwright、Pillow),理论上可以复现;训练基于开源的 Qwen-Image-Edit 模型和标准的 LoRA 微调流程;评估使用 Gemini-3-Pro 作为评判器,API 可获取。挑战方面:核心组件依赖多个大型模型——Qwen3-VL-235B 用于检测规划和奖励评估,Qwen-Image-Edit-2509 作为基座模型,Gemini-3-Pro 作为评估器,这些模型的算力需求巨大;SFT 训练 8000 步 + RL 训练 140 epoch 的总计算量不小;数据构建需要对 330K 样本进行多轮 VLM 推理和浏览器渲染。论文未明确报告 GPU 使用量和训练时间,但从模型规模推断需要多卡集群。项目页面已提供(https://huizhang0812.github.io/WeEdit/),但论文未提及是否开源代码和数据。