← 返回论文库
奠基 Groundwork CVPR 2023 (Highlight) · 2023 arXiv:2305.05665 ↗

ImageBind: One Embedding Space To Bind Them AllImageBind:一个嵌入空间绑定所有模态

只用「图像-其他模态」配对数据,就把图像、文本、音频、深度、热红外、IMU 六个模态对齐到一个空间。

作者:Rohit Girdhar, Alaaeldin El-Nouby, Zhuang Liu, Mannat Singh, Kalyan Vasudev Alwala, Armand Joulin, Ishan Misra
机构:Meta AI
发布:CVPR 2023 (Highlight)
时间:2023-05
多模态对齐音频
原文摘要 · English Abstract

We present ImageBind, an approach to learn a joint embedding across six different modalities — images, text, audio, depth, thermal, and IMU data. We show that all combinations of paired data are not necessary; only image-paired data is sufficient to bind the modalities together, extending zero-shot capabilities to new modalities just by using their natural pairing with images.

概要

ImageBind 是 Meta AI FAIR 提出的多模态对齐方法,目标是用一个联合嵌入空间覆盖图像/视频、文本、音频、深度、热成像与 IMU 六种模态。此前方法(如 CLIP、AudioCLIP)只能用某一对模态的成对数据训练,嵌入局限于那一对模态,而凑齐所有模态两两成对的数据几乎不可能。ImageBind 的关键洞察是把图像当作"枢纽":只需把每种模态分别对齐到图像嵌入,就能在从未见过配对样本的模态之间(如音频与文本)涌现出对齐能力,几乎不需要额外文本监督,就能在多个模态上匹敌甚至超过专门训练的专家模型。

此前的多模态对齐研究大多局限于单一模态对:CLIP、ALIGN 只对齐图像和文本,AudioCLIP 等把音频接入 CLIP 框架,但仍需专门收集音频-文本配对数据。这类嵌入只覆盖训练用到的那一对模态,视频-音频嵌入无法直接服务于图文任务,反之亦然。若要覆盖更多模态,理论上需要采集所有模态两两之间的成对数据,但深度、热成像、IMU 几乎不可能都和文本、音频收集到大规模配对样本,这是扩展的主要障碍。

关键贡献

  • 提出仅需「图像-其他模态」成对数据即可训练六模态联合嵌入空间的方法,无需模态两两之间的直接配对数据。
  • 发现并系统验证了「涌现对齐」现象:只用 (图像,模态1) 和 (图像,模态2) 训练,模型也能在模态1和模态2之间自动产生对齐能力。
  • 在音频、深度等非视觉模态上取得当时领先的涌现式零样本分类与检索效果,部分超过专门用该模态文本配对数据训练的监督模型。
  • 展示了嵌入空间的算术可组合性,把不同模态的嵌入直接相加就能合成语义、实现跨模态图像检索。
  • 通过替换预训练模型(Detic 检测器、DALL-E 2 扩散模型)中的 CLIP 文本嵌入为音频嵌入,无需重新训练即可把它们升级为声音驱动的检测与生成系统。
  • 系统性地做了图像编码器规模、温度、投影头、batch size、数据增强等消融实验,给出了训练涌现对齐的关键设计准则。

方法详解(完整做法)

总体思路

ImageBind 不去凑齐六种模态两两之间的成对数据,而是只用「图像, 某模态」这一种配对形式:图像与文本(网络图文对)、视频与音频(AudioSet)、图像与深度(SUN RGB-D)、图像与热成像(LLVIP)、视频与 IMU(Ego4D)。每种模态各自过一个编码器,输出被投影到同一维度的嵌入空间,用对比损失把该模态的嵌入拉向对应图像的嵌入。图像因此成为所有模态共享的「锚点」:只要每个模态都和图像对齐,模态之间即便训练时从未直接配对,也会在共享空间里自动靠拢。

用图像做「枢纽」绑定六种模态

给定一对模态 I,M\mathcal{I}, \mathcal{M} 及其配对观测 Ii,Mi\bm{I}_i, \bm{M}_i(例如一帧图像与它对应的音频片段),分别用编码器 ff、gg 得到嵌入:

qi=f(Ii),ki=g(Mi)\mathbf{q}_i = f(\mathbf{I}_i), \qquad \mathbf{k}_i = g(\mathbf{M}_i)

其中 I\mathcal{I} 始终固定为图像/视频模态,M\mathcal{M} 依次替换为文本、音频、深度、热成像或 IMU,六种模态共用同一套「以图像为一端」的训练范式,而不需要为每一对模态单独设计架构。

对比损失:InfoNCE 与对称化

把 qi\mathbf{q}_i 与 ki\mathbf{k}_i 的匹配作为正样本,mini-batch 内其余样本 kj≠i\mathbf{k}_{j\neq i} 作为负样本,用 InfoNCE 损失拉近正样本、推开负样本:

LI,M=−log⁡exp⁡(qi⊤ki/τ)exp⁡(qi⊤ki/τ)+∑j≠iexp⁡(qi⊤kj/τ)L_{\mathcal{I},\mathcal{M}} = -\log \frac{\exp(\mathbf{q}_i^{\top} \mathbf{k}_i/\tau)}{\exp(\mathbf{q}_i^{\top} \mathbf{k}_i/\tau) + \sum_{j \neq i}\exp(\mathbf{q}_i^{\top} \mathbf{k}_j/\tau)}

实际训练中把图像到模态、模态到图像两个方向各算一次并相加,得到对称损失:

L=LI,M+LM,I\mathcal{L} = L_{\mathcal{I},\mathcal{M}} + L_{\mathcal{M},\mathcal{I}}

τ\tau 是控制 softmax 平滑程度的温度标量,值越小分布越尖锐、模型对难负样本越敏感。

为什么绑定图像就能让未配对的模态涌现对齐

论文只用 (I,M1)(\mathcal{I},\mathcal{M}_1) 和 (I,M2)(\mathcal{I},\mathcal{M}_2) 两类配对训练,却观察到 M1\mathcal{M}_1 与 M2\mathcal{M}_2(例如音频与文本)之间也出现了对齐。原因在于对比损失把 M1\mathcal{M}_1 和 M2\mathcal{M}_2 的嵌入都拉向了同一批图像嵌入所在的区域:图像相当于共享坐标系里的公共锚点,两个从未见过对方的模态各自被锚定到相近的图像簇附近,几何上自然彼此接近。论文明确指出这与多语言机器翻译里「共享隐空间桥接、零样本翻译未配对语言对」的现象是同一种机制。

模态专属编码器与冻结的图文塔

六种模态各用独立编码器:图像/视频用 ViT,把 patch 投影层做时间维膨胀以处理 2 秒内采样的 2 帧视频;音频转成 128 维 mel 频谱图后当作二维图像,用 patch 16、stride 10 的 ViT 编码;深度图转成视差图、热成像当作单通道图像,两者都用 ViT;IMU 取 5 秒窗口的加速度计与陀螺仪读数,先用 kernel 为 8 的一维卷积投影,再送入 Transformer。每个编码器后接模态专属的线性投影头,映射到统一维度并归一化。图像与文本编码器直接用预训练 OpenCLIP(ViT-H 视觉塔 6.3 亿参数、文本塔 3.02 亿参数)初始化并冻结,只训练音频、深度、热成像、IMU 四个新编码器去对齐已经固定的图文空间。

温度、投影头与训练规模的消融发现

消融实验表明:更强的图像编码器(ViT-H 对比 ViT-B)能直接提升深度、音频等非视觉模态的零样本效果,说明图像塔的表征质量是涌现对齐强弱的决定因素;线性投影头比 MLP 投影头效果更好,这与 SimCLR 等自监督方法的经验相反;固定温度对深度、音频、IMU 分类比可学习温度更好;深度对齐依赖空间对齐的裁剪,换成随机不对齐裁剪反而明显掉分;不同模态的最优 batch size 也不同——数据量小的图像-深度对更适合小 batch,数据量大的视频-音频对则受益于大 batch。

小结

一句话:ImageBind = 只用「图像-模态」配对数据 + 对称 InfoNCE 对比损失 + 冻结的预训练图文塔,靠图像这个共享锚点让从未直接配对的模态之间涌现出跨模态对齐。

实验与关键数据

训练数据包括 OpenCLIP 大规模图文对,以及 AudioSet(视频-音频)、SUN RGB-D(图像-深度)、LLVIP(图像-热成像)、Ego4D(视频-IMU)等自然配对数据。评测覆盖涌现式零样本分类、跨模态检索、少样本分类,结果显示未见过配对的模态组合也能取得强零样本效果,效果随图像编码器增强而提升。

关键数据

MSR-VTT 文本→视频检索性能
36.1%
NYU-Depth 深度分类准确率(batch size 512)
47.3%
vs batch size 4096 39.9%(+7.4)
ESC-50 音频分类准确率(batch size 2048)
56.7%
vs batch size 512 39.4%(+17.3)
深度分类:ViT-H 对比 ViT-B 图像编码器带来的提升
+7%
音频分类:对视频做强数据增强导致的准确率降幅
-34%

影响与局限

ImageBind 把 CLIP 式的图文对比学习从两模态推广到六模态,证明「以图像为枢纽的隐式桥接」足以让从未直接配对的模态之间涌现出对齐能力,为后续多模态基础模型提供了不依赖全配对多模态数据集的训练范式;其「即插即用升级现有 CLIP 下游模型」的思路也启发了后续把预训练视觉-语言模型扩展到新模态的做法。

局限与不足

ImageBind 的嵌入是通用表征,没有针对具体下游任务优化,因此在专用任务上通常落后于专门训练的 specialist 模型——论文自己也承认这一点。涌现对齐的效果强烈依赖底层图像编码器的质量,如果图像塔较弱,非视觉模态的零样本效果会明显下降,等于把上限压在了图文预训练模型的天花板上。涌现的模态对齐主要在分类、检索这类「看起来相关」的任务上验证,缺少针对结构化预测任务(如检测、分割)的直接验证。深度、热成像等模态所用的训练数据集(SUN RGB-D、LLVIP)规模较小,需要复制 50 倍扩充才能训练,容易过拟合到数据集特有的偏置。论文也明确指出这只是研究原型,还不能直接用于真实场景部署。