CoME-VL:可扩展的互补式多编码器视觉语言学习框架 CoME-VL: Scaling Complementary Multi-Encoder Vision-Language Learning
通过熵引导层选择+正交融合+RoPE交叉注意力整合SigLIP与DINO双编码器,让VLM在视觉理解与定位任务上同时提升约5%
前置知识
Vision-Language Model (VLM)
视觉语言模型是把视觉编码器(如ViT/CLIP)与大语言模型对接、使其能根据图像回答问题或生成描述的系统。一般由视觉编码器、投影层和LLM三部分组成,常见架构包括LLaVA、Qwen-VL、Molmo等。
本文正是在VLM的基础上重新设计视觉编码器侧的多源融合方式,因此需要先理解VLM的标准pipeline(图像分块→视觉编码→投影→LLM自回归生成),才能理解作者改动哪些环节、为什么改动。
对比学习视觉编码器 (Contrastive Encoder, 如CLIP/SigLIP)
通过图像-文本对做对比学习训练得到的视觉编码器,特征空间天然与文本对齐,适合做检索、零样本分类,但监督是全局的,对细粒度空间定位不敏感。SigLIP用sigmoid损失替代softmax,能在更大batch下稳定训练。
本文使用的SigLIP2是这类编码器的代表,提供强语义对齐,但正因如此它在精确定位上弱——这是CoME-VL引入DINO作为补充的根本原因。
自监督视觉编码器 (DINO/DINOv2/DINOv3)
通过自蒸馏或掩码自重建等无标签目标训练的ViT,特征天然带有清晰的空间结构、目标边界和几何信息,在分割、检测等密集预测任务上表现优异,但在零样本语义对齐上不如CLIP系。
DINOv3为本文提供'定位能力',理解它为什么能产出强空间特征、为什么适合与SigLIP互补,是理解CoME-VL设计动机的前提。
正交约束与近等距映射 (Orthogonal Layer / Cayley transform)
在线性投影中要求权重矩阵 $Q$ 满足 $Q^\top Q = I$,保证映射是近等距的(不缩放、不丢失信息),同时让不同层映射到正交子空间以减少冗余。Cayley变换 $Q=(I+A/2)(I-A/2)^{-1}$ 是从反对称矩阵 $A$ 构造正交矩阵的可微参数化方法。
正交层是CoME-VL抑制多层冗余的核心机制之一,理解为什么'正交=不冗余'对读懂方法部分至关重要。
RoPE / 4D Cross-RoPE交叉注意力
旋转位置编码(Rotary Position Embedding)把位置信息编码为查询/键的相位旋转,使注意力分数依赖相对位置而非绝对位置。Cross-RoPE进一步把不同分辨率token网格的2D空间偏移注入到跨编码器注意力中,让异构token能空间一致地匹配。
本文用RoPE cross-attention而非简单拼接来对齐SigLIP(24×24)与DINOv3(14×14)两种不同分辨率的token网格,是它避免LLM上下文膨胀的关键。
层间熵 (Layer-wise Spatial Entropy)
把每一层特征图的注意力分布看作概率分布,计算其信息熵。熵高表示注意力分散、覆盖广;熵低表示注意力集中、目标明确。本文用此度量每一层特征是'语义多样'还是'空间精确'。
熵是本文选择融合层范围的核心判据,理解它才能解释为什么最终选SigLIP全部27层+DINO第10–23层,而不是任意切片。
研究动机
当前主流视觉语言模型(VLM)几乎清一色地使用单一CLIP/SigLIP对比式视觉编码器,并在最后一层或倒数第二层抽取特征送入LLM。这种做法存在三个具体痛点:第一,CLIP的图像-文本对比目标是全局对齐的,对颜色属性、细微空间关系、目标边界等细粒度线索不敏感,作者在Table 1中展示了MiniCPM-V 4.5、Qwen3-VL 4B/8B等模型在PixMo Pointing任务上'不支持'(无法输出像素坐标),而Molmo 7B也仅有53.79%/68.94% (@3px/@5px) 的精度;第二,VLM架构存在明显的'视觉-语言不对称',往往用较小的ViT-Large搭配7B–13B的LLM,视觉编码器成为整个系统的瓶颈;第三,已有工作(COMM、CLIP meets DINO、Eagle等)虽证明多编码器融合能改善通用理解,但要么只验证语义任务不验证定位任务,要么直接把多层多源特征拼接喂给LLM,造成视觉token数量大幅膨胀(COMM推理需2.2s/样本 vs 单编码器1.26s/样本),既贵又冗余。
本文的目标是本文的目标是构建一个名为CoME-VL(Complementary Multi-Encoder Vision-Language)的模块化多编码器融合框架,让VLM在保持高效推理的同时,把SigLIP风格的对比式语义特征与DINO风格的自监督空间特征有机结合,从而同时提升视觉理解(如图表、文档理解)和视觉定位(如pointing、counting、RefCOCO目标框预测)两类任务。具体而言,作者希望在PixMo理解类任务上比单编码器基线Molmo 7B平均提升约4.9%,在grounding类任务上提升约5.4%,并在RefCOCO上取得SOTA。
与已有工作不同的是,已有工作(COMM、CLIP-DINOiser、Frozen CLIP-DINO等)要么只关注语义任务不评估grounding,要么用简单拼接/平均方式融合多层特征,既没有'哪一层有用'的判据也没有'减少冗余'的机制,更没有解决不同编码器token网格分辨率不一致的问题。CoME-VL的独特切入角度是首次把'层间熵分析'作为层选择判据,把'正交层(OL)'作为显式去冗余机制,把'RoPE交叉注意力'作为异构token高效对齐方案——三者在同一个框架内联合优化,并明确针对grounding任务做验证,区别于以往工作只验证semantic任务。
核心方法
CoME-VL的整体思路可以一句话概括为:先用两个互补的视觉编码器(SigLIP2负责语义、DINOv3负责空间)各自抽取多层特征,再用'熵'作为判据选出最值得融合的层范围;接着给每一层加一个'正交层'投影以抑制层间冗余并做softmax加权聚合;最后用一个带RoPE的跨编码器交叉注意力把DINO的空间信息注入到SigLIP的token流里,再通过门控残差把紧凑的融合视觉token送进decoder-only LLM。整个设计的直觉是'分工→去重→对齐':让两个编码器做各自擅长的事→避免简单堆叠带来的冗余→以高效方式让LLM看到对齐后的紧凑表征。技术上建立在Molmo+Qwen2-7B基线之上,DINOv3参数全程冻结,SigLIP和融合模块随训练更新。
与已有方法相比,CoME-VL的本质区别在于三点:第一是'熵引导'——把视觉编码器每一层的空间注意力熵作为数据驱动的层选择准则,而不是凭经验切几段;第二是'正交去冗余'——给每层特征加上由Cayley变换参数化的正交投影 $Q^\top Q = I$,显式把不同层映射到正交子空间,让加权聚合后的特征占满更广的表示空间而非坍缩到高度相关的子空间;第三是'RoPE跨注意力对齐'——用SigLIP token作Q、DINO token作K/V并注入2D RoPE,使异构token网格能按空间位置精确匹配,并通过门控残差 $\tanh(\gamma)$($\gamma$ 初始化为0)控制DINO信息的注入强度,训练初期稳定、后期逐步放开。这三个机制是协同设计的:熵选层、正交保多样性、RoPE保空间一致性,共同支撑'多编码器但不冗余、不膨胀'的目标。
方法步骤详情
CoME-VL的训练与推理流程分为四步。**步骤1:双编码器多层特征抽取**。图像 $I \in \mathbb{R}^{H\times W\times 3}$ 分别送入冻结的SigLIP2-SO400M(384×384,patch=16,共27层)和冻结的DINOv3-Large(224×224,共24层),记第 $l$ 层隐藏态为 $Z^{(l)}_{\mathrm{Sig}} \in \mathbb{R}^{N_s\times d_s}$ 和 $Z^{(l)}_{\mathrm{DINO}} \in \mathbb{R}^{N_d\times d_d}$。**步骤2:熵引导候选层选择**。对每一层计算空间注意力熵,SigLIP2各层普遍较高(语义分散),DINOv3深层显著降低(空间聚焦),由此选定 $L_s = \{1,\dots,27\}$(全层)和 $L_d = \{11,\dots,24\}$(低熵深层)。**步骤3:正交化多层聚合**。对每个被选层先做LayerNorm、再过一个Orthogonal Layer(OL),OL通过反对称矩阵 $A_l$ 的Cayley变换 $Q_l=(I+A_l/2)(I-A_l/2)^{-1}$ 参数化正交投影,保证 $Q_l^\top Q_l = I$;随后用softmax权重 $w^{(e)}_l = \exp(\alpha_l)/\sum_k \exp(\alpha_k)$ 对同编码器内各层做加权求和,得到 $V_{\mathrm{Sig}}$ 和 $V_{\mathrm{DINO}}$。**步骤4:RoPE增强跨编码器对齐**。令Query来自SigLIP、Key/Value来自DINO,对Q/K施加2D RoPE得到空间相对位置感知的注意力分数;DINO侧可选2×2池化进一步压低token数;交叉注意力输出经 $\tanh(\gamma)$ 门控后加回SigLIP token,得到 $V_{\mathrm{fused}} = V_{\mathrm{Sig}} + \tanh(\gamma)\cdot\mathrm{CrossAttn}(V_{\mathrm{Sig}}, V_{\mathrm{DINO}})$,$\gamma$ 初始化为0以稳定早期训练;最终投影后的紧凑视觉token序列送入decoder-only Qwen2-7B LLM,自回归生成答案,对grounding任务额外用6 token序列 $\langle\mathrm{BOX}\rangle \langle x_1\rangle\langle y_1\rangle\langle x_2\rangle\langle y_2\rangle\langle\mathrm{END\_BOX}\rangle$ 序列化预测1000-bin坐标。
技术新颖性
技术新颖性体现在三处显式的'工程+理论'组合创新。**第一**,把'层间熵'从分析工具升级为融合层选择判据,给出了一个数据驱动的、跨编码器一致的层选择准则,区别于Cambrian-1、Eagle等凭经验/网格搜索选层的做法;**第二**,把正交层显式嵌入到多层融合流程中,借助Cayley变换保证可微优化,避免了简单concat/avg造成的'不同层映射到高度相关子空间'的退化问题;**第三**,把RoPE从单编码器自注意力的位置编码,扩展到跨编码器交叉注意力场景——通过Q/K上的2D RoPE让异构token网格按空间偏移精确对齐,避免直接concat导致的'语义相似但空间错位'误匹配;此外,门控残差 $\tanh(\gamma)$ 借鉴Flamingo设计但配合 $\gamma=0$ 初始化保证了训练的稳定性。三者联合使得CoME-VL在增加约20%推理时间(1.26s→1.52s/样本)的同时,带来4.9%/5.4%的精度提升,相比COMM的2.2s/样本效率显著更高。
实验结果
**核心发现1:PixMo理解任务全面超越单编码器基线**。Table 2显示CoME-VL 7B在Charts 57.24%、Diagrams 66.94%、Tables 70.75%、Others 81.84%四个子任务上均超过Molmo 7B(分别为52.39/62.41/66.25/76.26%),且在Counting上达到87.83%(vs Molmo 83.31%、vs InternVL2-8B 74.05%),同时CoME-VL是唯一报告Pointing指标的模型:@3px 58.56%、@5px 75.94%,相对Molmo 7B的53.79%/68.94%提升明显。**核心发现2:RefCOCO定位达SOTA**。Table 3中CoME-VL在val/testA/testB上分别取得92.57%/95.36%/90.51%,相比CLIP-to-DINO基线(91.73/94.06/88.85)提升+0.84/+1.30/+1.66,相比Qwen-VL(89.36/92.23/85.36)提升更显著,说明DINO的引入直接转化为定位精度。**核心发现3:熵引导的层选择效果显著**。Table 4显示 SigLIP(0–27)+DINO(10–23) 组合在所有任务上最强(平均推理1.52s),而 SigLIP(22–27)+DINO(0–9) 组合最弱(Pointing @3px 仅52.41% vs 58.56%),这印证了'SigLIP全层提供语义广度+DINO深层提供空间精度'的设计直觉。**核心发现4:自监督编码器越强越好**。Table 5对比DINOv1→v2→v3的演化,PixMo Chart从54.18→55.68→57.24%,Counting从85.91→86.12→87.83%,说明自监督空间表示的质量直接传导到下游grounding精度。**核心发现5:组件贡献可叠加**。Figure 6的stacked bar显示,在Molmo基线之上,单独加RoPE带来平均+3.0分、再加OL带来额外+1.2到+3.1分,组件互补而非互斥。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| PixMo Chart 理解 | Accuracy (%) | 57.24 | Molmo 7B: 52.39 | +4.85 |
| PixMo Diagrams 理解 | Accuracy (%) | 66.94 | Molmo 7B: 62.41 | +4.53 |
| PixMo Tables 理解 | Accuracy (%) | 70.75 | Molmo 7B: 66.25 | +4.50 |
| PixMo Counting 计数 | Accuracy (%) | 87.83 | Molmo 7B: 83.31 | +4.52 |
| PixMo Pointing @3px | Accuracy (%) | 58.56 | Molmo 7B: 53.79 | +4.77 |
| PixMo Pointing @5px | Accuracy (%) | 75.94 | Molmo 7B: 68.94 | +7.00 |
| RefCOCO val | IoU≥0.5 准确率 (%) | 92.57 | CLIP-to-DINO: 91.73 | +0.84 |
| RefCOCO testA | IoU≥0.5 准确率 (%) | 95.36 | CLIP-to-DINO: 94.06 | +1.30 |
| RefCOCO testB | IoU≥0.5 准确率 (%) | 90.51 | CLIP-to-DINO: 88.85 | +1.66 |
局限与改进
**作者承认的局限**:(1) 推理时间从Molmo的1.26s/样本增加到1.52s/样本(约+20%),虽然显著优于COMM的2.2s/样本,但相比纯单编码器基线仍存在不可忽略的开销,主要来自RoPE cross-attention和正交层投影;(2) 当前评估主要在PixMo和RefCOCO两个grounding密集型benchmark上,未充分验证对一般图文对话(open-ended VQA)的泛化性。**我的观察**:(1) 训练资源门槛较高——64块AMD MI210 GPU训练7天,相当于约448 GPU·日,中小实验室难以复现;(2) 框架依赖DINOv3作为高质量自监督编码器,对DINO版本的演化敏感(Table 5显示v1→v3带来约3个点的差距),如果未来DINO系列不再维护或更名,整套方案的迁移性需要重新验证;(3) RoPE cross-attention中Query固定来自SigLIP、Key/Value来自DINO的设计是单向的,没有探索双向融合,可能遗漏SigLIP特征对DINO分支的反馈增益;(4) 高分辨率图像最多切12个crops,对超高分辨率(4K+)文档或医学图像的可扩展性未讨论;(5) RefCOCO的box序列化依赖Molmo改造的6-token输出格式,相当于'借债'于基线能力,缺乏端到端的检测范式。
独立分析的弱点
**弱点1:单向信息流的潜在浪费**。当前架构只让SigLIP token作为Query去聚合DINO的Key/Value,但DINO分支无法反过来吸收SigLIP的全局语义。在需要'先定位再理解上下文'的场景(如医学影像先框出病灶再描述),单向融合可能导致DINO编码器对高层语义'听不懂'。**改进方向**:引入双向cross-attention或对称的token交换机制,让两路token互为Q/K。**弱点2:训练成本高、推理开销不可忽略**。64块MI210训练7天(约448 GPU·日),推理+20%时间,对工业部署是显著成本。**改进方向**:可以探索蒸馏版本(如把CoME-VL蒸馏回单编码器)或用更低秩近似(如LoRA化正交层、用Q-Former替代RoPE cross-attn)压缩到消费级GPU。**弱点3:层选择依赖离线熵统计**。$L_s$、$L_d$ 是基于在ImageNet/PixMo上预计算的熵统计得到的,与下游任务可能存在分布偏移。**改进方向**:引入可学习的层门控(如每层一个可微soft mask),让模型在训练中动态决定使用哪几层,避免离线启发式。**弱点4:grounding评测局限于英文PixMo/RefCOCO**。中文、多语言、长尾类别的grounding能力未验证。**改进方向**:在RefCOCO-CN、视觉7W等多语言数据集上做迁移测试,并对box token序列化做语言无关的离散化设计。**弱点5:bounding box生成借债于Molmo的特殊token**。整套坐标离散化(1000 bins、6 token序列)是Molmo特有的,未与Qwen-VL、LLaVA等其他VLM的输出范式对齐,跨基线迁移困难。**改进方向**:统一为业界标准的'数值坐标输出'或直接复用各基线原生格式。
未来方向
**作者提出的方向**:(1) 把CoME-VL扩展到视频模态,让SigLIP/Video-SigLIP提供时序语义、DINOv3的空间特征补充时空定位,应对视频grounding、video QA;(2) 探索更多互补编码器组合(如ConvNeXt、SAM、MAE)甚至编码异构模态(深度图、热成像);(3) 把正交融合和RoPE对齐抽象为通用模块,让其可插拔到其它VLM backbone。**基于成果可延伸的方向**:(1) **跨任务迁移**:在医学影像(CT/MRI)的病灶grounding、机器人操控的空间定位(DELAN等)上验证CoME-VL,因为这些任务对空间精度要求比PixMo更严苛;(2) **训练效率**:用知识蒸馏把CoME-VL 7B压缩到2B–3B,让'多编码器'思想在端侧可部署;(3) **统一表征空间**:当前SigLIP和DINO的隐藏维度需通过OL对齐到相同 $d$,未来可探索在embedding阶段就用共享codebook做早期对齐,避免维度转换损耗;(4) **理论分析**:从信息论角度量化'正交子空间'与下游任务的互信息关系,给出层选择和OL约束的数学最优性解释;(5) **结合RLHF**:在grounding任务上加入DPO/PPO,用人类反馈进一步提升pointing精度。
复现评估
**开源情况**:作者明确承诺将代码开源在GitHub(github.com/mbzuai-oryx/CoME-VL),模型权重发布在HuggingFace(huggingface.co/MBZUAI/CoME-VL),项目主页在mbzuai-oryx.github.io/CoME-VL,论文附有详尽附录解释Orthogonal Layer的Cayley参数化、bounding box token序列化方案、attention rollout可视化等细节。**数据**:训练数据为PixMo [15] 和RefCOCO [42],均为公开数据集;PixMo本身包含点选、计数、图表理解等多种任务标注,与本文评测一致。**算力门槛**:训练需要64块AMD Instinct MI210 GPU(每块64GB显存)跑7天,相当于约448 GPU·日,BF16精度、FSDP并行、全局batch size 24、序列长度2304;这远超一般实验室或学生项目的资源,**复现难度评估为高**——核心瓶颈不在算法而在算力。**易复现部分**:正交层Cayley参数化、RoPE cross-attention、softmax层聚合都是几十行代码的模块化设计,论文附录A给出了完整公式(Eq. 9–15),具备PyTorch基础的工程师可以在1–2周内复现核心模块;最难的反而是'完整跑通训练+评测',需要分布式基础设施经验。**对读者的建议**:若算力受限,建议先用Molmo 7B+Qwen2-VL基线在PixMo的小子集上验证grounding范式,再用DINOv3小模型(如DINOv3-S)替代DINOv3-L做消融,可在大致8块A100上2天内跑完一轮实验。
论文图表
该图由两个子图构成。子图(a) Layerwise Entropy Characterization:横轴为编码器层索引(0–27),纵轴为熵值,SigLIP2全部27层普遍保持较高熵(语义分散),DINOv3前9层熵较高、L10–23熵显著降低(空间聚焦),印证了双编码器互补性。子图(b) Performance Gains Via Optimization Fusion:横轴为不同层组合方案,纵轴为平均准确率,Molmo基线66.0%、S(0–22)+D(0–9) 64.3%、S(22–27)+D(0–9) 68.7%、S(0–22)+D(10–23) 73.0%、S(22–27)+D(10–23) 72.7%、完整CoME S(0–27)+D(10–23) 74.1%,并按Understanding(+4.9%)和Grounding(+5.4%)两个柱状图分别展示提升幅度。
这是整篇论文的'门面图',用熵证据+提升幅度一次性回答了'为什么选这两路编码器、为什么选这些层范围',是理解动机和设计的最重要视觉证据。
子图A展示DINO L0/L4/L10/L23和SigLIP L0/L4/L20/L26层的注意力mask(前30%注意力+等高线),可看出DINOv3从浅层就呈现目标中心、边界清晰的空间模式,且越深越聚焦;SigLIP2浅层注意力弥散、碎片化,深层才逐渐集中到少数语义显著区域。子图B是Grad-CAM rollout可视化,DINO(14–23层)和SigLIP(0–11层)各自捕捉到不同的语义/空间模式。
直接用视觉证据展示两种编码器在'语义vs空间'上的功能差异,是后续'分层融合'设计的最重要定性支撑。