通过层级词元压缩加速流式视频大语言模型
Accelerating Streaming Video Large Language Models via Hierarchical Token Compression
STC 的层级词元压缩思路可直接用于任何流式视频视觉语言模型:在 ViT 端缓存并复用时序相似帧的特征,在 LLM 预填充前按时空相关性剪枝视觉词元,从而在不重训练的情况下降低延迟,适合实时视频理解部署。
原文依据 STC retains up to 99% of accuracy on the ReKV framework while reducing ViT encoding latency and LLM pre-filling latency by 24.5% and 45.3%.
摘要译文
由模型翻译 · 术语统一 · 请对照原文核对关键结论流式视频大语言模型(VideoLLMs)在各类视频理解任务中表现出色,但由于处理连续视频流中密集视觉词元的计算成本高昂,其在实时部署中面临重大挑战。在流式视频场景中,主要瓶颈在于视觉 Transformer(ViT)编码阶段,对时序相似帧的冗余处理导致效率低下。此外,大语言模型预填充阶段膨胀的词元序列进一步加剧了延迟和内存开销。为解决这些挑战,我们提出流式词元压缩(STC),一个即插即用的层级框架,可无缝集成到现有流式 VideoLLMs 中,同时优化 ViT 编码和大语言模型预填充两个阶段以加速处理。STC 引入两个词元级加速器:STC-Cacher 通过缓存并复用时序相似帧的特征来降低 ViT 编码开销;STC-Pruner 在视觉词元序列进入大语言模型之前对其进行压缩,基于空间和时序相关性仅保留最显著的词元。在五个基准上对四个基线流式 VideoLLMs 的大量实验表明,STC 优于其他压缩方法。值得注意的是,STC 在 ReKV 框架上保持高达 99% 的准确率,同时将 ViT 编码延迟和大语言模型预填充延迟分别降低 24.5% 和 45.3%。
英文原文摘要 1465 字符
Streaming Video Large Language Models (VideoLLMs) have demonstrated impressive performance across various video understanding tasks, but they face significant challenges in real-time deployment due to the high computational cost of processing dense visual tokens from continuous video streams. In streaming video scenarios, the primary bottleneck lies in the Vision Transformer (ViT) encoding stage, where redundant processing of temporally similar frames leads to inefficiency. Additionally, inflated token sequences during LLM pre-filling further exacerbate latency and memory overhead. To address these challenges, we propose Streaming Token Compression (STC), a plug-and-play hierarchical framework that seamlessly integrates into existing streaming VideoLLMs, optimizing both ViT encoding and LLM pre-filling stages to accelerate processing. STC introduces two token-level accelerators: STC-Cacher, which reduces ViT encoding overhead by caching and reusing features from temporally similar frames, and STC-Pruner, which compresses the visual token sequence before it enters the LLM, preserving only the most salient tokens based on both spatial and temporal relevance. Extensive experiments on four baseline streaming VideoLLMs across five benchmarks demonstrate that STC outperforms other compression methods. Notably, STC retains up to 99% of accuracy on the ReKV framework while reducing ViT encoding latency and LLM pre-filling latency by 24.5% and 45.3%.