多模态大模型的grounding能力

发布时间 2023-10-26 11:19:13作者: shiiiilong

数据集

a)QW-VL:Visual Genome, RefCOCO, RefCOCO+, RefCOCOg,
b)CogVLM:Visual7W,Flickr30K-Entities
c)Kosmos2:GRIT

OFA

Unifying Architectures, Tasks, and Modalities Through a Simple Sequence-to-Sequence Learning Framework
将多模态任务统一为seq2seq,最大模型900M

文本,图片,物体离散到统一的词表中:
将文本用BPE转化为subwords,将图片简单切分成多个patch并使用image quantization转化为image code,抽取图片中的物体的标签和bounding box并将bounding box离散化为location tokens。统一词表是文本的subwords,图片的image code和物体的location tokens三者的并集。
box表示方式:将坐标映射到1-1000,对应词表中总共1000个location token,一个box即<x1><y1><x2><y2>

KOSMOS-2

KOSMOS-2: Grounding Multimodal Large Language Models to the World

kosmos-2的一个重要贡献是解锁了MLLM的grounding能力

为了解锁grounding能力,作者做了一个大规模grounded image-text pair数据集GRIT(caption-box pair)

模型大小follow kosmos-1
训练数据分为四类

  • grounded image-text pairs(新增)
  • text corpora
  • image-caption pairs
  • interleaved image-text data.

box表示:将图像分为32 * 32个bin(箱子,块),总共32 * 32个special token :\(<loc_n>\),每个图像用左上,右下两个bin的token表示
image