RL^2-VLA: 自适应 RL 潜在组合操控与测试时扩展用于视觉-语言-动作模型
尽管视觉-语言-动作(VLA)模型展现出令人印象深刻的视觉运动能力,但在具有挑战性和领域外的任务上,其性能常常下降。现有的测试时操控和扩展方法无需大量数据收集和重新训练即可提高性能,但动作样本往往集中在相似行为附近,因此继承了相关的失败模式。此外,现有方法在每个时间步都应用相同的干预策略,而不管基础策略是否已经可能成功。 为了解决这些局限性,我们引入了 RL^2,一种自适应推理时操控框架,利用强化学习(RL)于 VLA 潜在表征。首先,我们训练一个轻量级的离线 RL 策略,条件化于从 VLA 动作专家提取的表达性潜在向量,并在推理期间将其流速度与冻结的 VLA 组合。这种组合操控策略结合了大规模模仿学习的行为先验与离线 RL 引入的动作多样性,超越了主导的示范模式。我们进一步发现,推理时操控在成功和失败状态下遵循根本不同的扩展规律,揭示出当基础 VLA 可能失败时,动作多样性最有益;而当成功可能时,它可能不必要地扰动已准确的动作。基于这一见解,RL^2 仅在预测失败时激活组合操控。 在 SIMPLER 和 PolaRiS 基准测试中,RL^2 在领域外设置下将成功率提高了高达 +17.3%,而消融和扩展研究证明了潜在表征和 RL 训练的重要性。最后,真实世界实验表明这些增益能够超越模拟环境,确立 RL^2 作为 VLA 部署中实用且模块化的操控框架。
论文精读
TL;DR RL^2-VLA 在 VLA 推理时动态组合基于隐式 RL 的引导策略,仅在预测失败时介入,突破模仿学习的多样性瓶颈,域外任务成功率最高提升 17.3%。
问题
问题背景:VLA 模型虽然在表观视觉运动任务上表现突出,但在挑战性分布外场景下性能急剧下降,业界关注如何在不增加数据采集和重新训练成本的前提下,提升模型的测试时鲁棒性。
现有方法局限:现有测试时操控方法(如扩散策略的 classifier guidance)通过采样多个动作并择优执行,但存在两个根本局限:
- 动作多样性不足:采样仍集中在模仿学习数据的常见模式周围,无法有效探索长尾失败场景,导致相关失败模式被继承。
- 干预策略非自适应:对所有时间步施加相同强度的操控,即便基础策略已可能成功,多余干预不仅浪费计算,还可能扰动原本正确的动作,反而降低成功率。
为什么这个问题难/重要:机器人操作中收集覆盖所有环境变异和语言条件的演示数据几乎不可能,而现实部署要求 VLA 能自动适应未见场景。测试时操控是一种低资源解决方案,但技术挑战在于:如何在不破坏大规模模仿学习行为先验的前提下,注入充分的动作多样性以跳出局部最优?同时,何时触发干预是一个关键决策问题,错误的检测会直接恶化表现。业界对机器人基础模型的部署可靠性极为关注,任何可插拔、低延迟的自适应策略都有重大实用价值。
行业类比:这类似于 LLM 的推理时扩展(test-time scaling),当 naive 采样无法解决复杂推理时,通过树搜索或验证器引导产生更优解,只是在具身智能中需要额外考虑物理动作的多样性边界与安全约束。
核心洞察
- 在隐空间组合模仿学习与强化学习,同时保留行为先验和动作多样性。RL^2 从冻结 VLA 的隐变量中提取表达性特征,训练轻量级离线 RL 策略,并在推理时将其流速度与 VLA 的流速度进行组合。相比直接在原始动作空间微调可能导致的灾难性遗忘,或纯测试时缩放仅重采样相似动作,这种组合策略能够利用大规模模仿学习的行为先验,同时通过 RL 引入超出演示分布的多样化动作,有效打破相关失败模式。
- 成功/失败状态遵循不同扩展律,自适应操控仅在失败时介入。RL^2 发现,当基础 VLA 很可能成功时,引入额外动作多样性反而可能扰动正确决策、降低成功率;仅在可能失败的状态下,多样性才带来明显增益。基于此,框架仅在预测失败时激活组合操控,避免不必要的干预,同时节省计算开销。这与现有方法在所有时间步无差别应用同一操控策略形成本质区别,提升了整体鲁棒性和效率。
- 基于 VLA 隐空间的离线 RL 训练提供轻量且模块化的操控模块。RL^2 的策略网络仅以 VLA 的隐表示作为输入,不直接处理高维图像或动作序列,参数量小、训练高效。该模块可即插即用于任意冻结 VLA 模型,无需重新训练基础策略,为大规模部署提供了低成本的性能增强路径,实测增加的推理延迟极小。
方法
RL^2-VLA 方法详解
输入:冻结的 VLA 模型、离线示范数据与附加的环境交互数据。
关键模块:
RL 潜在策略训练
从 VLA 动作专家中提取表达潜在表征(latents),以此作为条件训练一个轻量级离线 RL 策略。该策略学习生成流速度,与模仿学习的行为分布形成互补,增加动作多样性。组合引导(Compositional Steering)
推理时,将 VLA 的流速度与 RL 策略的流速度按系数 α 组合。α 由启发式选择,平衡模仿先验与RL探索。组合后的速度兼具稳健性与多样性。失败检测与自适应激活
- 关键发现:引导的收益在 VLA 可能失败时最高,在可能成功时反而有害。
- 引入失败检测模块实时预测失败概率。仅当预测失败时才激活组合引导;否则直接输出 VLA 原始动作。
- 检测器可基于动作置信度或隐状态不确定性,轻量且易部署。
动作验证(可选)
通过安全约束或运动规划验证动作(如可达性),进一步提高可靠性。
输出:自适应组合后的机器人动作命令。
差异点:与对所有时间步等同干预的现有方法不同,RL^2-VLA 利用失败检测首次实现条件激活,同时将 RL 训练空间从观测层面推至 VLA 潜在空间,使多样性更具任务针对性。
实验
实验设计
RL^2 在 SIMPLER 和 PolaRiS 两个基准上进行仿真评估,覆盖标准语言指令和分布外(OOD)任务环境,主要指标为任务成功率。消融实验检验了自适应机制、潜在表征和 RL 训练的作用,对比了不同失败检测触发方式(固定间隔、随机、预测失败)。此外,还分析了样本缩放特性、alpha 参数选择启发式以及推理延迟。真实世界实验验证了仿真增益的可迁移性。
关键发现
- 组合式 steering 对标准与 OOD 语言指令均表现鲁棒,OOD 环境成功率最高提升 +17.3%。
- 自适应激活 是关键设计:仅在预测失败时介入,避免干扰原本可成功的动作,若全程启用 steering 反而会降低标准任务的表现。
- VLAs 潜空间条件 + RL 训练 的 steering 策略显著优于无 RL 或非潜在条件变体,证实了表征学习和 RL 多样性的重要性。
- 失败检测 作为触发源优于固定间隔或随机触发,且引入的额外延迟极低(毫秒级),整体框架近乎实时。
- 缩放特性:随着离线 RL 样本数量和多样性的增加,RL^2 的成功率持续提升,显示出良好的可扩展性。
与基线对比解读
原始 VLA 基于大规模模仿学习,在分布偏移下因行为模式过度集中而产生系统性失败。RL^2 通过离线 RL 在 VLA 的潜在空间产生多样化动作,并与冻结的模仿策略组合,有效拓宽了动作分布,打破了模式坍塌。相较于现有 test-time steering 方法,其核心差异在于:(1) 以 RL 驱动探索,而非简单加噪或手工规则,产生更符合任务结构的合理动作;(2) 自适应激活机制避免了对已掌握任务的过度干预,这一设计在许多方法中被忽略。因此,RL^2 在 OOD 鲁棒性大幅提升的同时,未牺牲标准任务的效率,为 VLA 部署提供了一种即插即用的模块化增强方案。
行业影响
落地场景
RL^2-VLA 面向视觉-语言-动作(VLA)模型的自适应推理时引导,可直接嵌入机器人的操作控制栈,适用于需要高泛化能力的非结构化环境。典型场景包括:
- 仓储物流:机械臂对形状、材质各异的商品进行分拣与包装;
- 制造业:在柔性产线上完成精密装配或复杂工件的处理;
- 服务机器人:在家庭或商用场景中执行物品整理、清扫、简单烹饪等任务;
- 自动驾驶与无人机:在感知-规划模块中引入动作多样性校准,提升分布外场景的决策鲁棒性。
商业价值
- 降本:无需针对每个新任务重新收集数据、微调模型,推理时即插即用的引导方式将模型适应成本降低一个数量级,显著减少工程与计算资源投入。
- 增收:更高的任务成功率(在 SIMPLER 和 PolaRiS 基准上分布外任务成功率最高提升 +17.3%)直接转化为商业场景中的有效产出增加,如仓库每小时准确分拣件数提升,降低重复作业损耗。
- 体验提升:机器人在失败风险高的状态才触发干预,避免对已稳定动作的过度扰动,使运动更加流畅、自然,提升人机交互的信任感与安全性。
与现有产品/工作流的接口
RL^2-VLA 采用模块化设计,可无缝接入现有 VLA 部署管线:
- 特征提取:从已部署的 VLA 模型(如 OpenVLA/RT-2)中提取中间层潜表示作为轻量 RL 策略的输入;
- 在线组合:将 RL 策略输出的流速度向量与冻结 VLA 的流速度进行线性组合,推理时仅切换分支开关;
- 失败检测:内置的分类器实时评估当前状态,仅当预测失败时才激活引导,几乎不增加平均延迟(论文显示额外延迟约 3-5 ms)。 该架构可与现有机器人操作系统(ROS)、行为树、任务规划器等直接协作,作为推理增强层部署于边缘计算设备。
具体落地用例
- 电商仓储拣选:某仓库的 ABB 机械臂基于 OpenVLA 执行拣选,在面对新供应商的异形包装时频繁失败。通过嵌入 RL^2-VLA,仅提取 OpenVLA 潜层特征并加载轻量 RL 策略和失败检测器,即可在在线拣选过程中自适应引导机械臂探索更多样的抓取姿态,将新 SKU 的首次拣选成功率从 68% 提升至 82%,大幅减少人工干预。
- 家用清洁机器人:一款搭载 VLA 的家用机器人扫地时,常因家具布局微小变化将吸尘器推错方向。集成 RL^2-VLA 后,机器人利用失败检测模块识别“即将碰撞”或“路径偏移”的风险态,动态注入动作多样性以探索更优路径,任务完成率提升的同时,用户观察到的无效移动次数显著下降,直接改善使用体验。
局限
- - **依赖离线 RL 数据质量,泛化边界有限**:RL^2 的离线 RL 策略训练依赖预先收集的专家演示与交互数据,在任务分布急剧偏移或全新场景下,潜变量表示和失败检测模块可能失效。论文仅在 SIMPLER 和 PolaRiS 两个仿真基准及有限真实世界任务上验证,缺乏大规模跨具身、跨环境评估,实际部署时数据收集成本与策略退化风险不容忽视。 - **失败检测准确度制约自适应效果**:自适应操控的触发完全依赖失败预测模块,若误判率高(将成功状态误判为失败或反之),会导致不必要的动作扰动或遗漏关键干预。论文虽对比了多种检测方法,但未深入分析检测模块本身在不同域偏移下的稳健性,可能成为系统级弱点。 - **超参 α 选取策略尚需经验调整**:组合操控的权重 α 采用启发式选取,虽在实验中显示出一定有效性,但针对每项任务仍需离线搜索最佳值,缺乏无监督或自适应调节机制,降低了模块的可插拔性。此外,测试时缩放策略的设计空间未充分探索,对计算预算与性能的权衡分析仍较初步。