研究导读语言模型与结构化学习
← 全部导读

从标记语言到图像的扩散模型

学习渲染 LaTeX、SMILES、HTML 与乐谱

Yuntian Deng、Noriyuki Kojima 和 Alexander M. Rush 的 Markup-to-Image Diffusion Models with Scheduled Sampling 学习把符号描述转化为图像。论文于 2022 年 10 月 11 日发布,发表于 ICLR 2023,提出了覆盖 LaTeX 公式、SMILES 分子、HTML 表格布局和 LilyPond 乐谱的扩散渲染方法。它也研究了扩散模型中的一个普遍问题:采样过程中遇到的状态与训练输入不同时,错误会逐步累积。

原始研究

Markup-to-Image Diffusion Models with Scheduled Sampling
Yuntian Deng, Noriyuki Kojima, Alexander M. Rush · ICLR 2023

动手理解一个例子

四类结构化输入与图像输出

原始任务学习从标记语言到渲染图像的映射。

数学公式

\frac{a}{b} + c^2

LaTeX → 公式图像

分子结构图

CC(=O)O

SMILES → 分子图像

表格布局

<table>…</table>

HTML → 表格图像

乐谱

{ c'4 d' e' f' }

LilyPond → 乐谱图像

打开 LaTeX 渲染演示 ↗

四个输入示例用于说明任务格式。生成结果与实验见论文原文和代码仓库。

输入是标记语言,输出是图像

学习式渲染器接收描述视觉对象的序列,预测渲染后的像素。LaTeX 描述数学记号,SMILES 描述原子与化学键,HTML 描述表格类布局,LilyPond 描述乐谱。这四个领域共同检验模型对符号内容、布局,以及跨越输入序列的依赖关系的处理能力。

图像到 LaTeX 的识别任务从图片恢复源代码;标记语言到图像的生成任务则朝相反方向进行。论文调整了包括 im2latex-100k 在内的配对数据,学习从标记语言到像素的正向映射。它训练的是条件扩散模型,推理时不调用传统渲染引擎。

来源: Markup-to-Image Diffusion Models with Scheduled Sampling

图像看起来像,并不等于符号正确

这些图像要求精确保留符号。多一个减号就会改变公式,化学键错误就会改变分子,音符移位就会改变乐谱。即使图像的整体外观符合相应领域,只要符号与输入不一致,渲染就失败了。布局错误还会传播:多出的符号会挤动后续内容。

因此,论文评估生成结果与参考渲染图像的对齐程度。主要指标把二值图像视为由图像列组成的序列,使用动态时间规整(dynamic time warping),以容纳小幅位移。像素误差和其他图像相似性指标提供补充测量。论文明确区分了这些指标与侧重视觉逼真度和宽泛语义的自然图像指标。

来源: Markup-to-Image Diffusion Models with Scheduled Sampling

让去噪器在自己产生的中间状态上训练

标准扩散训练从真实图像构造带噪输入;采样时,下一个去噪步骤接收的却是前面模型预测得到的结果。预测中的错误会改变后续步骤遇到的状态。论文把这种不匹配分析为一个错误会累积的序贯决策问题,将扩散采样与暴露偏差联系起来。

Scheduled Sampling 改变训练输入,让去噪器接触模型自己产生的状态,并学习修正自身错误。它将此前在序列生成中研究的原则用于扩散训练。实验在四个渲染领域中比较了 Scheduled Sampling 与基线。输出是否可辨认、符号是否忠实于输入,是两项不同的评估目标;论文同时报告了生成成功与失败的例子。

来源: Markup-to-Image Diffusion Models with Scheduled Sampling

查看原始示例与实现

开源仓库提供了公式、表格、乐谱和分子的并排动画,对照 Scheduled Sampling、基线和真实图像。这些示例展示了中间去噪轨迹。可以检查符号何时出现、消失、合并或移动,再把最终输出与精确的输入描述比较。

设计新的学习式渲染任务时,先确定配对的源数据与目标图像,识别哪些符号错误会改变含义,再选择能够暴露这些错误的指标。将渲染准确性与风格相似性分开。已发布的资源包括原始数据集、检查点、代码和记录的示例。

来源: Markup-to-Image Diffusion Models with Scheduled Sampling

论文与实现

ICML 2023

Input Perturbation Reduces Exposure Bias in Diffusion Models

Mang Ning, Enver Sangineto, Angelo Porrello, Simone Calderara, Rita Cucchiara

研究扩散模型采样过程中累积的错误,并在训练中扰动输入,以模拟推理时的预测误差。

通过合成输入扰动处理同一种训练与采样不匹配问题;Scheduled Sampling 则在模型生成的状态上训练。