近一年,多模态视觉&语言大模型架构演进汇总梳理

共 6343字,需浏览 13分钟

 ·

2024-07-15 22:00




↑ 点击蓝字 关注极市平台





作者丨Dreamweaver


来源丨吃果冻不吐果冻皮


编辑丨极市平台




极市导读

 



本文回顾了多模态LLM (视觉-语言模型) 近一年来的模型架构演进,对其中有代表性的工作进行了精炼总结,截止2024.06。 >>加入极市CV技术交流群,走在计算机视觉的最前沿







这篇综述一张图总结了多模态LLM的典型架构:





BLIP


【2022.01发布】[1]


统一视觉-语言理解和生成,使用captioner+filter高效利用互联网有噪数据





模型架构:


Image/text encoder: ITC loss对齐视觉和语言表征,基于ALBEF提出的momentum distillation


Image-grounded text encoder: ITM loss建模视觉-语言交互,区分positive/negative图文对,使用hard negative mining挖掘更高相似度的负例优化模型


Image-grounded text decoder: LM loss实现基于图像的文本解码,将双向self-attention替换为causal self-attention





BLIP的bootstrapping训练过程:





BLIP-2


【2023.01发布】[2]


使用相对轻量的Q-Former连接视觉-语言模态,通过两阶段训练:第1阶段基于冻住的视觉编码器,第2阶段基于冻住的LLM





第1阶段:同样优化ITC/ITM/LM loss,使用不同的self-attention mask,query和text端共享self-attention参数,使得可学习的query embedding提取与text语义最相关的视觉表征;使用BERT-base初始化,32个768维的query作为信息瓶颈


• ITC:计算每个query与text的相似度,取最大的;使用batch内negatives,不再使用momentum queue


• ITM:对每个query与text的分类logits取平均,使用hard negatives mining挖掘难负例


• LM:text token和frozen image encoder不能直接交互,要求query能提取有益的视觉特征





第2阶段:可基于decoder-only/encoder-decoder LLM进行适配,FC层对齐维度





LLaVA


【2023.04发布】[3]


• 使用仅文本模态的GPT-4生成视觉-语言指令遵循数据,用于微调多模态LLM


• 使用图片的dense captions和bounding boxes作为prompt,可以生成对话、细节描述、复杂推理等指令


• CLIP ViT-L/14 + Vicuna,使用简单的线性层进行映射


• 更复杂的:Flamingo中gated cross-attention,BLIP-2中的Q-former








• LLaVA模型的两阶段训练


• stage1. 预训练特征对齐:冻住vision encoder和LLM,只训练projection,学习一个兼容的visual tokenizer


• stage2. 端到端微调:冻住vision encoder,在单轮/多轮对话数据上微调projection和LLM


MiniGPT-4


【2023.04发布】[4]


stage1. 预训练:使用image-text pair微调linear projection layer,vision encoder和LLM保持冻住


stage2. 指令微调:指令格式为:###Human: ###Assistant:





InstructBLIP


【2023.05发布】[5]


stage1. 预训练:BLIP-2(使用image-text pairs进行两阶段训练)


stage2. 指令微调:只微调instruction-aware Q-former,冻住vision encoder和LLM


支持FlanT5(encoder-decoder)和Vicuna(decoder-only)





Qwen-VL


【2023.08发布】[6]


支持中英双语、多图像输入


Qwen-7B + OpenCLIP ViT-bigG,输入图像直接resize到视觉编码器输入


位置感知的VL adapter:使用基于Q-former的单层的cross-attention,将图像特征维度压缩到256,在query-key pairs中引入2D绝对位置编码增强位置信息


图像输入:256-dim图像特征


bounding box输入输出:
(X_topleft, Y_topleft), (X_bottomright, Y_bottomright)

,


标记box所指内容


三阶段训练:


stage1. 预训练:基于大规模、弱标注、网络爬取的图像-文本对,输入分辨率224x224,冻住LLM,训练ViT和Q-former,主要目的是模态对齐


stage2. 多任务预训练:基于7种下游视觉-语言理解任务的高质量、细粒度标注数据训练,输入分辨率448x448,图像/文本数据交错,训练整个模型


stage3. 指令微调:提升指令遵循和多轮对话能力,冻住ViT,训练LLM和Q-former





Qwen-VL-Plus和Qwen-VL-Max提升了视觉推理能力、图像细节的识别/提取/分析能力(尤其是文本导向的任务)、支持高分辨率和极端纵横比的输入图像;在部分中文场景超过了GPT-4V和Gemini


InternLM-XComposer


【2023.09发布】[7]


交错图文构成:自动在输出文本中插入合适的图片


EVA-CLIP ViT + InternLM-7B + Q-former (将图像特征压缩到64个embedding)


两阶段训练:


stage1. 预训练:冻住ViT,训练LLM和Q-former


stage2. 监督微调:包括多任务训练和指令微调,冻住ViT和LLM,训练Q-former,对LLM进行LoRA微调,增强指令遵循和图文混排能力





Fuyu-8B


【2023.10发布】[8]


模型架构和训练过程简单,易于scaling;支持任意图像分辨率;推理速度快


decoder-only的transformer,没有专门的图像编码器;image patch直接线性映射到transformer第一层





LLaVA-1.5


【2023.10发布】[9]


仍使用MLP作为模态连接,突出了训练的数据高效性





CogVLM


【2023.11发布】[10]


深度视觉-语言模态融合,而不影响LLM原有的语言能力:冻住LLM和ViT,在attention和FFN层训练一份视觉专家模块





CogAgent


【2023.12发布】[11]


针对GUI场景的多模态理解和导引,使用高分辨率-低分辨率双编码器,支持1120x1120的屏幕输入


高分辨率分支使用更轻量的ViT,基于cross-attention将高分辨率图像特征与LLM每层进行融合





VILA


【2023.12发布】[12]


探索了视觉-语言模型训练的设计选择:


• 预训练阶段冻住LLM虽然能取得较好的zero-shot性能,但上下文学习能力依赖对LLM的微调


• 图文交错的预训练数据是有益的,只用图文数据对效果不够好


• 将纯文本的指令微调数据加入SFT阶段有助于缓解纯文本任务的能力退化,同时也能够增强视觉-语言任务的准确性





LLaVA-Next


【2024.01发布】[13]


相对于LLaVA-1.5,保持了极简的设计和数据高效性:


• 提高了输入图像的分辨率 (4x),支持3种纵横比:672x672, 336x1344, 1344x336


• 更好的视觉推理和OCR能力:更好的指令微调数据配比


• 更好的多场景视觉对话:更好的世界知识和逻辑推理


• 更高效的部署和推理:SGLang


动态高分辨率:视觉编码器支持336x336的图像输入,对于672x672的图像,按照{2,2}的grid split成4个图像patch过encoder,downsample到336x336也过encoder,特征拼接作为visual tokens输入到LLM中





收集高质量用户数据,包括真实场景中反映用户更广泛意图的指令数据,利用GPT-4V进行数据构造


多模态文档/图表数据,增强文档OCR和图表理解能力





InternLM-XComposer2


【2024.01发布】[14]


提出了新的模态对齐方法partial LoRA:只在image token上添加LoRA参数,保证预训练语言知识的完整性,这样一个更轻量的视觉编码器同样有效


OpenAI CLIP ViT-L/14 + InternLM2-7B + partial LoRA (rank=256)








两阶段训练:


stage1. 预训练:冻住LLM,微调ViT和partial LoRA模块,包括通用语义对齐(理解图像基本内容)、世界知识对齐(进行复杂的知识推理)、视觉能力增强(OCR、物体定位、图表理解)


stage2. 监督微调:微调整个模型,包括多任务训练、自由形式图文排布


InternLM-XComposer2-4KHD


2024.04发布了4KHD版本:https://arxiv.org/abs/2404.06512


支持动态分辨率(336px → 4K (3840x1600)):改进了patch division范式,保持训练图像原有的纵横比,自动变化patch数目,基于336x336的ViT配置layout


动态图像划分:将输入图像resize and pad到336的整数倍宽高


结合图像的global和local视角:global视角由输入直接resize到336x336,使用sep token分隔两种视角的token


图像2D结构的换行符:可学习的\n token分隔图像token行





Mini-Gemini


【2024.03发布】[15]


使用双视觉编码器提取低分辨率embedding作为query,高分辨率特征区域作为key/value,两者之间做cross-attention,输出挖掘的tokens作为prompt前缀,输入到LLM做推理,外接图像解码器生成图像(SDXL)





引用链接


[1] 【2022.01发布】: https://arxiv.org/abs/2201.12086


[2] 【2023.01发布】: https://arxiv.org/abs/2301.12597


[3] 【2023.04发布】: https://arxiv.org/abs/2304.08485


[4] 【2023.04发布】: https://arxiv.org/abs/2304.10592


[5] 【2023.05发布】: https://arxiv.org/abs/2305.06500


[6] 【2023.08发布】: https://arxiv.org/abs/2308.12966


[7] 【2023.09发布】: https://arxiv.org/abs/2309.15112


[8] 【2023.10发布】: https://huggingface.co/adept/fuyu-8b


[9] 【2023.10发布】: https://arxiv.org/abs/2310.03744


[10] 【2023.11发布】: https://arxiv.org/abs/2311.03079


[11] 【2023.12发布】: https://arxiv.org/abs/2312.08914


[12] 【2023.12发布】: https://arxiv.org/abs/2312.07533


[13] 【2024.01发布】: https://llava-vl.github.io/blog/2024-01-30-llava-next/


[14] 【2024.01发布】: https://arxiv.org/abs/2401.16420


[15] 【2024.03发布】: https://arxiv.org/abs/2403.18814









公众号后台回复“数据集”获取100+深度学习各方向资源整理












极市干货









技术专栏:多模态大模型超详细解读专栏搞懂Tranformer系列ICCV2023论文解读极市直播


极视角动态欢迎高校师生申报极视角2023年教育部产学合作协同育人项目新视野+智慧脑,「无人机+AI」成为道路智能巡检好帮手!


技术综述:四万字详解Neural ODE:用神经网络去刻画非离散的状态变化transformer的细节到底是怎么样的?Transformer 连环18问!




点击阅读原文进入CV社区


收获更多技术干货







浏览 148
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报