基于Vicuna-1.5 7B
丰色 发自 凹非寺
量子位 | 公众号 QbitAI
字节&复旦大学多模态理解大模型来了：
可以精确定位到视频中特定事件的发生时间。
比如在下面这个视频中：
狗子转身看镜头时的时间戳是多少？
什么时候用爪子推开滑板？
在这里，视频中的宝宝什么时候推起眼镜、舒展了一下身体？又是什么时候翻的书？
对于这样的问题，这个叫做LEGO的模型全都读得懂，并毫不犹豫给出正确答案。
看起来，有了这些研究成果，以后我们看视频查资料都要方便一大截咯？
可精确识别局部信息的多模态LLM来了
LEGO全称是一个语言增强的多模态grounding模型。
它主要解决的是多模态LLM跨多种模态进行细粒度理解的能力，此前业内的成果主要强调全局信息。
为了实现该目标，作者主要先从数据集下手，打造了一套用于模型训练的多模式、多粒度问答形式数据集（即将开源）。
该数据集的构建涉及两个关键流程。
一是数据集转换（Dataset Conversion）。
在这个阶段，作者的目的是构建用于模态对齐和细粒度对齐的基础多模态数据集。
由于数据集质量相对较低，主要通过转换公开数据集获得。
如下图上部分所示，他们向GPT-3.5提供任务描述以生成特定于任务的问题库，最终生成单轮对话格式的问答对。
生成的数据集会进行过滤以确保其质量。
其中对于图像模态，作者利用LLaVA-pretrain595K数据集进行模态对齐，细粒度对齐则使用特定数据集如RefCOCO。
视频模态用Valley-Pretrain-703K进行模态对齐，Charades-STA数据集用于细粒度对齐。
二是指令调整数据集生成（Instruction-tuning Dataset Generation）。
这个数据集的目的是让模型更好地理解和遵循人类指令。
如上图下部分所示，作者也选择了公开可用的数据集（Flickr30K Entities、VCR、DiDeMo等）的子集进行人工注释，以创建上下文示例。它用于指导GPT-3.5在生成指令调整数据集时遵循类似的模式。
随后，特定任务的系统提示和随机选择的示例被输入到GPT-3.5中，以生成单轮或多轮对话。最后，进行数据过滤以确保数据集质量。
下面是经过三阶段训练产生的最终数据样本示例：
下面是LEGO模型的架构：
每个模态的输入通过独立的编码器进行处理，提取特征，然后使用适配器将这些特征映射到LLM的嵌入空间。
图中演示的是视频和图像模式的两个示例，蓝色方框表示视频作为输入，而黄色方框表示图像作为输入。
由于其基于模块化设计和适配器的架构，LEGO可以无缝集成新的编码器，处理额外的模态，如点云和语音，主打一个好扩展。
最后，LEGO使用Vicuna1.5-7B作为基础语言模型，训练由三个阶段完成：多模态预训练，细粒度对齐调整和跨模式指令调整。
下面是实验评估：
图像任务中，LEGO模型和其他模型在REC任务中的性能如下表所示，可以看到它在所有数据集上都表现出了比较有竞争力的性能。
视频任务中，由于LEGO侧重对于整个视频的理解，相比VideoLLaMA、VideoChat和Valley这三个模型，性能表现相当优异：
更多能力展示
如上所说，LEGO的能力不仅在于视频定位，对图片、音频等多模态任务都很在行。
指的就是以下这些：
- 图像内容解读
在这张风景图中，它准确给出了游玩风险提示。
在这个meme图中，它也准确发现这是一个炸鸡拼成的简单地图。
- 视频内容概括简介
可以看到它能识别出非常细节的城市坐标和景点。
- 音频解析
当然，这里测试的只是一个比较简单的纯雨声短音频。
- 声音定位
给一段狗叫音频+一张狗狗奔跑的图像，它可以准确圈出声音来源在狗嘴部。
作者介绍
本文一共12位作者。
除了一作Zhaowei Li来自复旦大学，还有一位叫做的Dong Zhang的也来自这里。
其余均为字节跳动员工，通讯作者为Tao Wang。
论文地址：
arxiv.org/abs/2401.06...