← 返回 2026-01-15

Fast-ThinkAct:通过可语言化的潜在规划实现高效的视觉-语言-动作推理 Fast-ThinkAct: Efficient Vision-Language-Action Reasoning via Verbalizable Latent Planning

Chi-Pin Huang, Yunze Man, Zhiding Yu, Min-Hung Chen, Jan Kautz, Yu-Chiang Frank Wang, Fu-En Yang 📅 2026-01-14 👍 56 2026-07-13 08:35
具身AI 实时控制 强化学习蒸馏 推理效率 机器人操作 潜在推理 视觉-语言-动作模型

通过压缩链式推理到紧凑的连续潜在表示,实现视觉-语言-动作任务的高效推理

前置知识

视觉-语言-动作模型(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概览
Figure 1: Fast-ThinkAct概览
Fast-ThinkAct方法细节
Figure 2: Fast-ThinkAct方法细节
长程任务可视化
Figure 4: 长程任务可视化
语言化潜在推理比较
Figure 7: 语言化潜在推理比较
OpenEQA推理轨迹比较
Figure 10: OpenEQA推理轨迹比较

实验结果

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倍加速。

具身推理基准定量评估
Table 2: 具身推理基准定量评估
训练目标和阶段消融研究
Table 3: 训练目标和阶段消融研究
LIBERO和SimplerEnv基准的详细结果
Table 5: LIBERO和SimplerEnv基准的详细结果
机器人操作基准的附加消融研究
Table 7: 机器人操作基准的附加消融研究
机器人操作和推理效率评估
Figure 3: 机器人操作和推理效率评估
少样本适应结果
Figure 6: 少样本适应结果
RoboTwin2.0轨迹可视化
Figure 8: RoboTwin2.0轨迹可视化
失败识别和分析能力
Figure 9: 失败识别和分析能力
查看结构化数据
任务指标本文基线提升
机器人操作 - 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等),但某些数据处理细节(如视觉轨迹标注)可能需要额外工作。代码和模型权重是否开源尚不明确,如果开源将极大促进复现。总体而言,方法原理清晰,实验设置详细,但计算资源需求较高,复现难度中等。