先看再做:增强视觉基础表征用于视觉语言动作模型 Look Before Acting: Enhancing Vision Foundation Representations for Vision-Language-Action Models
通过VL-MoT框架和动作引导的视觉剪枝策略,解决VLA模型深层视觉敏感性衰减问题
前置知识
Vision-Language-Action (VLA) 模型
VLA模型是一类将视觉观测、语言指令直接映射为机器人动作的端到端模型架构。它由三个核心组件构成:视觉编码器(如SigLIP)提取图像特征,LLM骨干网络(如Qwen3-VL)融合多模态信息进行推理,动作解码器将隐藏状态映射到机器人动作空间。VLA模型的核心优势在于能够利用大规模预训练VLM的知识,通过机器人演示数据微调实现通用的操控技能学习。典型的VLA模型包括OpenVLA(自回归离散动作预测)、π0(基于扩散/流匹配的连续动作预测)和OpenVLA-OFT(并行动作解码)。
本文的核心工作是改进VLA模型的视觉表征,理解VLA的基本架构是理解本文方法的前提
Transformer 层级敏感性分析
在深度Transformer模型中,不同层级对输入信息的利用程度存在显著差异。本文发现VLA模型的LLM骨干网络中,浅层对视觉token保持较强的敏感性(注意力集中在任务相关区域),而随着层数加深,对任务相关视觉区域的敏感性逐渐衰减。这种现象可以通过两种方式观测:一是可视化动作到视觉的注意力图(浅层集中、深层弥散),二是层级视觉token丢弃实验(遮蔽深层ROI token对动作预测影响很小)。
这一发现是本文的motivation核心,理解视觉信息在VLA深层的衰减机制才能理解为何需要注入额外视觉特征
Mixture-of-Transformers (MoT) 架构
MoT是一种多分支Transformer架构,允许不同模态或分支共享注意力计算但保持独立的处理路径。在本文的VL-MoT框架中,Vision Expert和VLA骨干网络的QKV表征被拼接后计算共享注意力,然后输出被分割回两个分支各自处理。这种设计的优势在于:(1)促进跨分支信息交换,(2)减少特征干扰,(3)稳定融合训练。与简单的特征拼接或对齐不同,MoT直接在注意力层面实现深层信息融合。
VL-MoT是本文的核心架构创新,理解MoT机制才能理解Vision Expert如何与VLA骨干网络交互
动作引导的视觉剪枝 (Action-Guided Visual Pruning, AGVP)
AGVP是一种利用VLA浅层动作到视觉的注意力图来识别任务相关视觉区域的策略。具体流程为:(1)从浅层计算动作token到视觉token的注意力图 $A^\ell \in \mathbb{R}^{N_a \times N_v}$,(2)对动作token维度取平均得到每层注意力图 $m^\ell$,(3)聚合多个浅层得到最终注意力图 $m$,(4)将注意力图插值到Vision Expert的高分辨率输入尺寸,(5)使用Top-K操作保留最重要的视觉token。这使得Vision Expert可以处理更高分辨率的输入(512×512 vs VLA的256×256),同时只关注任务相关的精细视觉区域。
AGVP解决了直接融合Vision Expert全量特征可能引入冗余和背景噪声的问题,是提高计算效率和任务性能的关键策略
DINOv3 视觉基础模型
DINOv3是一种自监督视觉基础模型(0.8B参数),通过DINO训练范式学习空间精细的视觉表征。与传统的图像-文本对齐模型(如SigLIP强调图文对齐)不同,DINOv3提供更强的细粒度空间表征和以物体为中心的特征,特别适合精确操控任务。本文选择DINOv3的最后几层Transformer特征作为Vision Expert的输出,因为这些深层特征编码了高级语义信息,更具有物体中心性和任务相关性。
DINOv3作为Vision Expert为VLA模型提供了传统视觉编码器(SigLIP2)缺失的精细空间表征,是本文方法能够提升精确操控性能的关键组件
Grad-CAM 视觉贡献分析
Grad-CAM是一种基于梯度的可视化方法,用于理解神经网络中不同区域对预测结果的贡献。在本文中,作者对每一层计算视觉token相对于预测动作的梯度贡献分数,生成逐层的token贡献图。结果显示:浅层的高贡献token主要集中在任务相关区域(机械臂、被操作物体、交互区域),而深层的贡献图变得弥散,转移到不太相关的区域。这种分析方法帮助作者发现了VLA模型中视觉信息衰减的现象。
这是本文方法论的关键分析工具,帮助作者定量地验证了深层视觉敏感性衰减的假设
研究动机
视觉语言动作(VLA)模型的可靠动作预测严重依赖于准确解读和整合视觉观测。然而,本文作者通过系统分析发现了一个关键问题:在动作生成过程中,VLA模型对视觉token的敏感性在深层逐渐衰减。具体而言,作者对三种代表性的VLA模型(OpenVLA、π0和QwenVLA-OFT)进行了层级分析,发现浅层的动作到视觉注意力图能够有效集中在任务相关区域(机械臂、被操作物体),但深层的注意力变得弥散且转移到不相关区域。通过层级视觉token遮蔽实验,作者定量验证了这一现象:遮蔽浅层的ROI token会导致动作预测误差显著增加,而遮蔽深层的ROI token对动作预测影响很小。例如在OpenVLA(32层)中,浅层遮蔽ROI token后动作MSE显著上升,但深层遮蔽后MSE几乎不变。这一现象的根本原因在于当前VLA模型普遍采用串行架构,视觉信息仅在LLM的第一层注入,随着信息在Transformer层间传播逐渐衰减。
本文的目标是本文的具体目标是:(1)系统分析VLA模型中视觉信息的处理机制,识别深层视觉敏感性衰减现象;(2)提出一种新的框架来增强VLA模型深层对任务相关视觉区域的敏感性,从而提高动作预测精度;(3)设计一种基于浅层注意力的视觉剪枝策略,过滤冗余视觉信息,使Vision Expert能够处理更高分辨率的输入;(4)在仿真和真实世界任务中验证所提方法的有效性,建立新的SOTA性能。
与已有工作不同的是,本文的独特切入角度在于:不同于现有方法将VLA的LLM骨干网络视为黑盒,本文深入分析了视觉信息在LLM内部各层的处理过程。现有增强VLA视觉能力的方法主要从四个角度入手:(1)引入视觉提示改善场景理解;(2)设计辅助视觉目标鼓励模型关注关键实体;(3)引入额外视觉模态(深度图、点云);(4)预测未来状态增强物理世界建模。然而这些方法都忽视了视觉信息在LLM深层的衰减问题。本文首次从LLM内部层的视角出发,发现深层视觉敏感性衰减是影响动作预测精度的关键瓶颈,并提出通过VL-MoT架构将Vision Expert的多层特征注入VLA深层来解决这一问题。这种"先分析问题、再针对性解决"的研究范式为VLA模型设计提供了新的思路。
核心方法
本文的方法DeepVision-VLA基于一个直观的观察:VLA模型的LLM骨干网络中,浅层能够有效保持对任务相关视觉区域的敏感性,但深层这种敏感性逐渐衰减。因此,本文提出将一个视觉基础模型(DINOv3)作为Vision Expert,通过VL-MoT(Vision-Language Mixture-of-Transformers)框架将其多层特征注入VLA的深层,增强深层对视觉信息的利用。技术路线为:首先通过系统分析发现深层视觉衰减现象,然后设计VL-MoT框架实现Vision Expert与VLA深层的共享注意力融合,再引入AGVP策略利用浅层注意力图剪枝Vision Expert的视觉token,最后在QwenVLA-OFT基线上实例化为DeepVision-VLA并在仿真和真实世界任务中验证。
本文的核心创新点在于发现并解决VLA模型深层视觉敏感性衰减问题。与已有方法的本质区别体现在三个方面:(1)分析视角不同——现有方法将LLM视为黑盒,从输入端增强视觉信息;本文深入LLM内部,发现深层视觉信息衰减是关键瓶颈,从信息传播的角度提供新见解;(2)融合方式不同——现有方法多在输入层拼接或对齐视觉特征;本文提出VL-MoT架构,在深层通过共享注意力机制直接融合Vision Expert的QKV表征,实现更精准的跨分支信息交换;(3)剪枝策略不同——现有方法处理全部视觉token;本文提出AGVP,利用浅层动作到视觉的注意力图识别任务相关区域,只保留Top-K重要token,既提高计算效率又增强任务相关性。
方法步骤详情
DeepVision-VLA的方法包含以下关键步骤:(1)**多分辨率输入**:将当前图像观测(256×256)输入VLA视觉编码器(SigLIP2-Large, 0.3B),同时将高分辨率图像(512×512)输入Vision Expert(DINOv3-H, 0.8B);(2)**浅层视觉分析**:在VLA骨干网络的浅层(如Layer 4-19)计算动作token到视觉token的注意力图 $A^\ell \in \mathbb{R}^{N_a \times N_v}$,对动作token维度取平均得到每层注意力图 $m^\ell = \frac{1}{N_a}\sum_{i=1}^{N_a} A^\ell_{i,:}$,聚合多个浅层得到最终注意力图 $m = \frac{1}{|\mathcal{L}_s|}\sum_{\ell \in \mathcal{L}_s} m^\ell$;(3)**视觉剪枝**:将注意力图插值到Vision Expert的高分辨率尺寸 $\tilde{m} = \mathcal{I}(m) \in \mathbb{R}^{N_d}$,使用Top-K操作保留最重要的视觉token $\mathcal{S}_K = \text{TopK}(\tilde{m}, K)$;(4)**VL-MoT融合**:将Vision Expert的最后16层QKV表征投影到LLM隐藏维度,与VLA深层的QKV拼接计算共享注意力 $Q=[Q_E; Q_Z], K=[K_E; K_Z], V=[V_E; V_Z]$,然后分割输出回两个分支各自处理;(5)**动作解码**:经过深层增强后的动作token隐藏状态通过两层MLP解码器映射到机器人动作空间。
技术新颖性
本文的技术新颖性体现在以下方面:(1)**层级视觉衰减现象的发现**:首次系统分析了多种VLA模型(OpenVLA、π0、QwenVLA-OFT)在不同动作生成范式(自回归、流匹配、并行解码)下的视觉信息处理过程,发现深层视觉敏感性衰减是普遍存在的现象,为VLA模型设计提供了新的理论见解;(2)**VL-MoT架构设计**:不同于简单的特征拼接或对齐,VL-MoT通过共享注意力机制实现Vision Expert与VLA深层的直接信息交换,同时保持各自独立的处理路径,减少特征干扰。特别是,本文选择Vision Expert的最后几层特征(而非前几层或均匀采样),因为这些深层特征编码了高级语义和物体中心表征,更适配VLA的任务相关特征;(3)**AGVP策略的双向协同**:AGVP利用VLA浅层可靠的动作-视觉注意力来指导Vision Expert的特征剪枝,形成了"浅层分析→深层增强"的闭环设计,既提高了计算效率(只处理Top-K token)又增强了任务相关性;(4)**多分辨率输入策略**:Vision Expert处理512×512高分辨率图像,VLA处理256×256标准分辨率图像,通过AGVP的注意力插值实现跨分辨率对齐,使Vision Expert能够捕获更精细的物体细节。
实验结果
本文在仿真和真实世界任务中均取得了显著的性能提升。在RLBench仿真环境中,DeepVision-VLA在10个操控任务上达到83%的平均成功率,超越所有基线方法。相比直接基线QwenVLA-OFT(69%)提升14%,相比HybridVLA(74%)提升9%,相比π0.5(65%)提升18%。在视觉挑战性任务上提升尤为显著:Sweep to Dustpan任务从0.15提升到0.95(80%提升),Wine at Rack任务从0.65提升到0.85(31%提升)。在真实世界实验中,DeepVision-VLA在4个复杂操控任务上达到91.7%的平均成功率,超越π0.5(84.2%)、QwenVLA-OFT(74.2%)和OpenVLA-OFT(71.7%)。特别是在"倒可乐到瓶子"任务中,DeepVision-VLA在两个阶段均达到100%成功率,而最强基线π0.5在第二阶段下降到70%。消融实验验证了各组件的有效性:VL-MoT框架(88% vs 基线65.5%)、使用DINOv3最后16层(88% vs 前16层61.5%)、动作到视觉注意力引导(88% vs CLS token 65.5%)。泛化实验表明,在未见背景和光照条件下,DeepVision-VLA相比基线具有更强的鲁棒性(成功率下降5% vs 18-22%)。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| RLBench 仿真环境(10个任务平均) | 平均成功率 (%) | 83% | HybridVLA 74%, π0.5 65%, QwenVLA-OFT 69% | 相比HybridVLA +9%, 相比π0.5 +18%, 相比QwenVLA-OFT +14% |
| Sweep to Dustpan(RLBench) | 成功率 (%) | 95% | QwenVLA-OFT 15% | +80% |
| 真实世界平均(4个任务) | 平均成功率 (%) | 91.7% | π0.5 84.2%, QwenVLA-OFT 74.2%, OpenVLA-OFT 71.7% | 相比π0.5 +7.5%, 相比QwenVLA-OFT +17.5% |
| 倒可乐到瓶子(真实世界) | Step 1 + Step 2 成功率 (%) | 100% + 100% | π0.5 100% + 70% | Step 2 +30% |
| 消融:VL-MoT vs 其他融合方式 | 4任务平均成功率 (%) | 88% (VL-MoT) | Early Fusion 73%, Mid Align 67%, 基线 65.5% | 相比Early Fusion +15%, 相比Mid Align +21% |
| 消融:特征选择策略 | 4任务平均成功率 (%) | 88% (Last 16 layers) | First 16 61.5%, Uniform 16 85% | 相比First 16 +26.5%, 相比Uniform 16 +3% |
局限与改进
尽管本文取得了显著成果,但仍存在以下局限性:(1)**Vision Expert选择单一**:本文仅使用DINOv3作为Vision Expert,未探索其他视觉基础模型(如SAM、CLIP等)的效果,可能限制了方法的通用性;(2)**计算开销增加**:引入额外的Vision Expert(0.8B参数)和高分辨率输入(512×512)会增加推理时的计算成本和内存占用,虽然AGVP剪枝缓解了部分问题,但与原始VLA相比仍有开销;(3)**浅层分析假设**:AGVP依赖浅层注意力图的可靠性来识别任务相关区域,但浅层注意力可能不总是准确的,特别是在复杂场景或动态环境中;(4)**任务范围有限**:实验主要集中在单臂操控任务(RLBench)和简单的真实世界任务,未涉及更复杂的双臂协调、长时间序列规划等场景;(5)**可解释性不足**:虽然作者分析了视觉敏感性衰减现象,但对于为何DINOv3最后几层特征最有效、以及VL-MoT如何具体改善深层视觉表征,缺乏更深入的理论分析;(6)**泛化能力待验证**:真实世界实验仅在特定的机器人平台(Franka Research 3)和有限的任务上验证,对于不同机器人类型、不同操控场景的泛化能力需要进一步验证。
独立分析的弱点
本文的弱点分析如下:(1)**Vision Expert的通用性不足**:本文仅使用DINOv3-H(0.8B)作为Vision Expert,未探索其他视觉基础模型(如SAM-2、DINOv2、CLIP等)的适用性。不同任务可能需要不同特性的视觉专家,例如语义分割任务可能更适合SAM,而精细操控可能需要更强的空间表征。改进方向:设计自适应的Vision Expert选择机制,或融合多个视觉专家的互补特征;(2)**AGVP的注意力图质量依赖**:AGVP使用浅层动作到视觉的注意力图来识别任务相关区域,但注意力图可能在复杂场景(多个相似物体、遮挡、动态背景)中产生噪声。改进方向:引入更鲁棒的显著性检测方法,如结合深度信息或时序一致性约束;(3)**固定层数的特征注入**:本文固定使用最后16层DINOv3特征注入最后16层VLA,未探索动态层数选择或自适应注入策略。不同任务的视觉复杂度不同,可能需要不同深度的特征注入。改进方向:设计可学习的层数选择机制或门控策略;(4)**缺乏与输入端增强方法的对比**:消融实验主要对比了不同的融合方式,但未与视觉提示、辅助视觉目标等输入端增强方法进行充分对比,难以评估深层融合相对于输入端增强的优势程度。
未来方向
基于本文成果,未来研究可以向以下方向延伸:(1)**多Vision Expert融合**:探索融合多个不同特性的视觉基础模型(如DINOv3的空间表征 + SAM的分割能力 + CLIP的语义理解),构建更全面的视觉增强框架;(2)**动态特征注入**:设计自适应的特征注入机制,根据任务复杂度和视觉场景动态调整注入层数和特征量,平衡计算效率和性能;(3)**扩展到更复杂任务**:将DeepVision-VLA应用于双臂协调、长时间序列规划、动态环境中的操控等更复杂的场景,验证方法的通用性;(4)**理论分析深化**:深入研究VLA模型中视觉信息衰减的根本原因,探索是否可以通过架构设计(而非额外注入)来根本解决这一问题;(5)**轻量化设计**:设计更高效的Vision Expert融合方式,减少额外参数和计算开销,使方法更适合实际部署;(6)**时序视觉增强**:将AGVP的注意力分析扩展到时序维度,利用历史帧的视觉信息增强当前帧的视觉表征,提高动态场景中的操控稳定性;(7)**与其他VLA范式的结合**:将VL-MoT框架应用于不同的VLA架构(如π0的扩散策略、OpenVLA的自回归范式),验证框架的通用性。
复现评估
本文的复现评估如下:(1)**开源情况**:论文未明确说明代码和模型是否开源,但提供了详细的架构描述和训练配置,理论上可以复现。项目主页(https://deepvision-vla.github.io/)可能提供更多资源;(2)**数据可用性**:预训练数据来自公开数据集(Open X-Embodiment、DROID、RoboMIND),共400K+轨迹,数据获取可行。RLBench仿真环境是公开的。真实世界数据采集需要Franka Research 3机械臂和Intel RealSense D455相机,硬件门槛较高;(3)**算力需求**:训练使用8块NVIDIA H20 GPU,预训练1个epoch,微调300个epoch。DINOv3-H(0.8B)+ Qwen3-VL(4B)的模型规模对算力要求较高,但相比更大规模的VLA模型(如RT-2的55B)仍在可接受范围;(4)**复现难度**:中等偏高。主要挑战包括:多分辨率输入的实现、VL-MoT共享注意力的正确实现、AGVP的注意力插值和Top-K操作、以及多数据集的预处理流程。消融实验的详细描述有助于理解各组件的贡献。
论文图表
展示了三种VLA模型(OpenVLA、π0、QwenVLA-OFT)的层级视觉分析结果。上图显示遮蔽ROI视觉token后动作预测MSE随层数变化的曲线(浅层MSE显著增加,深层影响很小);下图展示各层的动作到视觉注意力可视化(浅层集中在任务相关区域,深层变得弥散)。
这是本文发现的核心现象——深层视觉敏感性衰减的定量和定性证据,是整个工作的实验基础