通过动态检索减少视频LLM的缓存占用,实现边缘设备实时推理。
V-Rex: Real-Time Streaming Video LLM Acceleration via Dynamic KV Cache Retrieval
- 提出无需训练的ReSV算法,按时空相似性聚类帧间令牌以压缩缓存。
- 在边缘设备上实现3.9-8.3 FPS实时推理,精度几乎无损失。
- 软硬件协同设计,加速器仅占2%面积与功耗,性能提升1.9-19.7倍。
流式视频大语言模型(LLMs)广泛应用于实时多模态任务,如视频字幕生成、问答系统和增强现实。然而,持续视频输入导致键值(KV)缓存急剧增长,引发显著计算开销、数据传输负担及精度下降。尤其在边缘部署场景下问题更严重。本文提出V-Rex,首个软硬件协同设计的加速架构,全面解决算法与硬件瓶颈。核心是无训练的动态KV缓存检索算法ReSV,利用时空相似性进行令牌聚类,有效降低缓存冗余。配套硬件设计了低延迟动态缓存检索引擎(DRE),采用位级计算与早期退出机制。在边缘设备上实现3.9-8.3 FPS的实时推理,精度几乎无损;尽管DRE仅占2.2%功耗和2.0%面积,系统相较AGX Orin GPU仍实现1.9-19.7倍加速与3.1-18.5倍能效提升。该工作首次在算法与硬件层面系统性解决流式视频LLM的缓存检索难题。
原文摘要 · Abstract (English)
Streaming video large language models (LLMs) are increasingly used for real-time multimodal tasks such as video captioning, question answering, conversational agents, and augmented reality. However, these models face fundamental memory and computational challenges because their key-value (KV) caches grow substantially with continuous streaming video input. This process requires an iterative prefill stage, which is a unique feature of streaming video LLMs. Due to its iterative prefill stage, it suffers from significant limitations, including extensive computation, substantial data transfer, and degradation in accuracy. Crucially, this issue is exacerbated for edge deployment, which is the primary target for these models. In this work, we propose V-Rex, the first software-hardware co-designed accelerator that comprehensively addresses both algorithmic and hardware bottlenecks in streaming video LLM inference. At its core, V-Rex introduces ReSV, a training-free dynamic KV cache retrieval algorithm. ReSV exploits temporal and spatial similarity-based token clustering to reduce excessive KV cache memory across video frames. To fully realize these algorithmic benefits, V-Rex offers a compact, low-latency hardware accelerator with a dynamic KV cache retrieval engine (DRE), featuring bit-level and early-exit based computing units. V-Rex achieves unprecedented real-time of 3.9-8.3 FPS and energy-efficient streaming video LLM inference on edge deployment with negligible accuracy loss. While DRE only accounts for 2.2% power and 2.0% area, the system delivers 1.9-19.7x speedup and 3.1-18.5x energy efficiency improvements over AGX Orin GPU. This work is the first to comprehensively tackle KV cache retrieval across algorithms and hardware, enabling real-time streaming video LLM inference on resource-constrained edge devices.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。