视觉语言模型需要视觉Transformer吗?评估状态空间模型作为视觉编码器 Do VLMs Need Vision Transformers? Evaluating State Space Models as Vision Encoders
SSM视觉编码器VMamba在VLMs中可替代ViT,在定位任务上表现更优
前置知识
状态空间模型(SSM)
状态空间模型是一种通过状态向量动态更新的序列建模方法,在视觉领域如VMamba中,采用2D-Selective-Scan(SS2D)设计,在2D网格上进行四方向扫描,每个patch接收四个方向的状态更新,从而在架构层面保留空间结构信息
本文核心对比SSM与Transformer作为视觉backbone,理解SSM如何通过结构化扫描保持空间信息对理解其优势至关重要
视觉语言模型(VLM)
视觉语言模型由三部分组成:frozen视觉编码器提取图像特征为视觉tokens,轻量级连接器将视觉tokens映射到LLM嵌入空间,解码器语言模型处理多模态输入并生成文本输出。本文采用LLaVA风格设计,视觉编码器在指令调优期间保持冻结
理解VLM的模块化设计(特别是视觉backbone的冻结机制)是理解本文backbone-controlled swap评估框架的基础
视觉-语言接口
视觉-语言接口指视觉tokens如何传递给LLM的机制,包括连接器设计(如2层MLP,其中W表示权重矩阵,b表示偏置)和输入几何(图像分辨率、长宽比、token数量)。本文发现接口不稳定会导致定位崩溃(localization collapse)
接口设计是本文诊断的三要素之一(backbone、objective、interface),直接影响空间信息的传输和利用
定位崩溃(Localization Collapse)
定位崩溃指某些高分辨率检测预训练配置(如ViTDet-L/H和VMamba-T/B)在grounding基准上出现急剧的性能下降,例如ViTDet-L的加权平均定位分数从43.74(ViTDet-B)骤降至13.05。本文诊断这是视觉-语言接口失效而非视觉特征质量问题
定位崩溃是本文发现的关键失败模式,引导作者提出增强连接器容量和调整输入几何的稳定化策略
研究动机
现有VLMs的视觉backbone选择相对狭窄,大多数系统仍依赖ViT-family或基于Transformer的编码器。尽管有大量VLM训练配方的研究,但不同研究往往同时改变多个因素,包括视觉预训练目标、多模态训练流水线、分辨率和token化设置、连接器设计等,这使得难以区分性能差异究竟源于视觉架构本身,还是backbone家族的选择限制了视觉编码器能够传递的有用证据量。此外,从固定多模态token预算的图像中提取空间定位证据是VLM的 recurring 挑战,增加图像分辨率或视觉token数量会快速提高视觉编码器和LLM的计算与内存成本。
本文的目标是本文的目标是系统地评估SSM视觉backbone作为VLM视觉编码器的有效性,特别是使用VMamba作为纯SSM backbone的强基线。作者希望通过控制backbone swaps,在固定的多模态训练设置下比较不同架构家族,理解视觉backbone选择如何影响VLM行为,特别是对定位和grounding性能的影响。
与已有工作不同的是,本文的独特切入点在于:这是首次在VLMs中执行包含SSM视觉编码器的受控backbone swaps,同时匹配训练配方和视觉-语言接口。不同于大多仅关注VQA性能或同时改变多个因素的研究,本文采用严格的backbone-controlled swap框架,将影响VLM性能的因素解耦为三个维度:backbone架构(Transformer、SSM、混合)、预训练目标(分类vs密集目标)、视觉-语言接口(分辨率/几何和连接器设置)。
核心方法
本文采用LLaVA风格的VLM架构设计,由frozen视觉编码器、轻量级连接器(2层MLP)和解码器语言模型(Vicuna-7B)组成。核心方法是backbone-controlled swap:在固定训练配方下,仅交换视觉backbone checkpoint,从而将性能差异归因于视觉编码器而非其他混淆因素。评估在两个方案下进行:首先是matched IN1K/224 backbone swaps,严格匹配ImageNet-1K预训练、224乘224输入、L等于196视觉tokens;其次是dense-objective checkpoints比较,评估检测和分割预训练的影响。
核心创新点在于采用三维度解耦的backbone-controlled swap评估框架,将VLM性能的决定因素分解为:(i)backbone架构:比较Transformer(ViT)、SSM(VMamba)、混合架构(MaxViT、MambaVision);(ii)预训练目标:分类预训练vs密集目标预训练(检测、分割);(iii)视觉-语言接口:连接器容量和输入几何(分辨率、长宽比)。通过严格的控制实验,本文发现SSM-based视觉编码器(VMamba)在匹配设置下提升定位性能同时保持VQA竞争力,并能以更小的模型规模匹敌或超越更大的backbone。
方法步骤详情
方法分为五个主要步骤:首先,初始化视觉编码器和语言模型从预训练checkpoint,随机初始化连接器。其次,将输入图像馈送到视觉编码器提取特征为视觉tokens序列,其中L是视觉token数量,d_vision是token维度。第三,通过连接器将视觉tokens映射到LLM嵌入空间,连接器定义为线性变换加激活函数的形式。第四,将视觉嵌入与文本嵌入沿序列维度拼接,形成多模态输入嵌入。第五,解码器语言模型消费多模态嵌入并自回归生成输出文本。训练期间冻结视觉编码器参数,仅通过指令调优更新语言模型和连接器参数,固定所有优化超参数(优化器、学习率调度、批量大小、步数和精度)和随机种子。
技术新颖性
技术新颖性体现在四个方面:首先,这是首次在VLMs中执行包含SSM视觉编码器的受控backbone swaps,证明了SSM-based视觉编码器作为ViT-family替代方案的竞争力。其次,系统性地分析了预训练目标(分类vs密集目标)和视觉-语言接口(分辨率/几何和连接器设置)的影响,这是overlooked的研究方向。第三,发现了两个关键的失败模式:ImageNet准确率和naive backbone scaling不能可靠预测下游VLM性能,某些高分辨率检测预训练配置会出现定位崩溃。第四,提出了简单但有效的稳定化策略:增强连接器容量和调整接口几何,这些是architecture-agnostic且可以跨不同backbone受益。
实验结果
核心发现分为四个方面:第一,在严格匹配的IN1K/224、L等于196 swaps设置下,VMamba在整体性能上最强,VMamba-T/S在所有grounding基准上持续领先定位,VMamba变体在聚合VQA上达到top。具体而言,VMamba-S在VQA-v2达到65.24,GQA达到54.08,VizWiz达到44.95,加权平均VQA达到62.39;在RefCOCO达到56.48,RefCOCO+达到44.27,RefCOCOg达到49.88,OCID-Ref达到23.09,加权平均定位达到39.17,加权整体平均达到59.27。第二,密集预训练目标(检测/分割)可以改进SSM和Transformer-based视觉编码器的VQA和定位性能。检测预训练的ViTDet-B从IN1K预训练的ViT-B的13.92定位分数提升到43.74,提升约214%;分割预训练的VMamba-S在RefCOCO达到64.17,RefCOCO+达到53.98,RefCOCOg达到59.13。第三,ImageNet准确率和naive backbone scaling是下游VLM质量的不可靠预测器,MaxViT-L的IN1K准确率(84.93)高于MaxViT-T(83.41)但定位分数远低于(10.81 vs 15.79)。第四,某些检测预训练配置出现定位崩溃,ViTDet-L的定位分数从ViTDet-B的43.74降至13.05,VMamba-B从VMamba-S的47.94降至15.02。本文将崩溃诊断为视觉-语言接口失效模式,并提出简单、architecture-agnostic的稳定化策略可以恢复定位和整体性能。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| VQA-v2 | 准确率(%) | 65.24 (VMamba-S) | 60.07 (MaxViT-L, IN1K-84.93) | +5.17 |
| GQA | 准确率(%) | 54.08 (VMamba-S) | 49.55 (MaxViT-L) | +4.53 |
| 加权平均VQA | 得分(%) | 62.39 (VMamba-S) | 57.30 (MaxViT-L) | +5.09 |
| RefCOCO | 准确率(%) | 56.48 (VMamba-S, IN1K) | 26.66 (ViT-B, IN1K) | +29.82 |
| RefCOCO+ | 准确率(%) | 64.17 (VMamba-S, IN1K to ADE20K) | 52.17 (DeiT-B, IN1K to ADE20K) | +12.00 |
| 加权平均定位 | 得分(%) | 47.94 (VMamba-S, IN1K to COCO) | 13.05 (ViTDet-L, IN1K to COCO, 崩溃) | +34.89 (稳定化后恢复) |
| 加权整体平均 | 得分(%) | 60.78 (VMamba-S, IN1K to COCO) | 51.65 (ViTDet-L, IN1K to COCO, 崩溃) | +9.13 |
局限与改进
作者承认的局限性包括:首先,研究主要聚焦于VMamba作为SSM backbone的代表,虽然包含了MaxViT和MambaVision作为混合架构基线,但Transformer和SSM家族的覆盖仍有限,未来可以扩展到更多变体。其次,dense-task预训练仅包含检测和分割,未包含其他密集任务如实例分割或姿态估计。第三,接口问题的诊断仍是初步的,虽然提出了传输瓶颈和利用瓶颈的假设,但更深层的机制分析留待未来工作。本文观察到的额外局限性包括:研究主要关注frozen vision encoder设置,未探索finetune vision encoder的设置,这可能限制结论的泛化性。评估主要在标准VQA和grounding基准上,未涉及更复杂的多模态推理任务。实验主要在ImageNet-1K、COCO和ADE20K上预训练,未探索更大规模的自监督预训练(如CLIP、SigLIP、DINOv2),虽然作者在附录F中报告了结合对比预训练和SSL预训练的结果,但可比较的SSM-based backbone目前不可用。最后,稳定化策略虽然有效,但更多是启发式的,对为什么方形几何改进利用的深入分析留待未来工作。
独立分析的弱点
独立分析的弱点包括:第一,VMamba在更大规模(VMamba-B)时出现性能下降,定位分数从VMamba-T的39.20降至VMamba-B的27.89,这表明SSM backbone并非scale-proof。改进方向是探索如何在保持空间归纳偏差的同时提高大模型的稳定性,可能涉及重新平衡归纳偏差与模型容量。第二,高分辨率非方形输入的不稳定性问题未完全解决,虽然512乘512方形几何可以恢复性能,但这可能牺牲预训练分辨率的优势。改进方向是开发更鲁棒的连接器设计或自适应输入几何策略。第三,缺乏自监督SSM backbone的评估,限制了结论在预训练目标维度的完整性。改进方向是开发SSM架构的自监督预训练方法,或探索现有方法的适配。第四,接口设计仍相对简单(标准2层MLP),可能不是最优的。改进方向是探索更复杂的连接器架构(如cross-attention、Mamba-based连接器)或end-to-end接口学习。
未来方向
未来研究方向包括:作者提出的方向有(i)对为什么方形几何改进利用进行更深入的机制分析,可能涉及研究LLM如何从不同几何的视觉tokens中解释空间信息;(ii)扩展到更多SSM变体和混合架构,以验证发现的泛化性;(iii)探索其他密集任务预训练目标的效果,如实例分割、姿态估计等;(iv)探索更大规模的预训练数据,如更多图像-文本对或自监督学习。基于成果可延伸的方向包括(i)开发SSM架构的自监督预训练方法,如Mamba-based的对比学习或SSL方法,以与CLIP、SigLIP、DINOv2等baseline进行更公平的比较;(ii)探索更复杂的连接器架构,如cross-attention连接器、Mamba-based连接器或可学习的接口,以进一步提高空间信息的传输和利用效率;(iii)研究接口选择的自动化方法,根据任务需求自动选择最优的分辨率、几何和连接器配置;(iv)探索SSM backbone在其他多模态任务中的应用,如视觉-语言导航、图像生成、视频理解等;(v)研究SSM backbone与LLM的更深层次集成,如共享Mamba层或多模态Mamba架构。
复现评估
复现评估方面,作者基于验证过的代码库,使用Fully Sharded Data Parallel (FSDP)在4乘NVIDIA H200 GPU上进行训练。实验配置固定,包括优化器、学习率调度、批量大小、步数和精度,以及随机种子,这为复现提供了良好的基础。使用的视觉backbone(VMamba、ViT、MaxViT、MambaVision、ViTDet、DeiT)是开源的,预训练在标准数据集(ImageNet-1K、COCO、ADE20K)上进行。训练数据使用665K多模态指令调优示例,这些是公开可用的。评估基准(VQA-v2、GQA、VizWiz、TextVQA、POPE、TallyQA、RefCOCO、RefCOCO+、RefCOCOg、OCID-Ref)是标准且公开的。所有后处理和数据集特定阈值遵循[14],确保了比较的公平性。虽然作者未明确提供完整代码仓库的链接,但提到基于verified codebase,这表明代码质量较高。高分辨率输入(如1024乘1024、1333乘800)可能需要较大的计算资源,但中等分辨率的实验(如224乘224、512乘512)在更有限的资源下也应可行。总体而言,复现难度为中等,主要挑战在于获取足够的计算资源和正确设置训练配置。
论文图表