多模态Jev专题:FastJEV剪掉45.83%候选层,0.8B平均分还留97.52%

(1)《三年面试五年模拟》AIGC / LLM / AI Agent 算法工程师与开发工程师求职面试秘籍,独家资源见 WeThinkIn/AIGC-Interview-Book,欢迎 Star!

(2)AIGC / LLM / AI Agent 算法岗与开发岗求职面试内推学习社群 ,涵盖 AIGC、LLM 大模型、AI Agent、传统深度学习、自动驾驶、机器学习、计算机视觉、自然语言处理、强化学习、大数据挖掘、具身智能、元宇宙、AGI 等方向的最新面试干货与核心知识,欢迎加入(https://t.zsxq.com/DZRng)!


论文标题:FastJEV: Understanding Redundancy for Compact JEV Inference

论文作者:厦门大学,Jie Ma、Xiaoshuai Sun等

发表时间:2026年10月8日


目录

本文要点

(1)FastJEV优化的是OmniJev v1.1这种多模态决策模型的候选打分。图像和公共上下文仍用完整骨干编码,候选路径上做三件事,锚定、共享前缀、剪层。

(2)不额外训练,候选和决策头都保留。0.8B剪掉45.83%的候选层,六项平均保留97.52%的原分数。2B同预算保留94.92%。4B剪掉43.75%保留93.66%。

(3)默认只跳六层时,0.8B、2B、4B的平均保留率是100.97%、98.98%、100.21%。4B的AI2D从84.33%到84.29%,BLINK停在64.33%,ERQA从44.75%到46.50%。

(4)分数留下来,墙钟时间不一定降。选出的大预算上,三个尺寸相对原版的几何平均加速比是0.938、0.939、0.953,小于1。

(5)文末按论文第4.1节列T、S、P三个开关和校准集做法。仓库地址在论文里,正文没有安装命令。


2026年10月8日,厦门大学的Jie Ma、Jie Gao、Yihang Liu、Zhike Qiu、Junle Li、Chongyi Zhuang、Jiayi Ji、Xiaoshuai Sun把FastJEV提交到arXiv,编号2610.11379。PDF的Date行写October 9, 2026。我读的是v1。通讯邮箱是jiema100@stu.xmu.edu.cn和xssun@xmu.edu.cn。

OmniJev是他们用来打分的多模态决策模型,v1.1,尺寸0.8B、2B、4B。图像和问题编码一次之后,每个候选还要走语言层。候选之间会重复同一段token历史,也会各存一份推理状态,并且默认跑完所有层。FastJEV盯的是这三处,不改权重。

一、上下文锚定和候选前缀树

发布版推理已经把公共的视觉和文本前缀算一次。FastJEV在这之上再动状态和深度。

共享上下文锚定把前缀在每一层产生的循环状态当成只读缓冲。每个候选分支读这份初始状态,再维护自己的工作状态。视觉、token历史、位置和mask一致时,这种共享在精确算术下和原来的递推相同。卷积状态和注意力的KV缓存仍然按分支分开存。一条候选路径结束、后面没有人依赖它时,最终的循环缓存不再分配和写回。中间还要分叉的缓存留着。

候选前缀共享把后缀收成一棵前缀树。两个token能并到同一个节点,条件是父节点、token本身、三个原始位置id都相同。位置id沿用合并前的输入,不在合并后再重编。树按路径段执行,段在分叉或候选边界结束。同一父节点的段可以组batch,继承父节点的缓存当初始锚。有后代的段留下最终缓存,叶子段可以省掉最终循环缓存。

决策头仍要每个候选一条记录。作者保留token到树节点的映射,按原路径把对数概率捡回来,以原来的候选顺序送进没改过的决策头。头输出各候选logit和一个弃权logit。共享了整条树路径的候选,在读出里仍然是分开的条目。

二、用无标签样本决定跳哪些层

前缀共享不减少路径深度。剪层只发生在候选分支,公共前缀仍跑全部层。

对校准集里的每个输入,比较完整logit和只跳过第ℓ层之后的logit。两者都包含弃权logit,并在概率后处理之前测量。先去掉全体logit的共同平移,因为softmax不看这个平移。敏感度是相对分数变化的均方根,再除以原logit的尺度,分母有一个1e-8的下限。一层的分数是校准集上的平均敏感度。预算为k时,取敏感度最小的k层,平局按层号。

这套排序每个模型算一次,之后所有输入共用,不用标签,也不更新权重。校准集是32条OK-VQA,每条64个来自训练答案词表的固定候选。评价集的标签不参与选层。

图2里,三个尺寸都是靠后的一些层敏感度低,最后一块反而最高。敏感度不是越深越低,所以不能按"跳掉最后几层"来剪。

跳过的层换成恒等映射,注意力或循环模块以及前馈网络都不跑。

三、六个评测集上的保留率

公开集是AI2D测试3,088题,ScienceQA图像测试2,017题,POPE 9,000题,BLINK验证1,901题,ERQA 400题。另有从LIBERO-10演示重建的离线问答,20条评测episode、十个任务、188帧,每帧八种题,共1,504题。标签来自仿真器状态和固定轨迹规则。原题库不可得,所以LIBERO只代表这套重建题,不代表官方题面。

多选看给定选项里的最高分。二值题用原来的sigmoid头,阈值0.5。LIBERO的argmax包含弃权,弃权算错。平均保留率是六项准确率之比的不加权均值,比值在四舍五入前计算,不截断到100%。

计时在一张NVIDIA L20上,BF16骨干,SDPA。公开集用64道固定题测三轮时间、16道题测显存。LIBERO用20帧测三轮时间、10帧测显存。延迟含读图、预处理、上下文编码、候选执行和打分,不含加载模型、预热和序列化结果。

T是前缀共享,S是上下文锚定,P是剪层。完整方法是T+S+P。下表只列原版,以及平均保留率至少90%的最大预算。默认六层那一档另列,因为它是论文的默认实验。

模型 方法 剪掉层数 剪枝比例 AI2D SQA POPE BLINK ERQA LIBERO 平均保留
0.8B,24层 原版 0 0 62.27 69.01 84.76 43.98 28.75 55.45 100
0.8B FastJEV 6 25.00 62.37 69.31 87.79 43.40 29.50 55.65 100.97
0.8B FastJEV 11 45.83 58.19 66.63 88.50 41.45 28.00 54.92 97.52
2B,24层 原版 0 0 75.00 83.79 88.83 52.24 36.75 58.24 100
2B FastJEV 6 25.00 74.90 83.84 88.80 51.13 37.00 55.59 98.98
2B FastJEV 11 45.83 70.24 77.84 89.33 49.40 35.00 54.12 94.92
4B,32层 原版 0 0 84.33 94.30 90.77 64.33 44.75 65.23 100
4B FastJEV 6 18.75 84.29 94.10 90.64 64.33 46.50 63.76 100.21
4B FastJEV 14 43.75 81.96 92.56 90.46 64.70 40.00 50.20 93.66

再往上剪,0.8B剪14层、58.33%,平均保留掉到76.17%。2B同档73.81%。4B剪18层、56.25%,保留78.76%。论文把"至少90%"当成表里加粗的那一档边界。

默认六层时,18个对比里有13个绝对准确率变化不到1个百分点。2B和4B的LIBERO分别掉2.66和1.46个百分点。4B的ERQA从44.75升到46.50,这是少数变好的格子,不能当成剪层会涨点。

四、少算了,墙钟时间不一定短

默认六层预算下,完整FastJEV在18个对比里有6个总峰值显存更低。POPE上三个尺寸都比原版快,另外五个数据集更慢。LIBERO峰值显存分别降15.70%、8.43%、9.12%,每个尺寸的延迟都更高。

表1里那些保留率至少90%的较大预算,六集合上的几何平均加速比是0.938、0.939、0.953。小于1就是总体更慢。

4B的因子实验把T、S、P拆开,用同一个循环核,P启用时仍跳六层。S在T和P固定时,AI2D峰值显存降104.8MiB,LIBERO降224.6MiB,准确率不变。P在T和S固定时两边延迟都降,LIBERO准确率从64.89%到63.76%。T会改变分段方式和词表投影的调度。论文的实现结论写在摘要里。候选前缀共享可以减少重复计算,同时让延迟变高。共享粒度和执行顺序要一起设计,不能只看逻辑上少了多少token。

五、本地部署与推理实践

论文给出的代码地址是github.com/JieMaMagic/FastJEV。正文没有安装命令和命令行参数,下面按第4.1节整理开关,避免把没写过的flag当成官方接口。

text 复制代码
models: OmniJev v1.1, 0.8B / 2B / 4B
T: candidate prefix sharing
S: shared context anchoring
P: decision guided pruning
full method: T+S+P
calibration: 32 OK-VQA inputs, 64 fixed candidates, no answer labels
default P: skip 6 candidate blocks
  0.8B and 2B: 25% of 24 layers
  4B: 18.75% of 32 layers
prefix: all layers, all weights stay loaded
binary threshold: 0.5
LIBERO: abstention counts as an error
device in the paper: one NVIDIA L20, BF16, SDPA

剪层掩码跟模型走,不跟任务走。换一批图之前不用重新标定,除非要换预算k。公共前缀不能跟着剪,否则共享的上下文缓存和原版对不上。

六、总结与思考

FastJEV说明多模态JEV的候选路径里有可剪的层。最后一层最敏感,中间偏后的一些层可以跳,0.8B在剪掉将近一半候选层时还留着97.52%的平均分。4B在43.75%的预算上留93.66%,LIBERO从65.23掉到50.20,这项比平均保留率难看。

它没有变成一次更快的服务。几何平均加速比小于1,前缀树在这份实现里还可能加延迟。若目标是降延迟,论文自己的POPE结果和其余五集是反的,不能只报准确率保留。若目标是少占状态,S那一百到两百MiB更直接。我会先看显存和LIBERO,再决定要不要上T。


参考链接

  • 论文原文 arXiv 2610.11379 v1,2026年10月
  • 代码地址 JieMaMagic/FastJEV,以论文给出的链接为准
  • 被测模型OmniJev v1.1,论文引为Xu et al., 2026
  • 重建问答所用环境LIBERO-10,论文第4.1节
相关推荐
鲜于言悠9051 小时前
RAG工程化实战:从玩具Demo进化成企业可用知识库
人工智能
吴佳浩1 小时前
“优化 KV Cache”到底在优化什么?vLLM、SGLang、Strata 优化的是三个不同的维度
人工智能
a努力。1 小时前
MCP只连工具,不直接连技能
人工智能
neoooo1 小时前
🚀Claude Code 小技巧(一):一键进入危险模式
前端·人工智能·后端
阿里云大数据AI技术2 小时前
PAI 模型蒸馏全新发布「场景化蒸馏」:把最佳实践装进模板,让大模型能力"轻装上阵"
人工智能
亚川楼宇自控系统数据中心厂家2 小时前
机房 3D 可视化技术:从平面监控到立体化管理的升级
大数据·数据库·人工智能
LaughingZhu2 小时前
任务在飞书创建,开发在 Orca 继续:我把飞书项目接到了 Linear
大数据·人工智能·深度学习·神经网络·飞书
一水鉴天2 小时前
UML/MOF 元层级与 Model & Notation:整体设计核心术语的定稿探讨 20261012(豆包“查看聊天内容”继续)
人工智能·机器学习
陈嘿萌2 小时前
TIP 2026|没有分割标签,也能做任务导向图像融合:SSVIF的自监督语义注入
图像处理·人工智能·深度学习·计算机视觉·图像融合·学术研究与理论基础