Keep What Audio Cannot Say: Context-Preserving Token Pruning for Omni-LLMs
📄 Keep What Audio Cannot Say: Context-Preserving Token Pruning for Omni-LLMs #音视频 #高效推理 #多模态模型 #跨模态 ✅ 7.0/10 | 前25% | #音视频 | #高效推理 | #多模态模型 #跨模态 | arxiv 学术质量 6.5/8 | 影响力 0.5/2 | 可复现性 0.5/1 | 置信度 高 👥 作者与机构 第一作者:Chaeyoung Jung(Korea Advanced Institute of Science and Technology (KAIST)) 通讯作者:未明确说明(论文中列出的通讯作者地址为“Equal contribution”,与第一作者相同) 作者列表:Chaeyoung Jung(KAIST)、Kyeongha Rho(KAIST)、Joon Son Chung(KAIST) 💡 毒舌点评 这篇论文的“移除音频已知,保留音频未知”的剪枝思想确实新颖且直观,实验结果也非常亮眼,尤其是在Qwen2.5-Omni 7B上能以55%的压缩率保持甚至超越全token性能。但尴尬的是,方法最核心的“保留空间细节”分支(图2中标注2⃝)依赖于一个固定的全局空间保留比例ρspa,这在处理如球员球衣号码、面部表情等高度局部化、信息量集中的细节时显得非常脆弱,论文的失败案例也明确证实了这一点。一个自适应的、可能依赖内容或查询的空间预算分配机制,才是该方法从“有效”迈向“鲁棒”的关键下一步。此外,将所有实验仅限于短视频片段,其结论在长视频场景下的普适性存疑。 📌 核心摘要 要解决什么问题:全模态大语言模型(Omni-LLMs)在处理音视频输入时,会产生大量token,导致计算开销巨大,阻碍了实际部署。现有的token剪枝方法要么针对特定任务训练,要么过度依赖音频-视觉的强对齐信号,可能丢弃理解宽泛上下文所需的证据。 方法核心是什么:论文提出了一个名为ContextGuard的推理时token剪枝框架。其核心思想是将token压缩从“选择重要token”重新定义为“移除跨模态冗余,同时保留音频无法表达的视觉上下文”。具体实现包含三个主要步骤:a) 音频引导的语义剪枝:使用一个轻量级的音频到视觉语义预测器(A2V Predictor)从音频预测粗糙视觉语义,移除与预测语义高度相似(即可被音频解释)的视频token;b) 空间细节保留:通过网格采样和局部空间变化度量,额外保留一部分token以覆盖图像空间,保存颜色、姿态等音频无法指定的局部细节;c) 基于深度分数的时间合并:对时间上相似的chunk进行分段和合并,进一步压缩冗余帧。 与已有方法相比新在哪里:与现有训练基础的(如EchoingPixels, OmniSIFT)或基于对齐的(如OmniZip)方法不同,ContextGuard不追求为当前查询选择最相关的token,而是致力于去除被音频流冗余表达的视觉信息,并主动保留那些与音频不匹配的视觉证据(如场景文字、背景物体),以维护更广泛的上下文信息。 主要实验结果如何:在Qwen2.5-Omni(7B/3B)和Video-SALMONN2+(7B/3B)两个模型及六个音视频基准测试上,ContextGuard显著优于基线方法。关键结果如表1所示: Method Comp.↑ (%) World.↑ Daily.↑ Video-MME↑ OmniVid.↑ AVQA↑ video-SAL2.↓ Avg.↑ (%) Qwen2.5-Omni 7B Full Token 0 47.4 57.1 78.8 48.2 87.6 48.1 100.0 Random 50 45.7 52.4 78.4 43.4 86.7 50.4 95.4 FastV 50 45.6 56.6 77.9 47.6 87.0 49.8 98.1 OmniZip 54 46.8 56.6 77.1 47.0 87.0 52.8 97.3 ContextGuard 55 47.7 57.2 78.8 48.2 87.1 48.1 100.0 Qwen2.5-Omni 3B Full Token 0 47.7 57.7 75.8 44.0 87.6 53.5 100.0 Random 50 44.1 53.1 74.0 42.8 87.1 56.1 95.7 FastV 50 46.7 55.6 74.0 44.0 87.2 54.6 98.2 OmniZip 54 47.1 55.8 74.9 42.8 87.3 52.4 98.9 ContextGuard 61 47.7 56.6 75.8 45.2 87.3 52.3 100.5 Video-SALMONN2+ 7B Full Token 0 50.7 56.3 79.2 43.4 67.8 18.9 100.0 Random 50 47.3 51.6 76.2 39.5 66.8 25.7 90.7 FastV 50 46.3 51.8 79.2 42.2 66.1 24.0 92.8 OmniZip 49 47.8 53.8 79.2 42.8 68.8 22.4 95.7 ContextGuard 55 50.6 55.5 81.4 47.0 66.9 19.9 100.5 Video-SALMONN2+ 3B Full Token 0 48.1 57.7 76.2 45.8 81.6 22.7 100.0 Random 50 44.3 51.6 73.2 42.0 81.2 30.3 90.6 FastV 50 45.6 53.2 71.0 36.1 81.0 27.4 90.2 OmniZip 49 45.6 54.7 73.2 41.6 81.3 26.3 93.7 ContextGuard 56 47.1 55.5 73.6 44.6 81.4 24.2 96.9 特别是在Qwen2.5-Omni 7B上,剪枝55%的token后,在6个基准中的5个上达到了全token性能。 实际意义是什么:该方法为部署高效的全模态大模型提供了有力的工具,通过一个无需微调下游LLM、仅需一个独立训练的轻量预测器的推理时框架,大幅降低了音视频理解的计算成本和内存占用,对实时交互应用具有重要价值。 主要局限性是什么:方法依赖于固定的精细粒度空间保留比例(ρspa),这可能导致在需要非常具体局部细节的任务(如识别球衣号码、细微表情)上表现不佳,正如其失败案例所示。此外,所有评估均基于短于1分钟的视频片段,限制了结论对长视频场景的适用性。A2V预测器的鲁棒性在不同内容类型(如音乐视频)上未被充分讨论。 🔗 开源详情 代码:论文中未提及代码链接。 模型权重:论文中未提及模型权重链接。 数据集: WorldSense: 论文中未提供链接。 Daily-Omni: 论文中未提供链接。 Video-MME: 论文中未提供链接。 AVQA: 论文中未提供链接。 OmniVideoBench: 论文中未提供链接。 video-SALMONN2 test set: https://huggingface.co/datasets/videoSALMONN2/video-SALMONN_2_testset AudioSet: 论文中提及,未提供具体链接。 VGGSound: 论文中提及,未提供具体链接。 Demo:论文中未提及。 复现材料:论文提供了详细的附录,包括 A2V 预测器的架构、训练目标、训练数据(AudioSet 和 VGGSound)、超参数设置、以及各项分析实验的细节(如语义保留比例、空间细节保留策略、时序合并实现等)。未提及训练检查点或代码库。 论文中引用的开源项目: Qwen2.5-Omni: 论文中作为基线模型使用,未提供代码或权重仓库链接。 Video-SALMONN2+: 论文中作为基线模型使用,未提供代码或权重仓库链接。 OmniZip: 论文中作为对比方法,未提供代码或仓库链接。 FastV: 论文中作为对比方法,未提供代码或仓库链接。 🏗️ 方法概述和架构 整体流程概述:ContextGuard是一个在LLM解码器之前运行的、多阶段的音视频token剪枝框架。其输入为交替的音视频token序列,输出为压缩后的视觉token序列和完整的音频token序列。该框架并非端到端系统,而是由一个独立训练的轻量级A2V预测器和一个基于启发式规则的剪枝-合并流水线组成。 ...