VLS:通过视觉语言模型引导预训练机器人策略 VLS: Steering Pretrained Robot Policies via Vision-Language Models
提出训练无关的推理时引导框架,利用VLM生成可微奖励函数来引导冻结的机器人扩散策略适应分布外场景。
前置知识
扩散策略(Diffusion Policy)
扩散策略是一类基于去噪扩散概率模型(DDPM)的机器人动作生成方法。其核心思想是将动作生成过程建模为一个迭代去噪过程:从高斯噪声 $a^K_{t:t+T} \sim \mathcal{N}(0, I)$ 出发,通过训练好的噪声预测网络 $\epsilon(a^k, o, l, k)$ 逐步去噪,最终生成动作轨迹 $a^0_{t:t+T}$。在每个去噪步骤 $k$,网络根据当前噪声动作、观测 $o$ 和语言指令 $l$ 预测噪声,然后按照公式 $a^{k-1}_{t:t+T} = \frac{1}{\sqrt{\alpha_k}}(a^k_{t:t+T} - \frac{1-\bar{\alpha}_k}{\sqrt{1-\bar{\alpha}_k}}\epsilon(a^k_{t:t+T}, o, l, k)) + \sigma_k z$ 进行更新。这种方法能够建模多模态动作分布,但其生成过程与训练数据的空间配置紧密耦合。
理解扩散策略是理解本文核心贡献的前提,因为VLS正是通过干预扩散策略的去噪过程来实现推理时适应的。
流匹配(Flow Matching)
流匹配是另一种生成模型范式,它学习一个连续的速度场 $v$ 来建模从噪声分布到干净数据分布的转换过程。与扩散模型使用离散的去噪步骤 $\{K, K-1, ..., 0\}$ 不同,流匹配使用连续时间区间 $[0, 1]$,其中 $k=1$ 对应噪声分布,$k=0$ 对应干净动作轨迹。其动力学由常微分方程(ODE)描述:$\frac{da^k_{t:t+T}}{dt} = v(a^k_{t:t+T}, o, l, k)$。流匹配简化了去噪过程,同时保留了扩散模型的表达能力。在本文中,作者统一使用符号 $k$ 来表示两种框架中的去噪进程,避免冗余定义。
VLS框架需要同时适用于扩散策略和流匹配策略,因此理解两种生成范式的共同点和差异是理解本文通用性的关键。
分类器引导(Classifier Guidance)
分类器引导是一种在推理时引导生成模型的技术,最初由Dhariwal和Nichol在图像生成领域提出。其核心思想是在去噪过程中注入条件梯度信号,修改噪声预测或速度预测,使生成结果更好地满足给定条件。对于扩散模型,修改后的噪声预测为 $\hat{\epsilon} = \epsilon(a^k_{t:t+T}, (o,l)_{OOD}, k) - \lambda \cdot \sqrt{1-\bar{\alpha}_k} \cdot g(a^k_{t:t+T}, (o,l)_{OOD})$;对于流匹配策略,修改后的速度场为 $\hat{v} = v(a^k_{t:t+T}, (o,l)_{OOD}, k) + \lambda \cdot g(a^k_{t:t+T}, (o,l)_{OOD})$。其中 $\lambda$ 是控制引导强度的超参数,$g$ 是引导梯度函数。
VLS的核心技术贡献就是为机器人领域设计了有效的引导梯度函数 $g$,将其从简单的类别标签扩展到复杂的观测-语言条件对。
分布外(Out-of-Distribution, OOD)
在模仿学习语境下,OOD指的是测试时遇到的观测-语言对 $(o, l)_{OOD}$ 不在专家数据集 $\mathcal{D}_{expert}$ 中的情况。具体包括两类扰动:观测扰动($o_{OOD}$),如视觉背景变化、物体布局改变、新增干扰物等;语言扰动($l_{OOD}$),如未见过的指令、目标物体替换、任务逻辑改变等。论文指出,当前最先进的操作策略在面对这些OOD场景时会显著退化,即使所需的运动技能实际上已经存在于训练数据中,只是无法在新的空间结构下正确执行。
OOD问题是本文要解决的核心挑战,理解OOD的具体表现形式(观测扰动vs语言扰动)有助于理解VLS的设计动机和评估方案。
分段任意模型(Segment Anything Model, SAM)
SAM是由Kirillov等人于2023年提出的通用图像分割模型,能够根据提示(点、框、文本)对图像中的任意物体进行分割。在VLS中,SAM被用于从RGB-D观测中提取与任务相关的物体掩码 $\mathcal{M}$,这些掩码随后用于过滤DINOv2特征图,获得物体级别的语义特征。结合深度信息,这些掩码像素被重投影到3D点云空间,用于恢复物理结构信息。
SAM是VLS感知流程的基础组件,负责将视觉输入转化为结构化的物体表示,是实现OOD输入grounding的关键。
费曼-卡克引导(Feynman-Kac Steering)
费曼-卡克引导是一种基于粒子系统的梯度无关重采样方法,源于蒙特卡洛理论。在VLS中,一批动作提议被视为相互作用的粒子系统,在每个去噪步骤 $k$ 根据奖励势能进行重采样。具体地,第 $i$ 个粒子的势能定义为 $G^i_k = \exp(R_s(a^k_{t:t+T}[i], (o,l)_{OOD}))$,归一化权重 $w^i_k = G^i_k / \sum_{j=1}^{B} G^j_k$,然后进行多项式重采样。这使得高奖励粒子得以复制,低奖励粒子被修剪,有效地将生成过程的转移核倾斜向目标分布 $p(a | (o,l)_{OOD})$。
FK引导是VLS梯度无关探索机制的核心,与梯度引导互补,共同实现了在复杂多模态约束景观中的高效导航。
研究动机
当前基于扩散或流匹配目标的预训练机器人策略在分布内(in-distribution)任务上表现出色,但在面对轻微的环境变化时会严重退化。论文用一个直观的例子说明:一个学会了将杯子放在桌子中央的机器人,在被要求将杯子放在桌边、堆叠的书上或拥挤的柜子里时会失败。这种失败并非源于运动技能的缺失,而是因为动作生成与训练时特定的空间配置紧密耦合。具体来说,当测试时的观测-语言对 $(o, l)_{OOD} \notin \mathcal{D}_{expert}$ 时,基础策略 $\pi^\star$ 倾向于过拟合训练流形中的空间和语义相关性,导致在环境配置偏离训练分布时无法产生满足约束的行为。论文指出,通过重新训练或微调来解决这些问题是昂贵且概念上不匹配的,因为所需的行为已经存在于训练数据中,只是无法在测试时被选择性地适应。此外,扩展训练分布以覆盖所有可能的空间变化是对根本上属于推理时控制问题的暴力解决方案。
本文的目标是本文的目标是提出一个训练无关(training-free)的推理时适应框架,使预训练的机器人策略能够在不修改策略参数的情况下适应分布外场景。具体而言,给定一个冻结的基础策略 $\pi^\star$,VLS旨在通过在推理时重塑动作分布,使得生成的轨迹满足 $(o, l)_{OOD}$ 所隐含的空间和任务结构要求。这一目标的关键在于将技能执行与OOD任务规格解耦:基础策略提供可复用的运动原语(motor primitives),而推理时引导控制这些原语如何在OOD输入下被组合和实例化。论文希望实现的效果是:一个学会了在桌子中央放置物体的机器人,能够通过推理时引导可靠地将物体放置在桌边、书堆上或其他任意位置,而无需任何额外训练。
与已有工作不同的是,本文的独特切入角度在于将机器人策略适应问题重新定义为推理时控制问题,并借鉴大语言模型和图像生成模型中的推理时引导技术。与现有方法相比,VLS有三个关键差异:第一,与值函数/评论家引导方法(如V-GPS、VGD)不同,VLS不依赖辅助学习目标,而是直接利用VLM的开放世界理解能力生成任务特定的引导信号,避免了基础策略被评论家偏好重塑的问题;第二,与动力学/世界模型引导方法(如DynaGuide)不同,VLS不需要预测性建模或滚动式评估,而是通过VLM合成可微的程序化奖励函数,直接提供密集的动作空间梯度;第三,与人工/VLM在环的验证方法(如ITPS、FOREWARN)不同,VLS提供连续可微的引导而非离散的选择/拒绝,实现了细粒度的约束满足。这种角度的核心洞察是:VLM的跨模态推理能力可以被转化为可微的奖励函数,从而为冻结策略的去噪过程提供精确的梯度引导。
核心方法
VLS的核心思想是将推理时适应建模为一个可控制的去噪过程。直觉上,当机器人面对OOD场景时,基础策略的去噪过程会朝着训练时学到的模式方向去噪,但这些模式可能不满足新的空间约束。VLS的做法是在去噪过程中注入额外的梯度信号,将去噪轨迹"拉向"满足OOD约束的方向,同时保持基础策略的运动能力不变。技术路线分为三个阶段:首先,OOD输入grounding阶段,利用SAM和DINOv2将观测-语言对压缩为一组任务相关的3D关键点 $\mathcal{P} = \{p_i\}_{i=1}^n$,每个关键点 $p_i \in \mathbb{R}^3$ 锚定一个物理上有意义的空间约束;其次,程序化奖励生成阶段,利用VLM的跨模态推理能力将任务分解为 $S$ 个顺序阶段,并为每个阶段 $s$ 生成可微的奖励函数 $R_s(a^k_{t:t+T}, (o,l)_{OOD})$,该函数实现为PyTorch程序,由可微的张量操作(距离、点积、软约束)组成;最后,推理时去噪引导阶段,将奖励梯度 $g_s = \nabla_{a^k_{t:t+T}} R_s$ 注入扩散或流匹配的去噪更新中,结合RBF斥力项和费曼-卡克重采样机制,引导动作分布收敛到高奖励区域。
VLS的核心创新点在于利用VLM合成可微的、阶段感知的程序化奖励函数来近似引导信号 $g \triangleq \nabla_{a^k_{t:t+T}} \log p((o,l)_{OOD} | a^k_{t:t+T})$。与已有方法的本质区别体现在三个层面:第一,引导信号的来源不同。DynaGuide使用预训练DINO特征空间中的距离作为启发式引导,这种引导缺乏表达能力来捕获任务特定的空间需求;ITPS从预定义的引导函数集合中选择,受限于函数库的覆盖范围;而VLS利用VLM根据当前观测和指令动态合成定制化的奖励函数,实现了对任意OOD条件的灵活建模。第二,引导的连续性不同。ITPS和FOREWARN等方法提供离散的监督信号(选择/拒绝候选动作),迫使适应通过选择发生,样本效率低下;VLS提供连续可微的梯度引导,实现了在生成过程内部的精细控制。第三,探索机制的设计不同。VLS创新性地结合了三种互补机制:RBF斥力确保动作提议的多样性覆盖,梯度引导实现局部精细优化,FK重采样实现全局高效的粒子重分配。这种组合使得冻结策略能够在复杂的多模态约束景观中高效导航,避免了纯选择方法的样本低效和脆性问题。
方法步骤详情
VLS的完整算法包含以下步骤。第一步,OOD输入grounding:给定OOD输入对 $(o,l)_{OOD}$,使用VLM识别与操作任务相关的物体和区域,应用SAM获取物体掩码 $\mathcal{M}$,使用DINOv2提取语义对齐的密集视觉特征图 $\Phi \in \mathbb{R}^{H \times W \times d}$,将掩码像素重投影到3D点云空间,得到 $(d+3)$ 维表示,聚类获得任务相关关键点集 $\mathcal{P} = \{p_i\}_{i=1}^n$。第二步,程序化奖励生成:VLM查询以分解任务为 $S$ 个顺序阶段,并为每个阶段 $s$ 生成可微奖励函数 $R_s(a^k_{t:t+T}, (o,l)_{OOD}) = f_{VLM}(a^k_{t:t+T}, \mathcal{P}, s)$,该函数实现为PyTorch程序。第三步,多样化提议初始化:在每个环境时间步 $t$,采样 $B$ 个动作提议 $\{a^K_{t:t+T}[i] \sim \mathcal{N}(0, I)\}_{i=1}^B$,引入RBF斥力梯度 $g^{RBF}_k[i] = \nabla_{a^k_{t:t+T}[i]} \sum_{j \neq i} \frac{1}{\|a^k_{t:t+T}[i] - a^k_{t:t+T}[j]\|^2 + \epsilon}$ 防止提议过早塌缩。第四步,梯度引导优化:在每个去噪步骤 $k$,注入阶段特定的奖励梯度 $g_s = \nabla_{a^k_{t:t+T}} R_s$,采用随机优化进行多次内部更新以提高稳定性。第五步,FK重采样:定义势能 $G^i_k = \exp(R_s(a^k_{t:t+T}[i], (o,l)_{OOD}))$,计算归一化权重 $w^i_k = G^i_k / \sum_{j=1}^{B} G^j_k$,进行多项式重采样。第六步,闭环执行控制:根据当前奖励值相对于基准奖励的比值自适应调整引导强度 $\lambda_t = \lambda_{max} \cdot \text{sigmoid}(1 - R^s_t / R^s_{base})$,使用施密特触发器机制判断阶段切换信号 $Q_t$,避免边界处的振荡行为。
技术新颖性
VLS的技术新颖性体现在多个维度。首先,在问题建模上,VLS首次将机器人策略的OOD适应问题形式化为推理时控制问题,明确解耦了技能执行(由冻结的基础策略提供)和任务规格适应(由推理时引导控制),这一建模角度为该领域提供了新的思考框架。其次,在技术路线上,VLS创新性地将VLM的语义理解能力转化为可微的奖励函数,实现了从非可微的VLM到可微引导信号的"桥梁"构建。具体而言,VLM被约束输出程序化的奖励定义,实现为由可微张量操作组成的PyTorch函数,推理时梯度通过实例化的奖励函数反向传播,而VLM本身保持非可微、离图状态。这种设计既保留了VLM的强大推理能力,又满足了梯度引导的可微性要求。第三,在引导机制上,VLS首次在机器人策略引导中同时结合了三种互补机制:基于梯度的局部精化、基于RBF的多样性保持、基于FK理论的全局重采样。这种组合实现了探索与利用的有效平衡,克服了现有方法要么只提供离散监督(ITPS)、要么依赖启发式引导(DynaGuide)的局限。第四,在执行控制上,VLS引入了施密特触发器机制进行阶段切换,通过迟滞效应避免边界处的振荡,实现了对多阶段长时域任务的鲁棒协调。
实验结果
论文在仿真和真实世界环境中进行了全面评估,得出以下核心发现。第一,推理时引导对于处理观测和语言偏移是必要的。在LIBERO-PRO基准上,四个领先的VLA模型(OpenVLA、π0、π0.5、π0.5 LeRobot微调版)在联合观测-语言扰动下表现急剧下降,尽管它们使用了预训练的VLM骨干网络进行感知和指令理解。具体而言,OpenVLA和π0在所有OOD条件下成功率为0%,π0.5在任务扰动下成功率为0.75%,即使表现最好的π0.5 LeRobot版本也只有23.69%的总体成功率。这表明后训练在机器人数据上将空间推理与特定训练上下文纠缠,实际上削弱了VLM在执行环境偏离训练流形时的泛化能力。第二,VLS显著优于现有引导方法。在CALVIN基准上,VLS在可移动物体任务上达到94%的平均成功率,相比基础策略提升7.4倍;在铰接部件任务上达到87%成功率,提升9.6倍。相比ITPS和DynaGuide,VLS高出15-25个百分点。在LIBERO-PRO上,VLS将π0.5 LeRobot版本的总体成功率从23.69%提升至36.81%,提升幅度达13%。第三,消融研究证实了各组件的贡献。移除梯度引导导致性能严重崩溃,成功率降至接近失败水平,确认密集的轨迹可微引导是VLS有效性的主要驱动力。移除FK重采样或RBF多样性对成功率影响较小,但一致地降低了效率和稳定性。第四,真实世界部署验证了VLS的实用性。在Franka Emika机器人上,VLS在分布内任务上将平均成功率从50%提升至69%(提升19%),在最困难的物体级OOD情况下(目标物体被替换为未见过的马克杯),基线完全失败,而VLS在40%的试验中成功。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| LIBERO-PRO 任务扰动(Goal子集) | 成功率(%) | 33.50 | π0.5 LeRobot: 12.00 | +21.50百分点 |
| LIBERO-PRO 任务扰动(Spatial子集) | 成功率(%) | 54.00 | π0.5 LeRobot: 48.50 | +5.50百分点 |
| LIBERO-PRO 任务扰动(Object子集) | 成功率(%) | 41.00 | π0.5 LeRobot: 10.50 | +30.50百分点 |
| LIBERO-PRO 位置扰动(Overall) | 成功率(%) | 35.13 | π0.5 LeRobot: 24.25 | +10.88百分点 |
| LIBERO-PRO 总体 | 成功率(%) | 36.81 | π0.5 LeRobot: 23.69 | +13.12百分点 |
| CALVIN 可移动物体(立方体) | 平均成功率(%) | 94 | 基础扩散策略: ~12.7 | 7.4倍提升 |
| CALVIN 铰接部件(抽屉、开关、按钮、门) | 平均成功率(%) | 87 | 基础扩散策略: ~9.1 | 9.6倍提升 |
| 真实世界 分布内任务 | 平均成功率(%) | 69 | π0.5: 50 | +19百分点 |
局限与改进
论文明确指出了VLS的一个主要局限性:计算延迟。批采样、MCMC运行和FK重采样引入了较高的推理开销。具体而言,从Figure 4的运行时缩放分析可以看出,当批大小 $B$ 从1增加到10时,推理时间从178.6ms增加到1239ms,这在需要实时控制的场景中可能是不可接受的。此外,作者也暗示未来工作可以探索进度感知的奖励信号生成和推理时计算效率优化。从我的观察来看,还有几个未被充分讨论的局限性:首先,VLS高度依赖VLM的质量来生成准确的奖励函数,如果VLM对场景的理解出现偏差,引导信号可能会误导策略;其次,当前的阶段切换机制需要VLM查询来确定下一阶段,这增加了额外的延迟和不确定性;第三,论文的评估主要集中在桌面操作任务上,对于更复杂的长时域任务(如烹饪、清洁)的适用性尚未验证;第四,VLS假设基础策略的运动原语足以覆盖OOD场景所需的行为,但如果OOD场景需要全新的运动模式,VLS可能无法有效适应。
独立分析的弱点
VLS存在以下几个值得深入分析的弱点。第一,计算效率瓶颈:VLS在每个环境时间步需要进行多轮去噪迭代,每轮包含梯度计算、RBF斥力计算和FK重采样,当批大小 $B=10$、去噪步骤 $K=10$ 时,推理时间超过1秒。在需要高频控制的动态环境中(如抓取移动物体),这种延迟可能导致任务失败。改进方向包括:开发轻量级的奖励函数近似网络,避免每步都调用VLM;采用分层引导策略,只在关键决策点注入引导信号;探索异步引导机制,将引导计算与动作执行解耦。第二,VLM依赖的脆弱性:VLS的奖励函数质量完全取决于VLM的场景理解能力,但VLM在处理遮挡、透明物体、精细空间关系时可能存在系统性偏差。当VLM生成的奖励函数与真实约束不一致时,引导信号会将策略推向错误方向。改进方向包括:引入奖励函数验证机制,通过少量环境交互验证奖励函数的准确性;设计鲁棒的奖励函数模板库,降低对VLM创意生成的依赖;结合传统几何约束作为VLM生成奖励的正则化项。第三,阶段切换的粗糙性:当前的施密特触发器机制基于单一的奖励阈值进行阶段判断,对于需要细粒度阶段划分的任务(如精细装配)可能不够灵活。改进方向包括:设计连续的阶段进度表示,替代离散的阶段切换;引入基于不确定性的自适应阶段划分,当当前阶段的奖励不确定性高时自动细分。第四,缺乏长期记忆:VLS在每个时间步独立生成奖励函数,没有利用之前的执行历史来改进当前的引导决策。对于需要试错学习的场景(如探索未知物体的可操作性),这种缺乏记忆的机制可能效率低下。改进方向包括:引入执行历史编码器,将之前的尝试结果融入当前的奖励生成过程。
未来方向
论文作者提出了两个未来研究方向:进度感知的奖励信号生成和推理时计算效率优化。基于VLS的成果,还可以延伸出以下研究方向。第一,自适应奖励函数学习:当前VLS依赖VLM在每个时间步从头生成奖励函数,未来可以探索在线学习机制,根据执行反馈微调奖励函数,逐步减少对VLM调用的依赖。这可以通过维护一个轻量级的奖励函数参数化模型,利用执行轨迹的监督信号进行快速适应。第二,多策略协作引导:VLS当前针对单一基础策略进行引导,未来可以扩展到多策略场景,根据不同策略的优势在推理时动态组合或切换策略。例如,在粗定位阶段使用快速但不精确的策略,在精细操作阶段切换到慢速但精确的策略。第三,跨任务迁移:VLS的OOD适应能力可以被用于实现零样本跨任务迁移。通过设计通用的奖励函数模板,使得在一个任务上学到的引导模式可以迁移到结构相似但物体或场景不同的新任务。第四,人机协作引导:VLS的框架可以自然地扩展到人机协作场景,将人类的实时反馈(如手势、语音修正)融入奖励函数生成过程,实现更直观的策略引导。第五,理论分析:当前VLS的收敛性和最优性保证主要基于经验验证,未来可以从理论上分析引导梯度的性质、FK重采样的收敛速度、以及整体框架在什么条件下能够保证找到满足OOD约束的动作。
复现评估
从复现评估的角度来看,VLS具有较好的可复现性基础。论文使用了广泛认可的基准测试(CALVIN和LIBERO-PRO),这些基准都有公开的代码和数据。评估的基线方法(OpenVLA、π0、π0.5)都是开源的VLA模型,有公开的模型权重和推理代码。VLS的核心组件(SAM、DINOv2、VLM)也都是公开可用的模型。然而,论文的复现也面临一些挑战。首先,VLS的性能高度依赖VLM的质量,但论文没有明确说明使用的是哪个具体的VLM模型(如GPT-4V、Claude 3等),不同的VLM可能导致显著的性能差异。其次,VLS的超参数设置(批大小 $B$、去噪步骤 $K$、引导强度 $\lambda$、MCMC迭代次数等)需要针对不同任务进行调优,论文虽然提供了默认值(如 $B=10$、$K=10$、MCMC=4 for diffusion),但没有提供详细的超参数敏感性分析。第三,真实世界实验的硬件配置(Franka Emika机器人、RGB-D相机型号等)和软件环境(ROS版本、控制器频率等)细节需要从附录中获取。第四,VLS的计算需求较高,在 $B=10$ 时单步推理需要约1.2秒,这需要具有足够算力的GPU(论文未明确说明使用的GPU型号)。总体而言,如果具备适当的硬件条件和VLM访问权限,VLS的核心算法可以在合理时间内复现,但精确复现论文报告的所有结果可能需要仔细匹配VLM版本和超参数设置。
论文图表
左图展示了分布内任务的布局、语言指令和成功率。Level 1要求根据指令将橙子放在指定盘子上(红色或绿色),Level 2引入额外物体(香蕉),需要顺序选择目标物体和目标盘子。右图展示了三种OOD变体的结果:外观偏移(用未见过的黄色盘子替换红/绿盘子)、位置偏移(交换两个盘子的位置)、物体偏移(用未见过的马克杯替换香蕉)。每个任务评估20次试验,抓取正确物体贡献50%成功率,完全任务完成贡献100%。
真实世界部署是验证VLS实用性的关键证据,展示了VLS在物理机器人上处理各种OOD情况的能力,特别是最困难的物体替换情况下基线完全失败而VLS达到40%成功率。
该表格报告了VLA基线和冻结π0.5策略在有/无VLS情况下的成功率。实验环境包含LIBERO-PRO的任务和位置扰动,应用于LIBERO的四个子集:Goal、Spatial、10(Long)和Object,每个子集包含10个任务。每个任务在每个扰动下测试20个episode,报告平均成功率(%)。结果显示,尽管VLA模型使用了预训练的VLM骨干网络,但在联合观测-语言扰动下成功率急剧下降。VLS将π0.5 LeRobot版本的总体成功率从23.69%提升至36.81%,在任务扰动的Object子集上提升最为显著(从10.50%到41.00%)。
这个表格是论文最核心的量化结果,系统地展示了VLS在多种OOD条件和多个VLA基线上的性能提升,是评估VLS有效性的主要依据。