Fast-ThinkAct:通过可语言化的潜在规划实现高效的视觉-语言-动作推理 Fast-ThinkAct: Efficient Vision-Language-Action Reasoning via Verbalizable Latent Planning
通过压缩链式推理到紧凑的连续潜在表示,实现视觉-语言-动作任务的高效推理
前置知识
视觉-语言-动作模型(VLA)
VLA是结合视觉感知、语言理解和动作生成的基础模型,通常基于大型视觉-语言模型(VLM)骨干,并在机器人演示数据上训练以学习控制策略。这些模型能够处理复杂的视觉场景,理解自然语言指令,并在动态环境中执行适应性动作,是实现通用具身智能的关键技术。典型代表包括OpenVLA、π0等。
本文的核心研究对象就是VLA模型,需要理解其基本架构和挑战,才能理解推理延迟问题的严重性
链式推理(Chain-of-Thought, CoT)
CoT是一种让模型通过生成中间推理步骤来解决复杂问题的技术。在VLA任务中,CoT可以改善泛化能力,但传统的文本CoT生成冗长的推理链(如约250个token),导致显著的推理延迟。这种延迟在需要实时决策的具身AI应用中成为关键瓶颈。
本文的核心动机就是解决CoT带来的推理延迟问题,理解传统CoT的工作机制和局限性至关重要
强化学习与GRPO
GRPO(Group Relative Policy Optimization)是一种强化学习方法,通过组内相对奖励来优化策略。在本文中,GRPO用于训练教师模型生成文本推理链,并利用优势函数作为质量指标来构建偏好对。这种方法能够从多个推理轨迹中识别高质量和低质量的推理模式,为后续的偏好蒸馏提供训练信号。
GRPO是本文教师模型训练的核心算法,理解其工作原理才能理解偏好对的构建和蒸馏过程
知识蒸馏与偏好优化
知识蒸馏是将大型教师模型的知识转移到小型学生模型的技术。在本文中,蒸馏过程不仅涉及传统的能力转移,还结合了偏好优化,使用类似DPO的目标函数来引导学生模型学习高质量推理模式并抑制低质量模式。这种偏好引导的蒸馏确保了压缩后的潜在表示能够保留关键的推理能力。
这是本文方法论的核心创新点,理解偏好蒸馏机制才能理解如何将冗长文本推理压缩到紧凑的连续表示
潜在空间推理
与在离散文本空间中进行推理不同,潜在空间推理在连续的向量空间中进行。这种推理方式可以避免生成冗长的文本token,而是将推理过程编码为紧凑的连续表示。在VLA任务中,潜在推理需要处理时空理解,并桥接语义推理与具身控制,这与纯文本LLM的潜在推理有本质区别。
这是本文解决推理延迟的关键技术路径,理解潜在推理的优势和挑战才能把握论文的技术贡献
研究动机
视觉-语言-动作(VLA)任务要求在复杂视觉场景中进行推理并在动态环境中执行适应性动作。尽管最近的研究表明显式链式推理(CoT)可以改善泛化能力,但这些方法存在严重的推理延迟问题。具体来说,生成冗长的推理链需要每个决策数秒时间(例如0.1 Hz),而具身AI应用如机器人操作和自动驾驶需要高频决策(1-15 Hz)。例如,ThinkAct-7B模型需要7513毫秒进行推理,这完全无法满足实时要求。这种延迟不仅限制了实时性能,还在时间关键场景中带来安全风险。现有的高效推理方法如ECoT-Lite通过推理丢弃来加速,但直接减少文本推理长度会导致性能下降,因为可能丢失关键信息。
本文的目标是本文旨在解决VLA任务中推理能力与推理效率之间的核心矛盾。具体目标是:在保持强大推理能力(包括长程规划、失败恢复和少样本适应)的同时,显著降低推理延迟。作者希望通过引入一种新的推理框架,将冗长的文本推理压缩到紧凑的连续潜在表示中,实现高达89.3%的推理延迟降低,同时维持甚至提升任务性能。
与已有工作不同的是,本文的独特切入角度在于提出可语言化的潜在推理概念。与现有方法要么生成冗长文本推理(高延迟)要么直接减少推理长度(性能下降)不同,Fast-ThinkAct首次将潜在推理技术应用于VLA任务。关键创新在于:1)通过奖励引导的偏好蒸馏将文本推理压缩到连续潜在空间;2)引入视觉轨迹对齐来确保潜在表示编码了视觉规划能力;3)通过推理增强策略学习将高层规划与低层动作执行连接起来。这种方法不仅解决了延迟问题,还通过偏好优化过滤了冗余信息,使推理更加紧凑和高效。
核心方法
Fast-ThinkAct的整体思路是采用教师-学生框架,通过偏好引导的蒸馏将冗长的文本推理压缩到紧凑的连续潜在表示。具体来说,教师VLM(文本教师)首先通过GRPO训练生成显式推理链,然后学生VLM(潜在学生)学习将这些推理压缩为连续潜在向量。这些潜在向量可以通过一个语言化解码器(verbalizer)解码回自然语言,确保推理模式的可解释性。同时,通过视觉轨迹对齐确保潜在表示编码了视觉规划能力。最后,通过推理增强策略学习将高层规划连接到低层动作执行,实现端到端的高效推理和动作生成。整个过程可以形式化为:给定视觉观测 $o_t$ 和语言指令 $l$,模型生成 $M$ 个连续潜在向量 $\mathbf{z} = \{z_m\}_{m=1}^M$,其中 $z_m \in \mathbb{R}^d$,$d$ 是隐藏维度。
Fast-ThinkAct的核心创新点在于首次将潜在推理技术应用于VLA任务,并结合偏好优化和视觉轨迹对齐。与传统方法的本质区别在于:1)压缩表示:将约250个文本token的推理链压缩为仅6个连续潜在token,同时保留关键推理信息;2)偏好引导:利用教师模型的GRPO优势函数构建偏好对,通过类似DPO的目标函数引导学生学习高质量推理模式;3)视觉轨迹对齐:通过L2距离对齐教师和学生的轨迹级表示,确保视觉规划能力的转移;4)并行空间预测:使用可学习的空间token进行并行轨迹预测,避免自回归生成冗长文本。这种设计不仅大幅降低推理延迟(89.3%),还通过过滤冗余信息提升了推理质量。偏好蒸馏的目标函数为:$$\mathcal{L}_{ ext{verb}} = -\mathbb{E}\left[\log \sigma \left(?eta \left(\log ?rac{p_\psi( au^+|\mathbf{z})}{p_{ ext{ref}}( au^+)} - \log ?rac{p_\psi( au^-|\mathbf{z})}{p_{ ext{ref}}( au^-)} ight) ight) ight]$$
方法步骤详情
Fast-ThinkAct的方法包含三个主要阶段:1)教师GRPO训练:教师VLM在观测-指令对 $(o_t, l)$ 上通过GRPO训练生成文本推理链 $ au$,使用轨迹奖励计算优势函数 $A( au) = ?rac{R_ au - ext{mean}(\{R_i\}_{i \in G( au)})}{ ext{std}(\{R_i\}_{i \in G( au)})}$,构建偏好对 $( au^+, au^-)$。2)学生潜在蒸馏:学生VLM生成 $M$ 个连续潜在向量 $\mathbf{z} = \{z_m\}_{m=1}^M$,通过语言化损失 $\mathcal{L}_{ ext{verb}}$(类似DPO)确保潜在表示可解码为高质量推理;通过蒸馏损失 $\mathcal{L}_{ ext{distill}} = \|h_t^T - h_t\|_2^2$ 对齐教师和学生的隐藏状态;通过预测损失 $\mathcal{L}_{ ext{ans}} = \sum_{i=1}^K \|p_i - \hat{p}_i\|_2^2$ 预测 $K$ 个空间路标点。3)推理增强策略学习:提取空间token的KV缓存作为视觉潜在规划 $c_t$,连接到动作模型 $\pi_\phi$,通过模仿学习目标 $\mathcal{L}_{ ext{IL}}(\phi) = \ell(\pi_\phi(o_t, l, c_t), \hat{a}_t)$ 进行训练。推理时,学生VLM处理 $(o_t, l)$ 生成潜在推理和空间路标,动作模型基于 $c_t$ 预测动作 $a_t$。
技术新颖性
Fast-ThinkAct的技术新颖性体现在多个方面:首先,这是首次将潜在推理技术应用于VLA任务,解决了LLM潜在推理方法无法直接转移到VLA任务的挑战(需要时空理解和语义推理与具身控制的桥接)。其次,偏好引导的蒸馏机制创新性地利用GRPO优势函数构建偏好对,通过类似DPO的目标函数引导学生学习高质量推理模式,这比传统的知识蒸馏更有效。第三,视觉轨迹对齐确保了潜在表示编码了视觉规划能力,而不仅仅是语言推理模式。第四,并行空间token设计避免了自回归生成冗长文本,通过MLP同时预测多个路标点。最后,推理增强策略学习通过KV缓存连接高层规划与低层动作,实现了端到端的高效推理和执行。
实验结果
Fast-ThinkAct在多个具身推理和机器人操作基准上实现了最先进的性能,同时大幅降低推理延迟。在LIBERO基准上,Fast-ThinkAct在四个子任务(Spatial、Object、Goal、Long)上均取得最高成功率:92.0%、97.2%、90.2%、79.4%,显著超过OpenVLA、CoT-VLA、ThinkAct和MolmoAct等基线。在SimplerEnv-Google基准上,Fast-ThinkAct达到68.7%的成功率,超过ThinkAct-7B的68.3%。在RoboTwin2.0双臂操作基准上,Fast-ThinkAct在简单和困难设置下分别达到65.7%和26.4%的平均成功率,比RDT基线提升9.3%和3.6%。在具身推理基准上,Fast-ThinkAct在EgoPlan-Bench2上达到46.4%准确率,超过GPT-4V和Gemini-2.5-Flash;在RoboVQA上达到60.8 BLEU分数;在OpenEQA上达到51.2分。最关键的是效率提升:Fast-ThinkAct的推理延迟仅为805毫秒,相比ThinkAct-7B的7513毫秒降低89.3%,相比MolmoAct-7B的6723毫秒降低88.0%,相比ThinkAct-3B的5674毫秒实现7倍加速。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 机器人操作 - LIBERO-Spatial | 成功率 (%) | 92.0 | 88.3 (ThinkAct), 87.5 (CoT-VLA), 84.7 (OpenVLA) | +3.7% (vs ThinkAct) |
| 机器人操作 - LIBERO-Object | 成功率 (%) | 97.2 | 91.6 (CoT-VLA), 91.4 (ThinkAct), 88.4 (OpenVLA) | +5.6% (vs CoT-VLA) |
| 机器人操作 - LIBERO-Goal | 成功率 (%) | 90.2 | 87.6 (CoT-VLA), 87.1 (ThinkAct), 79.2 (OpenVLA) | +2.6% (vs CoT-VLA) |
| 机器人操作 - LIBERO-Long | 成功率 (%) | 79.4 | 77.2 (MolmoAct), 70.9 (ThinkAct), 69.0 (CoT-VLA) | +2.2% (vs MolmoAct) |
| 机器人操作 - SimplerEnv-Google | 成功率 (%) | 68.7 | 68.3 (ThinkAct), 64.9 (MolmoAct), 40.2 (OpenVLA) | +0.4% (vs ThinkAct) |
| 具身推理 - EgoPlan-Bench2 | 准确率 (%) | 46.4 | 44.0 (ThinkAct), 42.4 (Gemini-2.5-Flash), 32.6 (GPT-4V) | +2.4% (vs ThinkAct) |
| 具身推理 - RoboVQA | BLEU分数 | 60.8 | 55.3 (ThinkAct), 46.5 (RoboBrain2.0), 34.8 (Qwen2.5-VL) | +5.5 (vs ThinkAct) |
| 推理延迟 | 毫秒 (ms) | 805 | 7513 (ThinkAct-7B), 6723 (MolmoAct-7B), 5674 (ThinkAct-3B) | 降低89.3% (vs ThinkAct-7B) |
局限与改进
Fast-ThinkAct存在几个局限性:首先,语言化解码器(verbalizer)基于预训练的LLM构建,不可避免地继承了语言模型的局限性,包括幻觉问题,偶尔会产生看似合理但不准确的描述。不过,这并不影响推理时的动作执行,因为语言解码器仅用于可解释性,而动作预测使用来自视觉规划蒸馏的 grounded 潜在表示。其次,潜在推理的压缩可能丢失某些细节信息,虽然实验表明6个潜在token是最优平衡点,但在更复杂的任务中可能需要调整。第三,方法依赖于预训练的VLM骨干,对计算资源有一定要求,虽然推理效率大幅提升,但训练过程仍需要大量GPU资源(16个A100 GPU)。最后,虽然方法在多个基准上验证,但在真实物理机器人上的部署还需要进一步验证 sim-to-real 的迁移效果。
独立分析的弱点
Fast-ThinkAct的主要弱点包括:1)语言解码器的幻觉问题:虽然不影响动作执行,但可能影响可解释性和调试,需要引入定位感知目标或幻觉抑制技术来提高语言化推理的忠实度。2)潜在推理的固定长度:当前使用固定的M=6个潜在token,可能不是所有任务的最优配置,可以探索自适应长度机制。3)视觉轨迹对齐的局限性:当前的L2距离对齐可能过于简单,可以考虑更复杂的对齐损失如对比学习来更好地转移视觉规划能力。4)教师模型的依赖:需要先训练教师模型,增加了训练复杂性和时间,可以探索更高效的蒸馏方法。5)对动作模型的选择敏感:虽然论文验证了对不同动作模型的兼容性,但最优配置可能需要针对特定任务调整。
未来方向
基于Fast-ThinkAct的成果,未来研究可以探索多个方向:1)结合定位感知目标和幻觉抑制技术,进一步提高语言化推理的忠实度和可解释性。2)探索自适应潜在推理长度,根据任务复杂度动态调整潜在token数量。3)将潜在推理扩展到更复杂的具身任务,如长期导航、多机器人协作等。4)研究更高效的蒸馏方法,减少对教师模型的依赖。5)探索潜在推理在其他模态(如触觉、力觉)的扩展。6)在真实物理机器人上进行大规模部署和测试,验证 sim-to-real 迁移效果。7)研究潜在推理与强化学习的进一步结合,实现在线适应和持续学习。
复现评估
Fast-ThinkAct的复现性评估:论文提供了相对详细的实验设置和训练细节,使用Qwen2.5-VL 3B作为VLM骨干,基于开源模型和数据集。训练过程需要大量计算资源(16个A100 GPU,80GB内存),这对许多研究机构可能构成挑战。数据集方面,论文使用了多个公开数据集(OXE、AIST、RoboVQA等),但某些数据处理细节(如视觉轨迹标注)可能需要额外工作。代码和模型权重是否开源尚不明确,如果开源将极大促进复现。总体而言,方法原理清晰,实验设置详细,但计算资源需求较高,复现难度中等。
论文图表
图5展示了Fast-ThinkAct在RoboFAC基准上的失败恢复能力。左侧是模拟和真实机器人的定性例子,展示如何识别错误并提供恢复指导。右侧是定量评估,Fast-ThinkAct在模拟和真实设置上分别超过第二名基线10.9和16.4分。
这张图展示了Fast-ThinkAct的关键优势之一:能够识别执行失败并提供具体恢复计划,这对机器人实际应用至关重要。
表1展示了在RoboTwin2.0基准上的定量评估结果,包含10个任务的简单和困难设置。Fast-ThinkAct在平均性能上显著超过DP、ACT、π0、RDT和ThinkAct基线,在简单和困难设置下分别达到65.7%和26.4%的成功率。
这个表格提供了Fast-ThinkAct在复杂双臂操作任务上的详细性能数据,是验证方法有效性的关键证据。
表4展示了将Fast-ThinkAct扩展到更大模型(7B)时在具身推理基准上的结果。Fast-ThinkAct-7B在所有基准上均取得 strong 性能,验证了方法的可扩展性。
这个表格证明了Fast-ThinkAct的方法不仅适用于3B模型,还能有效扩展到更大规模,具有良好的通用性。
表6比较了Fast-ThinkAct与几种高效文本推理替代方案。包括完全移除推理、限制文本长度为6个token、使用RL长度惩罚等方法。这些方法都导致性能下降,而Fast-ThinkAct使用6个潜在token实现最佳性能。
这个表格证明了潜在推理相比文本推理压缩的优越性,是论文核心创新点的重要验证。
表8研究了潜在推理步数M的影响。M=1限制推理能力,M=30或100可能引入冗余或噪声信息,M=6实现最优性能。
这个表格验证了论文选择M=6的合理性,展示了潜在推理步数与性能之间的权衡关系。