总结
BLIP2
论文地址:https://https://arxiv.org/pdf/2301.12597
发布时间:2023.06.15
模型结构:
Vision Encoder:ViT-L/14
VL Adapter:Q-Former
LLM:OPT (decoder-based),FlanT5(encoder-decoder-based)
论文主要提出Q-Former(Lightweight Querying Transformer)用于连接模态之间的gap。BLIP-2整体架构包括三个模块:视觉编码器、视觉和LLM的Adapter(Q-Former)、LLM。其中Q-Former是BLIP-2模型训练过程中主要更新的参数,视觉Encoder和大语言模型LLM在训练过程中冻结参数。
BLIP-2的预训练包括两个阶段:
Stage 1)Vision-and-Language Representation Learning. Q-Former与冻结的Image Encoder(ViT-L/14)连接,在和文本交互中学习图文相关性表示(3个预训练任务)。
Stage 2)Vision-to-Language Generative Learning. 第一个阶段训练得到的Q-Former的输出接入一个大语言模型,学习视觉到文本生成(1个预训练任务)。
下面分别介绍两个阶段:
阶段1:Q-Former的结构如上图所示,包括两个Transformer子模块(共享Self-Attention层),一个image transformer与image encoder交互提取视觉表征(图中黄色区域左侧),一个text transformer既作为text encoder也作为text decoder(图中黄色区域右侧)。首先创建一个可学习的query向量(Learnable query embeddings)作为image transformer的输入,queries通过self-attention层进行自我交互,然后与冻结参数的image features(来自image encoder)通过cross-attention层(插入每隔一个block)进行交互,此外这个query向量还与text通过同一个self-attention进行交互。在不同的预训练任务中,使用不同的self-attention masks来影响查询文本的交互,在这个阶段的预训练中,一共通过三个任务进行学习(类似BLIP),分别为:
ITM:图文匹配任务,使用双向self-attention mask, 不进行掩码,该任务目标是学习细粒度的图文表示对齐。训练过程中queries和text可以完全互相看到,query embeddings的输出Z包含了多模态的信息,将它输入一个2分类linear层得到一个logit,平均来自所有queries的logits作为最终的匹配分数,论文采纳了难负样例挖掘的策略创建更具信息量的负样本对。
ITG:图引导文本生成,使用causal self-attention mask, 由q-former的架构可以看到,冻结的image encoder没有和text tokens直接交互,而生成(generation)所需要的信息必须首先通过queries提取视觉信息以及通过self-attention传给text tokens,因此queries被迫学习抽取可以描述文本信息的视觉特征。论文使用multimodal causal self-attention mask来控制query和text的交互(同UniLM中的使用),queries可以看到自己但看不到text tokens,text tokens可以看到queries以及已经生成的text tokens, 用[DEC] token取代[CLS] token作为第一个text token来指示解码任务的开始。
ITC:图文对比学习,使用单模态self-attention mask, 对text的token全部进行掩码,使得queries和text互相看不到,这个任务对齐的是来自image transformer的query表示和来自text transformer的文本表示([CLS] token)。
在论文实验中,选用两种视觉编码模型 1)CLIP预训练的ViT-L/14. 2)EVA-CLIP预训练的ViT-g/14. 并移除ViT的最后一层,只使用倒数第二层的输出特征(实验中效果更好),与queries交互。Learned Queries侧使用32个query,每个query维度768,q-former输出维度为32*768,这样比冻结的image features维度要小很多(比如对于使用ViT-L/14来说是257*1024),因此第一个阶段主要目标是训练queries可以提取蕴含语义信息的视觉表示。
阶段2:Q-Former通过一个FC层使得Q-former的输出query embedding Z对齐到与LLM输入同样的维度,并前置于input text embedding一起输入LLM,这种拼接方式有点像软视觉提示(soft visual prompt),q-former在前面预训练任务中已经学到如何抽取蕴含语言信息的视觉表示,该阶段中可以起到把最重要信息输入给LLM同时去除错误没有意义的视觉信息的作用,从而降低LLM学习视觉语言对齐的负担,同时也缓解了灾难遗忘的问题。
论文实验了两种LLM,对于decoder-based LLM,预训练使用language modeling loss. 对于encoder-decoder-based LLM,预训练使用prefix language modeling loss, 将text分成两部分,前一部分与q-former输出拼接一起作为编码器的输入,后面部分作为解码器的生成目标。
InstructBLIP
论文地址:https://arxiv.org/pdf/2305.06500 (https://arxiv.org/pdf/2305.06500)
发布时间:2023.06.15
模型结构:
Vision Encoder:ViT-g/14
VL Adapter:Q-Former
LLM:FlanT5-xl(3B), FlanT5-xxl(11B), Vicuna-7B, Vicuna-13B
InstructBLIP的模型结构与BLIP-2类似,区别在于输入文本换成了指令数据Instructions. Q-Former抽取指令感知的视觉特征(Instruction-aware vision model),根据指令的不同获取不同的视觉特征。然后将这些视觉特征作为LLM的软视觉提示(soft prompt),使用language modeling loss和指令微调模型生成回复。
训练过程(Vision-Language Instruction Tuning):3阶段训练以及zero-shot预测
Stage 1:预训练,训练Q-Former和Projection Layer,冻结image encoder。使用image caption数据,学习视觉文本相关性表示。
Stage 2:预训练,训练Projection Layer,冻结LLM。使用image caption数据,学习对齐LLM的文本生成。
Stage 3:指令微调,训练Q-Former和Projection Layer。使用Instruction任务数据,学习遵循指令生成回复的能力。
训练数据:收集11个任务以及相应的26个数据集,如下图所示。对于每个任务,人工编写10-15个自然语言的指令模版,作为构造指令微调数据的基础。对于偏向较短回复的开源数据集,在指令模版中使用'short/briefly'降低模型过拟合为总是生成较短回复(防止过拟合的方式是在指令中有所体现)。
Qwen-VL
论文地址:https://arxiv.org/pdf/2308.12966
发布时间:2023.10.13
模型结构:
Vision Encoder:ViT-bigG/14
VL Adapter:a single-layer cross-attention(Q-former的左侧部分)
LLM:Qwen-7B
VL Adapter创建一组可训练的queries向量和image features一起做cross-attention,将视觉特征压缩至256的固定长度,同时为了提升细粒度的视觉理解,在cross-attention中也加入图像的2D绝对位置编码。
Image Input使用特殊token(<img> and </img>)分隔,Bounding Box Input使用特殊token(<box> and </box>)分隔,bounding box的content referred使用特殊token(<ref> and </ref>)分隔。
训练过程:
Stage 1:预训练,训练Cross-Attention和ViT,冻结QwenLM。
Stage 2:多任务预训练(7 tasks同时),全参数训练。
Stage 3:指令微调,训练Cross-Attention和QwenLM,冻结ViT。
训练数据:
第一个阶段使用image-text pairs数据,77.3%英文、22.7%中文,一共14亿数据训练,图片size=224*224.
第二个阶段使用质量更高的image-text pairs数据,包含7个任务,图像size=448*448. 在同一个任务下构造交错图像文本数据,序列长度为2048. 训练目标与Stage1一致。
第三个阶段使用Instruction数据,训练指令遵循和对话能力,通过LLM self-instruction构造,一共350k条。
Qwen2-VL
论文地址:https://arxiv.org/pdf/2409.12191
发布时间:2024.09.18
模型结构:
Vision Encoder:ViT/14
VL Adapter:Cross-Modal Connector
LLM:Qwen2-1.5B, Qwen2-7B, Qwen2-72B
Qwen2-VL相较于Qwen-VL的主要改进点(除了一些VQA等基础能力的提升之外):
1)支持视频理解,支持context上下文长度到128k token(20分钟左右视频)。
2)Visual Agent能力,支持实时视频对话。
3)图像位置编码采用2D-RoPE,一张224*224分辨率的图像经过ViT/patch_size=14等一系列转换之后会被压缩至66个token输入到LLM。
训练过程:
Stage 1:训练ViT,使用大量image-text对。
Stage 2:全参数微调,使用更多的数据提升模型全面理解的能力。
Stage 3:指令微调,训练LLM。
LLaVA
论文地址:https://https://arxiv.org/pdf/2304.08485
发布时间:2023.12.11
模型结构:
Vision Encoder:ViT-L/14
VL Adapter:/
Projection Layer:a linear layer
LLM:LLaMA
训练过程:
Stage 1:Pre-training for Feature Alignment. 训练Projection Layer
Stage 2:Fine-tuning End-to-End. 训练Projection Layer和LLM
LLaVA-1.5
论文地址:https://arxiv.org/pdf/2310.03744
发布时间:2024.05.15
模型结构:
Vision Encoder:Clip预训练 Vit-L/336px
VL Adapter:MLP
LLM:Vicuna v1.5 13B
MiniGPT-4
论文地址:https://arxiv.org/pdf/2304.10592
发布时间:2023.10.02
模型结构:
Vision Encoder:ViT-G/14
VL Adapter:Q-Former
Projection Layer:a single linear
LLM:Vicuna
训练过程:
Stage 1:只训练Linear Projection Layer来对齐视觉特征和大语言模型。使用大量text-image pair数据。
Stage 2:指令微调,使用少量高质量text-image instruction数据
指令模板:###Human: <Img><ImageFeature></Img><Instruction>###Assistant:
MiniGPT-v2
论文地址:https://arxiv.org/pdf/2310.09478
发布时间:2023.11.07
模型结构:
Vision Encoder:ViT
VL Adapter:/
Projection Layer:Linear
LLM:Llama2-7B
训练过程:
Stage 1:预训练,使用大量弱监督image-text和细粒度数据集的混合数据训练,让模型获取多样化知识
Stage 2:多任务训练,只使用细粒度高质量数据集训练模型在不同任务上的能力。
Stage 3:多模态质量微调,让模型具备Chat哪里
训练数据: