OmniSIFT:面向高效全能模态大语言模型的模态非对称Token压缩 OmniSIFT: Modality-Asymmetric Token Compression for Efficient Omni-modal Large Language Models
提出模态非对称token压缩框架,先修剪视频冗余再视觉引导音频选择,显著提升Omni-LLMs效率
前置知识
Omni-modal LLMs(全能模态大语言模型)
能够同时处理文本、视觉和音频等多种模态信息的大语言模型,如Qwen2.5-Omni。这些模型通过模态特定编码器将不同模态映射到统一的token空间,然后由LLM骨干网络进行联合推理。典型架构包括视觉编码器、音频编码器、跨模态投影层和生成式LLM骨干。
本文的核心研究对象就是Omni-LLMs的token压缩问题,理解其基本架构是理解压缩方法的前提
Token压缩
在多模态大模型中,将输入的视觉、音频等模态信息转换为token序列后,通过去除冗余token来减少序列长度的技术。常见方法包括基于注意力分数的重要性评估、相似度计算、随机剪枝等,目标是在保持模型性能的同时降低计算开销。
本文提出的就是一种新的token压缩方法,需要理解这个概念才能明白研究问题的本质
Cross-attention(跨注意力)
一种注意力机制,其中查询(Query)来自一个序列,而键(Key)和值(Value)来自另一个序列。允许模型学习两个不同序列之间的对齐关系和依赖关系,常用于多模态融合、机器翻译等任务。计算公式为 $H = \text{Softmax}\left(\frac{QK^\top}{\sqrt{d}}\right)V$。
OmniSIFT的VGAS模块使用跨注意力来实现视觉引导的音频token选择
Straight-through Estimator(STE,直通估计器)
一种在离散选择操作(如TopK选择)中实现梯度反向传播的技术。前向传播时执行离散选择(生成二进制掩码),反向传播时使用恒等映射作为代理梯度,使得梯度能够直接流向重要性分数,从而实现端到端训练。
OmniSIFT使用STE来解决TopK选择的不可微问题,是实现可训练压缩的关键技术
模态非对称性
指音频和视频两种模态在信息冗余和重要性判断上存在本质差异的现象。视频冗余可以通过帧内结构和帧间重叠来估计,而音频的重要性更依赖于上下文(如是否与可见说话者或视觉事件相关)。这种非对称性意味着有效的压缩应该以视觉为主导来指导音频选择。
这是OmniSIFT的核心设计哲学,理解模态非对称性是理解整个方法的关键
研究动机
Omni-modal LLMs在音频-视频理解任务中表现出强大能力,但依赖长多模态token序列导致计算开销巨大。一个典型的20秒多模态片段会产生超过20K个token,这种长序列显著增加了计算成本,特别是在长视频理解场景中。现有token压缩方法存在明显局限:模态解耦压缩方法(如DyCoke)独立处理音频和视频,完全忽略了跨模态语义依赖;模态对称压缩方法(如OmniZip)将两种模态视为同等重要,依赖注意力分数进行压缩,但这种方式限制了与FlashAttention等高效算子的兼容性,且将压缩过程简化为选择显著时间位置而非捕获模态特定语义线索。此外,OmniZip的音频引导视频剪枝策略在高压缩比下会导致关键视觉特征被错误丢弃,如比分牌变化等重要视觉信息可能因为音频信号的低重要性评分而被移除。
本文的目标是本文旨在为Omni-LLMs设计一个高效、轻量且兼容现有高效算子的token压缩框架。具体目标包括:(1)仅增加极少参数(如4.85M)实现有效压缩;(2)在仅保留25%原始token的情况下保持甚至超越全token模型的性能;(3)实现比训练无关基线(如OmniZip)更低的推理延迟和内存占用;(4)压缩框架需兼容FlashAttention等高效注意力算子,支持实际部署。
与已有工作不同的是,本文的独特切入角度在于发现并利用音频-视频数据的模态非对称性。人类在处理音视频内容时本身就存在感知非对称性——在联合音视频输入中,视觉信息通常占主导地位,音频信号的显著性往往依赖于是否与视觉场景提供语义锚点(如可见说话者或视觉接地事件)。基于这一洞察,本文提出三个设计原则:(1)模态非对称、视觉引导的压缩;(2)轻量级压缩;(3)兼容高效算子。与OmniZip的音频引导视频压缩相反,OmniSIFT采用视频引导音频压缩的范式,先去除视频冗余获得紧凑视觉锚点,再用这些锚点选择信息丰富的音频token。
核心方法
OmniSIFT的核心思想可以用一个类比来理解:在观看视频时,我们首先会注意到画面中的关键视觉元素(如移动的物体、变化的场景),然后根据这些视觉线索来判断哪些声音是有意义的(如与说话者相关的声音、与可见事件相关的音效)。OmniSIFT正是模仿这种人类感知模式:先对视频进行时空剪枝去除冗余视觉信息,再用保留的视觉锚点来引导音频token的选择。技术路线分为两个阶段:第一阶段是时空视频剪枝(STVP)模块,在chunk级别操作,通过计算帧内空间显著性和帧间时间显著性来选择最具信息量的视觉token;第二阶段是视觉引导音频选择器(VGAS)模块,利用剪枝后的视觉token作为键和值,音频token作为查询,通过跨注意力机制生成视觉上下文化的音频表示,然后通过MLP和sigmoid计算音频token的重要性分数并进行TopK选择。整个框架通过直通估计器(STE)实现端到端训练。
OmniSIFT与已有方法最本质的区别在于其模态非对称压缩范式。现有方法要么独立压缩各模态(模态解耦),要么将两种模态视为同等重要进行对称压缩。OmniSIFT则基于音频-视频数据的内在非对称依赖关系:视频冗余可以从视觉流的内部结构(帧内结构冗余和帧间重叠)估计,而音频信号的显著性更依赖于上下文(是否与视觉场景相关)。因此,OmniSIFT先通过STVP去除视频中的时空冗余获得紧凑视觉锚点,再用这些视觉锚点通过VGAS来选择最相关的音频token。这种视觉引导音频的策略比音频引导视频(如OmniZip)更有效,因为视觉信息通常更稳定且能提供场景语义锚点。另一个关键创新是VGAS的轻量级设计:仅使用单层跨注意力机制(8头、512维隐藏层),总参数量仅4.85M,却能有效捕获跨模态依赖。
方法步骤详情
OmniSIFT的完整处理流程如下:首先,输入的视频和音频分别通过模态特定编码器和投影层转换为token序列 $Z_v \in \mathbb{R}^{N_v \times D}$ 和 $Z_a \in \mathbb{R}^{N_a \times D}$。这些token按2秒粒度对齐形成多模态chunk $C_t = [Z_v^{(t)}; Z_a^{(t)}]$。然后进入第一阶段STVP:对于每个chunk的视觉token(对应两帧 $F_1^{(t)}, F_2^{(t)}$),计算空间显著性分数 $s_{1,i}^{(t)} = 1 - \frac{v_{1,i}^{(t)} \cdot \bar{v}_1^{(t)}}{\|v_{1,i}^{(t)}\|\|\bar{v}_1^{(t)}\|}$(基于与帧均值的余弦距离)和时间显著性分数 $s_{2,i}^{(t)} = 1 - \frac{v_{2,i}^{(t)} \cdot v_{1,i}^{(t)}}{\|v_{2,i}^{(t)}\|\|v_{1,i}^{(t)}\|}$(基于相邻帧对应patch的余弦距离),然后通过TopK选择保留比例为 $\alpha_v$ 的最显著token,得到剪枝后的视觉序列 $\hat{Z}_v^{(t)}$。第二阶段VGAS:以音频token为查询、剪枝后的视觉token为键和值进行跨注意力计算 $H_a^{(t)} = \text{Softmax}\left(\frac{Q_a K_v^\top}{\sqrt{d}}\right)V_v$,加上残差连接得到 $\tilde{H}_a^{(t)} = H_a^{(t)} + Z_a^{(t)}$,再通过两层MLP和sigmoid计算音频显著性分数 $s_{a,j}^{(t)} = \sigma(\text{MLP}(\tilde{h}_{a,j}^{(t)}))$,最后通过TopK选择保留比例为 $\alpha_a$ 的音频token。训练时使用STE解决TopK的不可微问题。
技术新颖性
OmniSIFT的技术新颖性体现在以下几个方面:首先,首次明确提出并利用Omni-modal token压缩中的模态非对称性,提出视觉引导音频的压缩范式,这与OmniZip的音频引导视频策略形成鲜明对比。其次,STVP模块设计了双通道显著性评估——空间显著性捕获帧内结构冗余,时间显著性捕获帧间重叠,两者互补实现全面的视频冗余去除。第三,VGAS采用极简的单层跨注意力设计(实验证明3层变体反而性能下降),在保持跨模态依赖建模能力的同时最大化效率。第四,整个框架仅增加4.85M参数(不到7B模型的0.1%),却能实现显著的效率提升。最后,压缩框架与FlashAttention兼容,解决了OmniZip依赖注意力分数导致的算子兼容性问题。
实验结果
OmniSIFT在五个音频-视觉基准测试上展示了强大且一致的性能。在Qwen2.5-Omni-7B上,仅保留35% token时,OmniSIFT在WorldSense上达到50.0分,超越全token模型的49.7分;在VideoMME上达到68.3%平均准确率,超越全token的67.6%。在更激进的25%保留率下,OmniSIFT仍保持49.9和68.2的高分,而OmniZip分别降至48.1和66.0。效率方面,OmniSIFT将7B模型的总推理时间减少40%以上,峰值GPU内存降低4.6GB以上。在DailyOmni细粒度类别中,OmniSIFT在Event Sequence上达到66.7(OmniZip仅61.8),在AV Event Alignment上达到68.9(OmniZip仅59.7),展示了在需要复杂时空和跨模态推理任务上的显著优势。消融实验证明STVP和VGAS两个模块都不可或缺:去除空间或时间组件会导致准确率下降3-4%,去除视觉引导(仅用音频自注意力)会导致3.9%和2.9%的下降。范式消融显示,在所有保留率下,OmniSIFT的非对称策略都优于对称变体,且保留率越低优势越明显。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| WorldSense | Accuracy (%) | 50.0 | 49.7 (Full Tokens) | +0.3 |
| VideoMME (Avg) | Accuracy (%) | 68.3 | 67.6 (Full Tokens) | +0.7 |
| DailyOmni (Average) | Accuracy (%) | 73.2 | 72.2 (Full Tokens) | +1.0 |
| WorldSense (25%) | Accuracy (%) | 49.9 | 48.1 (OmniZip) | +1.8 |
| VideoMME (25%) | Accuracy (%) | 68.2 | 66.0 (OmniZip) | +2.2 |
| DailyOmni (25%) | Accuracy (%) | 72.5 | 66.2 (OmniZip) | +6.3 |
| 推理延迟 (7B) | E2E Latency (s) | 2.86 | 4.94 (Full Tokens) | -42% |
| GPU内存 (7B) | Peak Memory (GB) | 22.91 | 27.59 (Full Tokens) | -4.68GB |
局限与改进
作者承认的局限性包括:(1)音频和视觉信号之间的精确对齐仍然具有挑战性,因为这类任务通常依赖高度局部化的证据,对token剪枝敏感;(2)STVP是轻量级的chunk级别模块,目前仅在每个chunk内建模时间冗余,尚未探索chunk间和长程音视频结构的时间冗余;(3)OmniSIFT使用固定的、与查询无关的token预算来支持广泛的全能模态任务。从个人观察来看,OmniSIFT在3B模型上的提升幅度明显小于7B模型,暗示该方法的效果可能与模型容量相关;此外,论文未充分讨论不同音频-视频质量(如低质量音频、模糊视频)下的鲁棒性;VGAS的单层跨注意力设计虽然高效,但可能限制了复杂跨模态关系的建模能力。
独立分析的弱点
基于对论文的独立分析,OmniSIFT存在以下弱点及改进方向:(1)固定压缩预算问题:当前使用全局固定的视频和音频保留率,但不同场景的信息密度差异很大(如静态讲座vs动态体育比赛),改进方向是引入自适应预算分配,根据视频内容动态调整压缩比,论文中的初步实验显示自适应策略未带来明显提升且增加延迟,但这可能是因为实现方式不够精细,值得进一步探索;(2)chunk级时间建模局限:STVP仅在2秒chunk内建模时间冗余,忽略了chunk间的长期依赖,改进方向是引入跨chunk注意力或层次化压缩策略;(3)单层VGAS的表达能力限制:虽然实验证明3层变体性能下降,但这可能是因为训练数据不足或正则化不够,而非架构本身的问题,可以探索更深层但带适当约束的VGAS设计;(4)缺乏任务自适应:当前压缩与具体查询无关,对于需要特定模态信息的任务(如语音识别需要完整音频),可能造成关键信息丢失,改进方向是引入查询感知的压缩策略。
未来方向
作者在论文末尾提出了几个未来研究方向:(1)探索自适应预算分配策略,针对信息密度非均匀的视频进行动态token分配;(2)引入查询引导的剪枝,实现任务特定的压缩;(3)在多轮对话场景中保持上下文一致性;(4)更好地保留音频触发的视觉证据。基于OmniSIFT的成果,还可以延伸以下方向:(5)将模态非对称压缩思想扩展到更多模态(如深度、红外等);(6)探索压缩与量化、剪枝等其他效率技术的协同;(7)研究压缩对下游任务(如视觉问答、视频生成)的长期影响;(8)开发压缩友好的预训练策略,使模型天生更适合压缩。
复现评估
OmniSIFT的复现条件相对友好:(1)代码已开源在GitHub(https://github.com/dingyue772/OmniSIFT),提供了完整的训练和评估代码;(2)训练数据使用公开的AVoCaDO SFT数据集(107K音视频描述对);(3)基座模型Qwen2.5-Omni-7B和3B均为开源模型;(4)VGAS模块仅需4.85M额外参数,训练资源需求相对较低;(5)论文提供了详细的实现细节,包括学习率(1e-5)、批大小(128)、压缩比率配置等。复现难度主要在于:需要8卡GPU进行训练(批大小128),推理评估需要加载7B模型,以及需要正确配置Qwen2.5-Omni的推理环境。总体而言,对于有中等算力资源的研究团队,复现难度为中等偏低。
论文图表