← 返回 2026-03-06

面向社交感知机器人的轻量化视觉推理方法 Lightweight Visual Reasoning for Socially-Aware Robots

Alessio Galatolo, Ronald Cumbal, Alexandros Rouchitsas, Katie Winkle, Didem Gürdür Broo, Ginevra Castellano 📅 2026-03-04 👍 0 2026-07-13 08:35
人机交互 多模态推理 意图识别 机器人感知 视觉-语言模型 轻量化模块

用LLM隐藏态回灌视觉编码器,构建视觉-语言双向推理回路。

前置知识

视觉-语言模型(VLM)

VLM 是由视觉编码器(如 CLIP)与大语言模型(LLM)通过跨模态对齐训练拼装而成的模型,能同时理解图像与文本并按指令生成回答。其主流架构采用 Flamingo 范式:图像先被编码器切分为 patch 嵌入,再与文本 token 拼接送入 LLM,LLM 输出最终回复。Qwen 2.5 VL、Gemma 3、LLaVA-OneVision 都属于此类。

本文提出的视觉推理模块就插入在 VLM 的视觉编码器与 LLM 之间,因此必须先理解 VLM 的前向流程才能明白作者在哪个环节插入反馈。

因果掩码(Causal Masking)与隐藏态(Hidden States)

因果掩码让 LLM 在自回归解码时只能看到当前及之前的 token,从而保证生成顺序。隐藏态是 LLM 每一层输出的中间表征向量,蕴含了对已读上下文(包括图像 token)的丰富语义抽象。本文的'视觉推理模块'直接消费的就是最后一层 LLM 在图像 token 位置上的隐藏态。

作者利用因果掩码的方向性解释为何'图像在前、文本在后'的输入顺序会让 LLM 对图像的隐藏态无法被后续文本条件化——这是消融实验中一个反直觉结论的关键。

LoRA(低秩适配)

LoRA 通过在原始权重旁路注入两个低秩矩阵 $A \in \mathbb{R}^{d \times r}$ 与 $B \in \mathbb{R}^{r \times d}$(其中 $r \ll d$)来实现参数高效微调,训练时冻结原权重、只更新低秩矩阵。本文中 LoRA 只在第一趟前向中启用,用于辅助 LLM 生成更好的视觉反馈;第二趟前向时关闭。

论文总可训练参数 <3% 中很大一部分来自 LoRA,理解它才能正确评估方法的额外成本。

门控 MLP(Gated MLP)

门控 MLP 的形式为 $\sigma(W_g x) \odot W_p(\text{Dropout}(W_2 \cdot \text{GELU}(W_1 x)))$,其中 $\sigma$ 是 sigmoid 门控、$\odot$ 是逐元素乘积。它通过一个可学习的门控信号对前馈输出做软筛选,比普通 MLP 表达更细粒度的调制。本文用其把 LLM 隐藏态投影回视觉编码器输入空间。

这是视觉推理模块的核心组件,去掉它会导致 Mementos 与导航性能明显下滑——理解其结构才能看懂消融实验。

Patch Unmerger

LLM 输出的图像 token 数量通常与视觉编码器输出的 patch 数量不同(因 cross-attention 或 perceiver resampler 压缩过)。Patch Unmerger 把 LLM 隐藏态'展开'回视觉编码器所期望的 patch 网格形状,从而保证反馈信号能直接加到原始图像嵌入上。

它是视觉反馈链路能闭环的工程前提;消融实验中即使移除 MLP,Unmerger 仍能部分补偿,说明它是必要但不充分的设计。

Furhat 社交机器人与意图识别数据集

Furhat 是瑞典 Furhat Robotics 公司的拟人化社交机器人,可投影面部表情并对话。本文作者在 10 名参与者(3 女 6 男 1 未知,平均年龄 30.9 岁)的视频-音频记录中标注了 5 类交互行为(等待、接近打断、平静示意、急切示意、同时对话),共 188 个事件,最终扩成 376 个多选题样本。

理解这一自建数据集的构造方式与视角(机器人第一人称)有助于判断其外部效度与对真实部署的可迁移性。

研究动机

现有部署于人类共享空间的机器人不仅要导航和避障,还必须理解动态、常常不可预测的人类行为,例如在人群中识别他人意图、礼让插队者、回应求助等。尽管 CLIP、InstructBLIP、RoboVQA、MMRo 等预训练 VLM 在物体识别、指令跟随上取得长足进步,但它们普遍把视觉当作文本推理的'被动上下文'——图像只被编码一次就再也不更新,缺乏对人类细微线索(目光朝向、肢体张力、接近节奏)的深度整合。作者指出即便 SOTA 模型在 MMRo 这类基础感知任务上仍频繁出错属性、颜色、空间位置,而 HRI 场景里'意图'这种高级语义更要求语言与视觉的深度耦合,而不是浅层拼接。

本文的目标是论文目标是为现成的 VLM 设计一种轻量化插件,使其能在不大幅改动原模型的前提下,把 LLM 对当前上下文的语义理解反向注入到视觉编码器,从而让机器人在导航、场景描述、人类意图识别三个机器人中心任务上都获得稳定增益。具体而言:额外参数 <3%,推理可在单卡 RTX 3090 上完成,并在 Qwen 2.5 VL 7B、Gemma 3 4B、LLaVA-OneVision 1.5 4B 三类主干上保持通用性。

与已有工作不同的是,作者切入了一个被忽视的架构不对称问题:当代 VLM 多是严格的视觉→语言前馈(perception→interpretation),没有让语言去'回头看'视觉。这一缺口源于:(1) 现有视觉推理方法多依赖 Chain-of-Thought 等文本侧推理,把视觉信息当成提示而非参与推理的'一等公民';(2) 用边界框、分割掩膜等外部工具干预视觉的做法会切断端到端梯度流,使 VLM 自身失去在推理时自由操控视觉的能力;(3) 反射式(reasoning/reflection)思路在纯文本 LLM 中已证明有效(test-time compute scaling),但在多模态场景下尚未被架构化地实现。本文正是用一条紧凑的'语言→视觉'反馈回路填补这条缺口。

核心方法

方法的整体直觉是:让 LLM 先'看一遍'图像,把对图像的语义理解压缩成一段视觉提示(visual hint),再用一段小型 MLP 把这段提示'翻译'回视觉编码器能消化的空间,加回到原始图像嵌入上,从而让图像在第二次编码时已被任务相关的语义所调制。技术路线为:在每个训练步执行两次前向,第一趟冻结视觉编码器、启用 LoRA 跑 LLM,从图像 token 的最后隐藏态中抽取 hint $z$;第二趟把 $r(z)$ 与原始图像嵌入相加再走一次视觉编码器,然后再次跑 LLM(关 LoRA),最终损失只从第二趟回传,反向更新视觉推理模块与 LoRA。推理时执行完全相同的两次前向,但 LoRA 已合并入 LLM,因此是无额外训练参数的纯推理管线。整个模块可加在任何 Flamingo 式 VLM 上。

核心创新点是把'语言→视觉'的反馈回路从无到有地结构化进 VLM 里:用一个仅占原模型 <3% 参数的 gated MLP + patch unmerger,把 LLM 对图像 token 的隐藏态再注入视觉编码器,使图像嵌入能够基于语言上下文被'重读'。与已有方法的本质区别有三:(1) 不同于 DDCoT、Visual-CoT 等依赖外部 VQA 工具或纯文本 CoT 的方案,本方法在 VLM 内部建立端到端可微的反馈回路,不切断梯度;(2) 不同于 Image-of-Thought、Cocot 等把视觉作为生成对象的方法,本方法让 LLM 去'操控'视觉编码器的输入而非生成新图像;(3) 不同于依赖边界框或分割等外部干预的做法,本方法不需要任何预训练检测器,使模型在推理时具备文本式 LLM 那种 test-time 可扩展性。

方法步骤详情

完整训练流程如论文 Algorithm 1 所示,可拆为六步:(1) 给 LLM 注入 LoRA 适配器(可选),视觉推理模块 $r$ 与 patch unmerger 作为唯一新增可训练参数;视觉编码器与 LLM 主干全部冻结。(2) 第一趟前向:图像 $x_v$ 经视觉编码器得到 $H_v = f_{VE}(x_v)$;连同用户查询 $x_q$ 一起送入 LLM,开启 LoRA,得到所有 token 的隐藏态 $H = f_{LM}(x_q, H_v)$。(3) 从 $H$ 中切出图像 token 在最后一层的隐藏态作为视觉提示 $z \leftarrow H_{\text{last}}^{\text{img}}$。(4) 视觉推理模块计算 $r(z)$:先经 patch unmerger 把 $z$ 投影回与图像 patch 数量一致的空间,再经过 gated MLP $\sigma(W_g z) \odot W_p(\text{Dropout}(W_2 \cdot \text{GELU}(W_1 z)))$,输入输出维度等于 LLM 隐藏维度 $d$,隐藏维度为 $2d$。(5) 第二趟前向:把 $r(z)$ 与原始 $H_v$ 相加得到 $\bar{H}_v = f_{VE}(x_v + r(z))$,连同 $H_v$ 一起送入 LLM(关闭 LoRA),得到预测 $\hat{y} = f_{LM}(x_q, H_v, \bar{H}_v)$。(6) 计算 $\mathcal{L}(\hat{y}, \text{labels})$ 并仅沿 LLM→视觉编码器→视觉推理模块路径反向传播,更新 $r$ 与 LoRA。推理时执行完全相同的两趟前向(但 LoRA 已合并),最终只读取 $\hat{y}$ 输出。

技术新颖性

技术新颖性体现在三方面:第一,架构上首次把'LLM→视觉编码器'显式闭合为一条端到端可微的反馈回路,而不依赖任何外部工具或提示工程,使视觉嵌入在每次推理中都能被语言上下文'重写';第二,参数效率极高:可训练参数不到 1.7%,即便加上 LoRA 也不到 3%,并因此在推理时仅带来 <3% 显存增加(15.9 GB→16.32 GB),TFLOPs 增至三倍但吞吐量仍可达 1.27 samples/s,可部署在单卡 RTX 3090;第三,训练策略上把 hint 抽取与 hint 注入分到两次前向完成,并辅以 Linear interpolation 式的两配置融合(基于 MME-CoT 验证集挑选 top-2 后合并),让小参数也能稳定收敛。整篇文章的'挑战 VLM 主流前馈范式'立意也颇具新意:他们用实验结果证明冻结的视觉编码器依然能被语言调制而受益,为'架构不对称+反馈'的多模态设计打开了新方向。

Overview of the Visual Reasoning approach: A module connects an LLM's hidden states for image tokens back to the vision encoder through a gated MLP, creating a reasoning loop between text and vision.
Fig. 1: Overview of the Visual Reasoning approach: A module connects an LLM's hidden states for image tokens back to the vision encoder through a gated MLP, creating a reasoning loop between text and vision.
Examples of the datasets used for evaluation: Mementos-Robotics (scene description), Navigation benchmark (robot trajectories), and the custom human-intention recognition dataset.
Fig. 2: Examples of the datasets used for evaluation: Mementos-Robotics (scene description), Navigation benchmark (robot trajectories), and the custom human-intention recognition dataset.

实验结果

Table I 报告了三种主干在三个任务上的全面对比:在 Qwen 2.5 VL 7B 上方法全面获胜,导航距离从 7.787 缩到 7.530(↓3.3%)、Mementos 描述分从 2.261 升到 2.318(+0.057)、意图识别准确率从 34.04% 升到 36.97%(+2.93%);Gemma 3 4B 与 LLaVA-OneVision 1.5 4B 上则呈现'场景描述与意图识别强增益、导航弱退化'的非一致模式:Gemma 在 Mementos 上 +0.111(1.693→1.804)、意图识别 +10.81%(20.84%→31.65%),但导航距离反而由 7.977 微涨到 8.014;LLaVA 在 Mementos +0.055(2.201→2.256)、意图识别 +4.79%(20.74%→25.53%),导航 7.832→8.114。Table II 的消融(基于 Qwen)揭示三件事:(1) 去掉原始图像只用第二张'幻觉图像'会让 Mementos 从 2.318 跌到 1.950、意图识别 36.97%→34.31%,说明视觉'重读'优于视觉'替换';(2) 移除 MLP(仅保留 unmerger)同样显著掉点(Mementos 1.980、导航 7.831),证明门控调制本身不可替代;(3) 输入顺序'图像在前'比'文本在前'在所有指标上都更好(意图 28.46% vs 25.53%,导航 7.685 vs 8.056),与作者基于因果掩码的初始假设相反,作者归因为这些 VLM 的预训练阶段就偏好这种顺序,交换顺序会导致基线掉点高达 30%。Table III 给出资源开销:在 RTX 3090 上平均 TFLOPs 从 7.06 升至 20.39、吞吐从 4.24 samples/s 降到 1.27 samples/s、显存仅从 15.9 GB 增至 16.32 GB,与'轻量化'的标题一致。

任务指标本文基线提升
机器人导航(Habitat 模拟器) 到目标的最终距离(↓ 越小越好) Qwen 7B: 7.530;Gemma 4B: 8.014;LLaVA 4B: 8.114 Qwen 7B: 7.787;Gemma 4B: 7.977;LLaVA 4B: 7.832 Qwen 降低 3.3%(唯一正收益);Gemma 与 LLaVA 分别退化 0.46% 与 3.60%,差异不显著甚至负向
场景描述(Mementos-Robotics) LLM-as-judge 1–5 分(↑ 越高越好) Qwen 7B: 2.318;Gemma 4B: 1.804;LLaVA 4B: 2.256 Qwen 7B: 2.261;Gemma 4B: 1.693;LLaVA 4B: 2.201 三主干全部正增益:Qwen +0.057、Gemma +0.111(最大相对增益 +6.6%)、LLaVA +0.055
人类意图识别(自建 HRI 数据集 376 题) 多选准确率(↑ 越高越好) Qwen 7B: 36.97%;Gemma 4B: 31.65%;LLaVA 4B: 25.53% Qwen 7B: 34.04%;Gemma 4B: 20.84%;LLaVA 4B: 20.74% 全主干正增益:Qwen +2.93 pp、Gemma +10.81 pp(相对 +51.9%)、LLaVA +4.79 pp(相对 +23.1%),是方法最强的赛道
消融:无原始图像(仅 LLM 生成的第二张图) 导航 / Mementos / 意图(Qwen 7B) 7.764 / 1.950 / 34.31% 完整方法 7.530 / 2.318 / 36.97% 去掉原图后 Mementos 大幅退化(-0.368),意图 -2.66 pp,验证'视觉重读 > 视觉替换'
消融:移除 MLP(保留 unmerger) 导航 / Mementos / 意图(Qwen 7B) 7.831 / 1.980 / 37.50% 完整方法 7.530 / 2.318 / 36.97% 导航 +0.301、Mementos -0.338 退化明显;意图反常地 +0.53 pp,作者归因于 unmerger 学到的隐式补偿
消融:输入顺序(image-first vs prompt-first) 导航 / Mementos / 意图(Qwen 7B) image-first: 7.685 / 2.000 / 28.46%;prompt-first: 8.056 / 1.744 / 25.53% 完整方法 7.530 / 2.318 / 36.97% image-first 在所有指标上更优,反驳了'利用因果掩码让文本先看图像可获得更好隐藏态'的假设
资源消耗(Qwen 7B,RTX 3090) TFLOPs / samples per second / 显存 GB 20.39 / 1.27 / 16.32 7.06 / 4.24 / 15.9 TFLOPs ×2.89,吞吐 ÷3.34,显存仅 +2.6%;可单卡消费级部署

局限与改进

作者明确承认的局限有三:(1) Gemma 与 LLaVA 的导航任务未获收益,反而因模型自身的 JSON 输出格式问题导致动作解析失败,作者把这归咎于主干而非方法本身;(2) LLaVA 在导航中频繁出现'refusal'行为(拒绝给出动作),只在场景描述与意图识别上观察到增益;(3) 两趟前向使 TFLOPs 接近三倍、吞吐量降至 1.27 samples/s,对实时高频感知仍是负担,需要进一步量化或硬件加速。我自己的额外观察:(a) 人类意图识别数据集仅 376 题且未做 train/val/test 划分,全部用于评估,意味着无法判断是否过拟合该特定场景;(b) 训练数据是通用 Visual-CoT 而非机器人专用,作者解释为避免污染评估,但代价是导航这类需要空间常识的任务只获得 Qwen 上的微弱增益;(c) Patch Unmerger 在去掉 MLP 后仍能撑住意图识别性能,说明其本身已是一个非平凡的可学习组件,论文中并未充分分析其独立贡献;(d) 评测场景局限在英文指令 + Furhat 机器人,多语言、跨文化、跨机器人形态的可迁移性尚未验证。

独立分析的弱点

独立审视后我看到四个值得改进的弱点:(1) **双前向开销过高**:TFLOPs ×2.89 让推理吞吐降到 1.27 samples/s,对真实机器人 30 Hz 视觉流严重不友好。改进方向是缓存第一趟 LLM 中间层的图像隐藏态,或用更便宜的 early-exit/蒸馏式单趟版本(knowledge distillation 让单趟 LLM 模仿两趟行为)。(2) **依赖主干输出结构化 JSON 的能力**:导航任务在 Gemma/LLaVA 上掉点主要因为格式错误而非视觉推理失败。可以加一个轻量级的 action parser 或约束解码(grammar-constrained decoding)来兜底,或在 Visual-CoT 训练时混入导航指令数据。(3) **Patch Unmerger 行为不透明**:它被消融为'非必要但能补偿'的隐式模块,应单独做可视化或注意力分析,明确它到底在'展开'什么语义;也可以尝试更显式的 perceiver resampler 替代方案以提高可解释性。(4) **自建 HRI 数据集规模小且无划分**:376 题 + 188 事件 + 10 名参与者,无法支撑统计显著性检验,也无法做少样本泛化研究;建议至少扩展到 50 人以上并按说话人划分训练/测试,再加入'非英语+非 Furhat'作为跨域评测。

未来方向

作者在结论里点出两条延伸:(1) 用更具针对性的机器人数据集替代 Visual-CoT 来训练视觉推理模块(甚至可以基于 Habitat 仿真自动生成合成数据);(2) 把模型量化或部署到专用加速器以抵消双前向开销。基于成果还可拓展:(a) 把'视觉重读'的思路泛化到视频、音频、触觉等多模态场景,构建一个多模态可循环推理的通用骨干;(b) 引入 test-time compute scaling,让 LLM 在推理时迭代多轮生成 hint $z$,把每次重读当作文本 CoT 的视觉对偶;(c) 把 hint $z$ 显式可视化(如 attention rollout 或 logit lens)以提高模型可解释性,让社交机器人能在拒绝响应时给出原因;(d) 与 RLHF/DPO 结合,让视觉反馈回路接受人类对'是否真正理解意图'的偏好对齐;(e) 在真实机器人(Pepper、QTrobot、Hiwonder)而非仅 Furhat 上做实机验证。

复现评估

复现友好度较高:作者在论文 III.A 节明确给出训练/评估代码仓库 https://github.com/alessioGalatolo/VLM-Reasoning-for-Robotics;训练使用 Visual-CoT 公开数据集 + 自建 HRI 数据集(按声明需进一步发布);硬件门槛为 4×A100 40GB,bf16 半精度,单次训练 1h30–2h45;推理可在单卡 RTX 3090 上完成,峰值显存 16.32 GB。算法细节在 Algorithm 1 完整列出,包含两趟前向、LoRA 注入位置、unmerger 与 gated MLP 结构、学习率扫描范围 $\eta \in [10^{-2}, 10^{-5}]$ 等 7 个值、MME-CoT 验证集选 top-2 后线性插值融合等关键超参。复现难点主要在三处:(1) 三种 VLM 主干各自的训练数据格式与对话模板差异较大,需要分别适配;(2) patch unmerger 的输出维度与各 VLM 的视觉 patch 数耦合(CLIP vs SigLIP),容易错位;(3) 自建 HRI 数据集涉及 10 名参与者视频与伦理声明,需要走完 IRB 流程。总体而言属于'代码可跑、但需较高工程投入'的复现难度。