ReVisiT:揭示并引用视觉 Token 内隐语义以引导 LVLM 无训练解码
论文信息:Beomsik Cho, Jaehyung Kim (Yonsei University) 发表期刊/会议:ACL 2026 (
arXiv:2506.09522,doi:10.18653/v1/2026.acl-long.262) 代码仓库:bscho333/ReVisiT
1. 核心思想要览
主流训练无关(Training-Free)的多模态幻觉抑制方法(如 VCD、M3ID、SID、CODE、AGLA)大多通过构造扰动视图(加噪、掩码、移除图像)并执行双路或多路解码前向传播(Multi-Pass Forwarding) 来抑制语言先验,导致推理延迟普遍翻倍(
ReVisiT(Referencing Vision Tokens) 从可解释性探针(Logit Lens)视角出发,首次揭示了一个关键事实:即便在模型发生幻觉的解码步,真实物体(Ground-Truth Object)的语义依然编码在图像视觉 Token 的隐状态中;只要将词表投影限制在与当前上下文语义相容的候选子集内,视觉 Token 在语言空间中的物体语义便会清晰浮现。 基于这一发现,ReVisiT 提出了一种极其轻量的单次前向解码算法:
- 预填充一次性视觉 Logits 缓存(Prefill-Time Caching):在首个生成步(Prefill 阶段),将候选解码层集合
中所有视觉 Token 的隐状态 通过语言模型输出头(LM Head)一次性投影到完整词表 ,缓存在显存矩阵 中,后续自回归解码步零额外前向传播、零重复矩阵乘法; - 上下文感知自适应词表子集约束(Context-Aware Vocabulary Subset):在每个解码步
,利用自适应合理性约束(APC)根据原始输出分布筛选出合理的局部候选词表子集 ,滤除全词表下干扰视觉投影的函数词与高频噪声; - 约束子集上的最小 JSD 视觉 Token 选择(Constrained Min-JSD Selection):通过简单的张量切片与掩码,在子集
上计算原始输出分布与缓存的各视觉 Token 投影分布之间的 Jensen-Shannon 散度(JSD),选出与当前生成上下文最契合的视觉参考对 ; - KL 正则化专家乘积对数空间融合(Product-of-Experts Log-Space Fusion):将原始约束分布与所选视觉 Token 约束分布在对数空间直接相加并重归一化,以仅
的微弱延迟开销(相比 VCD/SID 提速近 ),在 CHAIR、POPE、HallusionBench、VQAv2 与 MMMU 五大基准、8 个主流模型(7B 至 32B)上取得领先或极具竞争力的抗幻觉表现。
2. 先导实证分析:视觉语义在幻觉发生时去哪了?
作者在 Qwen2.5-VL-7B(词表规模
该集合共包含 176 个单 Token 词项,覆盖 80 个 COCO 类别中的 62 个类别。在默认 Greedy 解码生成的 500 条图像描述中,CHAIR 共检出 275 处幻觉词,其中幻觉词与至少一个同图真实(GT)物体同义词均为单 Token 的对齐位置共有
2.1 观察一:幻觉发生时,真实物体仍在高概率候选前列
在
| Top- | 占完整词表比例( | 幻觉步真实物体召回率 At-Least-One Recall@ |
|---|---|---|
| Top-50 | 63.2% | |
| Top-100 | 73.7% | |
| Top-500 | 95.8% |
这表明:绝大多数幻觉并非源于模型“完全没看见真实物体”,而是真实物体的排名在局部竞争中略逊于受语言先验驱动的幻觉词。
2.2 观察二:视觉 Token 持续向解码分布注入正向接地增益
记解码步
- 在全部
个解码位置上,Wilcoxon 符号秩检验显示 ( ),说明视觉 Token 在全局范围内显著拉升了真实物体相对于幻觉词的概率优势; - 但在发生幻觉的
个局部步骤上, 且 ( ),即隐式注意力传递在这些步骤上未能扭转 的逆序,亟需在解码端显式调用视觉 Token 语义予以矫正。
2.3 观察三:子集约束能将视觉 Token 的显式语义召回率提升 20 倍
将 500 张图像全部
| 评测范围 | 全词表投影 | 语义相关物体子集投影 |
|---|---|---|
| Top-1 GT Recall | 40.44% | |
| Top-5 GT Recall | 69.56% | |
| Top-10 GT Recall | 79.78% | |
| Top-30 GT Recall | 89.24% |
核心洞见:在完整词表
3. 方法详解:ReVisiT 三步解码机制
设输入序列由
3.1 预填充阶段的视觉 Logits 一次性缓存
在首个生成步(
- 对每一候选层
与每一个视觉 Token 下标 (注意仅遍历视觉 Token,不遍历文本 Token),计算其投影 Logits 向量:
- 将所有候选层与视觉 Token 的投影结果拼装并缓存为矩阵:
由于视觉 Token 位于序列最前端,在因果掩码(Causal Attention Mask)下,其隐状态
3.2 步骤一:上下文感知的自适应词表子集构建
在每个解码步
其中
3.3 步骤二:约束空间内的投影与最小 JSD 视觉 Token 选择
对缓存矩阵
接着,计算每个候选视觉 Token 分布与当前原始约束输出分布之间的 Jensen-Shannon 散度(JSD),并联合选取散度最小(即语义最契合当前生成语境)的视觉 Token 与解码层对
与 DoLa 的本质区别:DoLa 在同一文本位置对比不同层,选取 JSD 最大(
)的早熟层做减法(Contrastive Subtraction)以剥离浅层语法偏置;而 ReVisiT 是在 个视觉 Token 候选中搜寻与当前解码意图语义同质(处于同一候选子集流形内)的视觉锚点,因此必须取 并做正向融合。消融实验证明,若改为 ,CHAIR 上的物体 F1 会从 崩溃至 。
3.4 步骤三:基于 KL 正则化专家乘积(PoE)的分布修正
记选中的最优视觉参考分布为
该优化问题具有显式闭式解(Product-of-Experts, PoE):
取默认等权重设置
最终输出分布在子集
在确定性解码(Greedy,
4. 实验评估与核心结果
4.1 评测基准与架构设置
- 评测模型(涵盖 3 大家族 8 个模型规模):主实验评测
LLaVA-1.5-7B、Qwen2.5-VL-7B、InternVL3-8B;扩展实验涵盖LLaVA-1.5-13B、InternVL2.5-8B/26B、InternVL3-14B、Qwen2.5-VL-32B。 - 候选层池
配置:对 LLaVA-1.5与InternVL3默认仅取最后一层(last,即);对 Qwen2.5-VL默认取所有偶数解码层(all)。 - 对比基线:Greedy、DoLa、VCD、M3ID、CODE、SID(全部统一在
、最大生成长度 512 tokens 下评测)。
4.2 CHAIR 开放式图像描述抗幻觉结果(500 张 COCO Val 图像)
| 方法 | LLaVA-1.5-7B | LLaVA-1.5-7B | LLaVA-1.5-7B | Qwen2.5-VL-7B | Qwen2.5-VL-7B | Qwen2.5-VL-7B | InternVL3-8B | InternVL3-8B | InternVL3-8B |
|---|---|---|---|---|---|---|---|---|---|
| Greedy | 53.8 | 14.66 | 82.33 | 35.2 | 8.43 | 79.85 | 29.4 | 7.91 | 81.11 |
| DoLa | 53.6 | 14.45 | 82.27 | 31.0 | 7.63 | 79.33 | 28.8 | 8.49 | 81.37 |
| VCD | 52.8 | 15.61 | 81.84 | 35.0 | 9.07 | 80.10 | 30.8 | 8.43 | 81.56 |
| M3ID | 57.0 | 16.57 | 82.06 | 33.8 | 9.80 | 78.80 | 29.2 | 8.02 | 81.81 |
| CODE | 45.2 | 13.13 | 81.48 | 17.0 | 9.35 | 48.50 | 29.0 | 8.25 | 81.32 |
| SID | 50.6 | 13.58 | 83.34 | 37.6 | 8.84 | 80.40 | 33.0 | 9.58 | 81.82 |
| ReVisiT (Ours) | 50.6 | 13.43 | 83.17 | 29.8 | 7.04 | 81.16 | 27.6 | 7.27 | 82.29 |
注:CODE 在 Qwen2.5-VL-7B 上虽然
降至 ,但物体 F1 暴跌至 (因过度缩减有效生成内容);而 ReVisiT 在三个骨干上均实现 显著下降与 F1 稳步提升的双赢。在大尺寸模型(LLaVA-1.5-13B、InternVL3-14B、InternVL2.5-26B、Qwen2.5-VL-32B)上,ReVisiT 同样一致降低 并提升 F1。
4.3 POPE、HallusionBench、VQAv2 与 MMMU 综合评测
- POPE(MS-COCO / A-OKVQA / GQA
3 种采样共 9 个子集平均): LLaVA-1.5-7B:Accuracy 从 Greedy 的提升至 81.80%,Precision 从 提升至 78.03%,F1 达 83.45% (全场第一,超越 SID 的 与 VCD 的 ); InternVL3-8B:Accuracy 达 89.21%,F1 达 89.44% (全场最高)。
- HallusionBench(Visual Dependent 子集 591 题,GPT-4 评判):
- 在
LLaVA-1.5-7B上,问题对准确率(qAcc)从 Greedy 的大幅跃升至 20.22%,图像准确率(fAcc)从 升至 26.52%,总准确率(aAcc)从 升至 51.44%;在 Qwen2.5-VL-7B(aAcc 61.76%)与InternVL3-8B(qAcc 33.21%、fAcc 39.13%)上同样全面领先。
- 在
- 通用视觉问答与多学科推理(VQAv2 & MMMU):
- 许多对比解码基线在通用问答上出现明显退化(如 VCD 在 InternVL3-8B 上 VQAv2 从
跌至 ,SID 跌至 ); - ReVisiT 在
LLaVA-1.5-7B(VQAv2 70.40%,MMMU 35.13%)、Qwen2.5-VL-7B(VQAv2 67.60%,MMMU 51.18%)和InternVL3-8B(MMMU 54.14%)上均刷新最高分,证明其不会牺牲通用多模态理解与复杂推理能力。
- 许多对比解码基线在通用问答上出现明显退化(如 VCD 在 InternVL3-8B 上 VQAv2 从
4.4 推理延迟与消融实验(NVIDIA RTX A6000 48GB)
| 方法 | 前向传播次数 | LLaVA-1.5-7B 延迟(ms/token) | 相对开销 | Qwen2.5-VL-7B 延迟(ms/token) | 相对开销 |
|---|---|---|---|---|---|
| Greedy | |||||
| DoLa | |||||
| VCD | |||||
| M3ID | |||||
| SID | |||||
| CODE | |||||
| ReVisiT (Ours) |
关键消融结论(Qwen2.5-VL-7B on CHAIR):
- 去掉词表子集约束(Full Vocabulary + min-JSD):模型迅速陷入单 Token 无限重复退化,物体 F1 从 81.16 暴跌至 1.52,证实上下文子集约束
是防止全词表噪声主导视觉投影的承重墙; - 选择准则对比:
min-JSD(F1)显著优于 min-KL(F1)与 Cosine Similarity(F1),而 Random(F1)与 max-JSD(F1)完全失效。
5. 复现审计与关键实现细节
- 缓存张量
的显存管理: 的维度为 。在 LLaVA-1.5-7B上(层, , ),FP16 仅占 ; - 但在
Qwen2.5-VL-7B上(all偶数层,动态分辨率视觉 Token 可达 ,词表 ),若以 FP32 缓存将占用 显存!工程实现中必须使用 bfloat16/float16存储并在每条样本生成结束后立即释放缓存。
- 伪代码下标修正:论文 Algorithm 1 中写有
Given {h_i^j} for i = 0, ..., T-1,但紧接着的内层循环明确为for i = 0 to |v| - 1。由于视觉 Token 拼接在文本 Token 之前(下标),切片时务必只取视觉 Token 区间,切勿混入文本提示 Token。 - 阈值
的任务特性:开放式长文本描述(如 CHAIR)适合较宽松的候选子集( ),以保留丰富的实词候选;而短答案视觉问答(如 VQAv2)更适合较紧的候选子集( ),避免引入无关干扰项。