Text to image论文精读CogView: Mastering Text-to-Image Generation via Transformers(通过Transformer控制文本生成图像)

2022-08-23 832

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： CogView是清华大学和阿里巴巴达摩院共同研究开发的一款用Transformer来控制文本生成图像的模型。该论文已被NIPS（Conference and Workshop on Neural Information Processing Systems，计算机人工智能领域A类会议）录用，文章发表于2021年10月。论文地址：https://arxiv.org/pdf/2105.13290v3.pdf代码地址：https://github.com/THUDM/CogView本博客是精读这篇论文的报告，包含一些个人理解、知识拓展和总结。

CogView是清华大学和阿里巴巴达摩院共同研究开发的一款用Transformer来控制文本生成图像的模型。该论文已被NIPS（Conference and Workshop on Neural Information Processing Systems，计算机人工智能领域A类会议）录用，文章发表于2021年10月。

论文地址：https://arxiv.org/pdf/2105.13290v3.pdf

代码地址：https://github.com/THUDM/CogView

本博客是精读这篇论文的报告，包含一些个人理解、知识拓展和总结。

一、原文摘要

文本到图像的生成在一般领域一直是一个开放的问题，这需要一个强大的生成模型和跨模态的理解。我们提出了CogView，一个带有VQ-VAE标记器的40亿参数变压器来解决这个问题。我们还展示了各种下游任务的微调策略，例如风格学习、超分辨率、文本图像排名和时装设计，以及稳定预训练的方法，例如消除NaN损失。CogView在模糊的MS COCO数据集上实现了最先进的FID，优于之前基于GAN的模型和最近的类似工作DALL-E。

二、为什么提出CogView

2.1 文本生成图像的任务难度

目前的各种文本生成图像任务，我们期望模型具有

（1）从像素中分离形状、颜色、手势和其他特征；

（2）理解输入文本；

（3）将物体和特征与对应的单词及其同义词对齐；

（4）学习复杂的分布，以生成不同物体和特征的重叠和组合。

显然这些要求已经超出了基本的视觉功能，模型需要拥有更高水平的认知能力。

2.2 现有模型的不足

（1）基于GAN的模型可以在简单和特定领域的数据集中进行合理的合成，但在复杂场景中其效果远不如人意。

（2）基于Transformer的GPT模型虽然在自然语言生成和语言理解方面取得很大进展，但是无法承受图像生成的计算量（即使是最大的ImageGPT也仅仅达到96*96分辨率）

（3）VQ-VAE（Vector Quantized Variational AutoEncoders）框架训练一个编码器将图像压缩到一个低维离散潜在空间重振了 CV中的自回归模型，但是未曾在文本生成图像领域有所应用。

CogView就是使用VQ-VAE对文本（中文）和图像进行大规模生成性联合预训练，并通过提出的 Precision Bottleneck Relaxation 和 Sandwich Layernorm 解决大规模的文本到图像生成预训练中的不稳定问题，实现文本生成图像的任务的模型。

三、模型结构

3.1 框架结构

在这里插入图片描述

CogView整体思路为：

首先将文本部分转换成token，利用的是已经比较成熟的SentencePiece模型；
然后将图像部分通过一个离散化的AE(Auto-Encoder)转换为token
文本token和图像token拼接到一起，之后输入到GPT模型中学习生成图像。
训练后，在处理文本图像生成类任务时，模型会通过计算一个Caption Score对生成图像进行排序，从而选择与文本最为匹配的图像作为结果。

3.2 理论推导

模型对数似然和置信下界（ELBO）为：

或者可以写为：

符号解释：

不能识别此Latex公式:
{(\mathbf{X}, \mathbf{T})=\left{x_{i}, t_{i}\right}_{i=1}^{N}}

为数据集，其由N对独立同分布的图像变量x及其描述文本变量t的样本组成；
假设图像x可以通过涉及潜在变量z生成：首先由先验P(t;θ)生成，由生成 ,由生成 ;
ϕ为编码器，ψ为解码器；
是编码器ϕ的输出，为变分分布；
学习过程：
学习编码器ϕ 和解码器ψ 最小化重构损失（reconstruction loss）；
GPT通过学习token化的文本和序列最小化两个负对数似然损失（NLL loss for text 和NLL loss for z）

理论推导看不懂暂时也没关系。总之要知道：整个模型分成两个阶段

第一个阶段在做Tokenization（文本的tokenizer已经成熟，忽视他的重构损失，而图像的tokenizer是一个离散化的自编码器，其重构损失reconstruction loss为整体损失函数的第一个部分）

第二个阶段是一个transformer，用的是目前最强大的GPT模型，其输入为第一阶段文本和图像的Tokenization序列，其目的是最小化另一个损失（NLL loss）

3.3 第一阶段：Tokenization

3.3.1 文本的tokenizer

关于文本的tokenizer已经有很好地研究，在CogView中，使用的是SentencePiece，基础是一个大型中文语料库。

3.3.2 图像的tokenizer

在这里插入图片描述

图像的tokenizer是离散化的自编码器，类似VQ-VAE。

编码器 ϕ 将维度为 H×W×3 的图像x映射到一个维度为 h×w×d 的Encoder中Enc（x）中，然后将每个d维向量量化为codebook ，量化的结构可以由h×w的嵌入索引来表示，最终编码器编码成。解码器ψ 将量化向量映射回（模糊的）图像以重建输入。

文章在分析了四种图像的tokenizer方法后，选择了由VQ-VAE思想转化来的最近邻映射的直通估计器，这种方法的一个问题就是当codebook太大时，会出现维数灾难，不过本文未碰见这种情况。

在这里插入图片描述

（四种图像的tokenizer方法均能收敛到相似的水平）

3.4 第二阶段：Auto-regressive Transformer

CogView的主网络是单向Transformer（GPT）。Transformer有48层，隐藏的大小为2560，40个注意力头和40亿个参数。

在这里插入图片描述

如图所示，在每个序列中添加四个分隔符标识，指示文本和图像的边界。

[RO|1] ：标识对应文本的开头
[BASE]：标识对应文本的结束
[BO|1]：标识图像的开头
[EO|1]：标识图像的结束

预训练的pretext任务是从左到右预测字符，也称语言建模，在本模型中，文本和图像都以同样方式训练预测，在实验中，本文发现文本建模是文本到图像预训练成功的关键所在，如果文本标记的权重设置为零，模型将无法找到文本和图像之间的连接，并生成与输入文本完全无关的图像。

本文假设文本建模将知识抽象在隐藏层中，并在之后有效利用这些知识进行图像建模。

四、训练过程的维稳

在16位精度下，text-to-image任务预训练会非常不稳定，保持训练的稳定是CogView最具有挑战的部分。在分析模型训练后，发现有两种不稳定性：溢出（NAN loss）和下溢（loss 不收敛），因此提出以下维稳技术：

4.1 Precision Bottleneck Relaxation (PB-Relax)

在分析了训练的动态性之后，作者发现溢出（NAN loss）总是发生在两个瓶颈操作上，即最后一层LayerNorm或注意层。

在深层网络中，LayerNorm中输出的值可能会爆炸到10000+，导致溢出。解决的方法是令LayerNorm(x)=LayerNorm(x/max(x))，即通过除以x的最大值来减小爆炸。
注意力分数可能明显大于输入元素,故将计算顺序改为，又由于，故将注意力变更为：。如此，注意力分数被除以α来防止溢出。

4.2 Sandwich LayerNorm (Sandwich-LN)

Transformer中的 LayerNorm对于稳定训练至关重要。LayerNorm的输出为 ,基本上与x的隐藏维度大小的平方根成比例,但有些维的输入值明显大于其他维，会导致对应维的输出值加大，在残差分支中，这些较大的值被放大并放回到主支中，加剧transformer层的现象，最后导致深层的value explosion。