Quant VideoGen:基于2-Bit KV缓存量化的自回归长视频生成 Quant VideoGen: Auto-Regressive Long Video Generation via 2-Bit KV-Cache Quantization
通过语义感知平滑与渐进残差量化,实现7倍KV缓存压缩的自回归长视频生成
前置知识
KV缓存(Key-Value Cache)
在自回归Transformer模型中,为避免重复计算,每一步推理时会将之前所有token的Key和Value向量缓存起来,这就是KV缓存。在视频生成模型中,由于每一帧对应大量空间token,KV缓存会随着生成帧数线性增长,迅速占据大量GPU显存。具体来说,对于一个有 $L$ 层、隐藏维度 $d$ 的模型,存储一段空间尺寸 $H \times W$、时间长度 $T$ 的视频的KV缓存需要 $\text{Mem}_{\text{KV}} = 2 \cdot L \cdot (HWT) \cdot d \cdot \text{Byte}_{\text{BF16}}$ 字节的显存。
KV缓存是本文要解决的核心瓶颈,理解其内存占用公式和增长规律是理解本文动机的前提
自回归视频扩散模型
与传统双向注意力的视频扩散模型(如Wan2.1、HunyuanVideo)不同,自回归视频扩散模型引入时间因果性,帧块按时间顺序逐步生成,每一步只依赖已有的历史帧。代表方法包括CausVid和Self-Forcing。这种范式使得模型可以流式输出、支持实时交互和长视频生成,但也引入了KV缓存随历史增长的系统瓶颈。
本文的方法专为自回归视频扩散模型设计,理解该范式与双向扩散的区别才能明白为何KV缓存成为关键瓶颈
对称分组整数量化(Symmetric Per-group Integer Quantization)
一种常见的低比特量化方案:将浮点张量按组(group)进行量化,每组独立计算缩放因子 $S_x = \frac{\max(|X_{\text{BF16}}|)}{2^{b-1}-1}$,然后将值映射到 $b$ 比特整数 $X_{\text{INT}} = \text{Round}(\frac{X_{\text{BF16}}}{S_x})$。量化误差与缩放因子 $S_x$ 成正比,而 $S_x$ 由组内最大绝对值决定,因此离群值(outlier)会导致严重的量化误差。
理解量化误差的数学形式(误差 $\propto S_x$)是理解本文Semantic-Aware Smoothing为何有效的关键
k-means聚类
一种经典的无监督聚类算法,将 $N$ 个数据点划分为 $C$ 个簇,使得每个点到其所属簇中心(centroid)的距离之和最小。在本文中,k-means被用于根据token的隐藏表示相似度将KV缓存token分组,使得组内token在数值分布上更加同质,便于后续的质心减除和残差量化。本文还提出了流式质心缓存策略,用前一个chunk的分配结果初始化下一个chunk的质心,将k-means开销降低3倍。
k-means是Semantic-Aware Smoothing的核心组件,理解其聚类逻辑和本文的加速优化对理解方法至关重要
渐进式残差量化(Progressive Residual Quantization)
受流式视频编码器的启发,该技术将残差张量在多个阶段(stage)中逐步量化:每个阶段对上一阶段的残差再次执行语义感知平滑和量化,从而从粗到细地捕获信息。第一阶段能带来最大幅度的MSE降低(5.83倍),后续阶段逐步减少误差但边际收益递减。这种设计使得压缩率和质量之间可以灵活权衡。
这是本文的第二个核心创新,理解其多阶段迭代机制和收益递减规律对评估方法性能很重要
研究动机
自回归视频扩散模型虽然在流式生成和实时交互方面具有优势,但面临一个严重的系统-算法耦合瓶颈:KV缓存内存。具体来说,以LongCat-Video为例,生成一段5秒480p视频需要约38K个latent token,对应的KV缓存约占34 GB显存,已经超过单张RTX 5090 GPU的显存容量。而模型参数本身仅占27 GB。随着生成时长增加,KV缓存线性增长,迅速成为GPU显存的首要瓶颈。更严重的是,为了控制显存占用,现有系统被迫采用固定长度的滑动窗口(如Self-Forcing默认21帧,HY-WorldPlay仅保留20帧),但这等价于缩小了模型的工作记忆,导致长视频生成中出现身份漂移、场景布局不一致和运动语义退化等问题。论文通过可视化实验(Figure 2a)清晰展示了使用完整KV缓存可以解决漂移问题但受限于内存,而截断KV缓存虽然可行但质量严重下降。
本文的目标是本文的目标是通过KV缓存低比特量化来同时解决自回归视频生成中的两个瓶颈:(1)显存瓶颈——将KV缓存从BF16压缩到INT2甚至更低,实现数倍的内存节省,使得原本无法在单GPU上运行的长视频生成变得可行;(2)质量瓶颈——在大幅压缩的同时保持接近无损的生成质量,避免现有量化方法在视频模型上导致的严重退化。具体而言,作者希望在H100或消费级RTX 4090/5090上实现HY-WorldPlay-8B等大模型的高效长视频生成,并在超过700帧甚至1400帧的长序列上保持视觉质量。
与已有工作不同的是,尽管KV缓存量化在LLM领域已有大量成熟工作(如KIVI、KVQuant、QuaRot),但直接将这些方法移植到视频扩散模型会导致严重的质量退化。根本原因在于视频模型的激活统计特性与LLM存在本质差异:视频模型的KV缓存呈现出高度异质的数值分布,在token维度和channel维度上都不均匀——某些token的离群channel对其他token并非离群(Figure 3a-b)。这种行为源于视频token对应多样的空间区域和运动模式,其注意力投影产生强烈非均匀的激活尺度。此外,现有LLM量化方法没有显式利用视频特有的时空冗余性:相邻帧和相邻空间patch的token在潜空间中高度相似(cosine similarity接近1),这种冗余为压缩提供了天然的结构化先验。本文正是从这个独特的视频特性出发,设计了专门面向视频KV缓存的量化框架。
核心方法
QVG的核心直觉是:视频KV缓存虽然数值范围大、分布不规则,但存在强烈的时空冗余——相邻token高度相似。如果能将相似的token分组并减去它们的公共部分(质心),剩余的残差就会变得数值小、分布集中,非常适合低比特量化。基于这个直觉,QVG包含两个核心组件:Semantic-Aware Smoothing(语义感知平滑)负责利用时空冗余生成量化友好的残差张量,Progressive Residual Quantization(渐进残差量化)则通过多阶段迭代进一步降低量化误差。整体技术路线为:首先用k-means对KV缓存token按隐藏表示相似度聚类,然后对每组减去质心得到残差,最后对残差进行多阶段的渐进量化。整个过程无需额外训练,是纯推理时的即插即用方案。
QVG与已有KV缓存量化方法的本质区别在于两个层面。第一,它不是直接对原始KV缓存做量化,而是先通过Semantic-Aware Smoothing变换数值分布,使之更适合量化。已有方法(如KIVI)假设LLM的KV缓存具有相对规则的分布,直接应用per-group量化即可,但视频模型的KV缓存分布高度异质,直接量化会导致极大的误差。第二,它显式利用了视频的时空冗余结构。LLM的token序列缺乏这种结构化冗余,因此LLM量化方法无法利用这一特性。具体来说,QVG观察到:(1) 固定空间位置的token在相邻帧间cosine similarity接近1(Figure 2b);(2) 空间邻近patch的token也有很高的cosine similarity(Figure 2c)。这种冗余意味着k-means聚类后组内token的质心能捕获大部分共同信息,减去质心后残差幅值大幅降低。实验表明,Semantic-Aware Smoothing可将Key缓存的量化误差降低约6.9倍,Value缓存降低约2.6倍(Figure 6)。
方法步骤详情
QVG的完整处理流程分为以下步骤。输入为一段视频的KV缓存张量 $X \in \mathbb{R}^{N \times d}$,其中 $N$ 为token数,$d$ 为head维度。第一步,语义感知平滑(Semantic-Aware Smoothing):(a) 将 $N$ 个token按隐藏表示用k-means聚类为 $C$ 组 $G = \{G_1, G_2, \ldots, G_C\}$;(b) 计算每组质心 $C_i \in \mathbb{R}^d$;(c) 对每组减去质心得到残差 $R_i = X_{G_i} - C_i$。第二步,渐进残差量化(Progressive Residual Quantization):设 $R^{(0)} = X$,共 $T$ 个阶段。在每个阶段 $t$,对 $R^{(t-1)}$ 再次执行Semantic-Aware Smoothing得到 $R^{(t)}$、$C^{(t)}$、$\pi^{(t)}$。第三步,最终量化:对最后一个阶段的残差 $R^{(T)}$ 执行对称分组整数量化 $X_{\text{INT}}, S_x = Q(R^{(T)})$,存储到全局显存。同时存储所有阶段的质心 $C^{(t)}$ 和分配向量 $\pi^{(t)}$,中间残差 $R^{(t)}$ 作为临时结果丢弃。反量化时从阶段 $T$ 到阶段 $1$ 逐步恢复:先反量化得到 $\hat{X}^{(T-1)}$,然后逐阶段加上对应质心 $\hat{X}^{(t-1)} = \hat{X}^{(t)} + C^{(t)}_{\pi_i}$,最终得到重建张量 $\hat{X}^{(0)}$。
技术新颖性
QVG的技术新颖性体现在多个方面。首先,它首次系统性地识别并利用了视频KV缓存的时空冗余性来进行压缩,这在LLM KV缓存量化文献中从未被考虑。其次,Semantic-Aware Smoothing的设计巧妙地将减均值思想与语义聚类结合,使得残差张量的数值范围从原来的 $\max|K| \sim 10^2$、$\max|V| \sim 10^3$ 量级大幅下降,INT2量化变得可行。第三,Progressive Residual Quantization借鉴流式视频编码器的多尺度编码思想,在量化领域首次实现了coarse-to-fine的多阶段残差迭代量化,第一阶段即带来5.83倍的MSE降低。第四,算法-系统协同设计的思路值得关注:流式质心缓存策略将k-means开销降低3倍;融合的反量化内核将多阶段质心恢复操作在寄存器中完成,避免反复读取全局显存;采用pre-RoPE key caching使Key分布更规则;使用FP8 E4M3 per-group缩放因子减少元数据开销。这些工程优化使得整体端到端延迟开销控制在1.3%至4.3%之间。
实验结果
本文在三个自回归视频生成模型上进行了全面评估。在LongCat-Video-13B上,QVG在INT2设置下实现6.94倍压缩比的同时达到28.716 PSNR,而KIVI仅为20.317、QuaRot为21.573。在INT4设置下,QVG以3.72倍压缩比达到37.141 PSNR,远超所有基线。在HY-WorldPlay-8B上,INT2设置下QVG以7.05倍压缩比达到29.174 PSNR。更值得注意的是,QVG使得HY-WorldPlay-8B首次可以在单张RTX 4090上运行,且PSNR超过29。在长视频质量保持方面(Figure 5a),Self-Forcing模型的评估显示:在700帧的长序列上,QVG和QVG-Pro保持接近无损的Image Quality分数,而KIVI在约100帧后急剧退化。在1400帧(约90秒)的超长评估中(Table 2),QVG维持67.28的Image Quality,而KIVI降至35.85。在720p分辨率的可扩展性评估中(Table 3),QVG以INT2量化达到25.99 PSNR,显著优于KIVI的17.66和QuaRot的20.88,且无需增加聚类中心数。端到端延迟开销方面:LongCat-Video增加2.1%,HY-WorldPlay增加1.5%,Self-Forcing增加4.3%,在RTX 5090上也仅为1.9%(Table 6)。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| LongCat-Video-13B INT2 KV缓存量化 | PSNR / 压缩比 | QVG: 28.716 dB / 6.94×; QVG-Pro: 30.376 dB / 4.97× | RTN: 20.872 dB; KIVI: 20.317 dB; QuaRot: 21.573 dB(均6.40×) | QVG的PSNR比最强基线QuaRot高7.14 dB,压缩比更高(6.94× vs 6.40×) |
| LongCat-Video-13B INT4 KV缓存量化 | PSNR / 压缩比 | QVG: 37.141 dB / 3.72×; QVG-Pro: 37.095 dB / 3.05× | RTN: 32.984 dB; KIVI: 32.158 dB; QuaRot: 33.744 dB(均3.55×) | QVG的PSNR比QuaRot高3.40 dB,且压缩比更高 |
| HY-WorldPlay-8B INT2 KV缓存量化 | PSNR / 压缩比 | QVG: 29.174 dB / 7.05×; QVG-Pro: 31.562 dB / 5.20× | RTN: 24.199 dB; KIVI: 24.272 dB; QuaRot: 25.207 dB(均6.40×) | QVG的PSNR比QuaRot高3.97 dB,压缩比从6.40×提升到7.05× |
| HY-WorldPlay-8B INT4 KV缓存量化 | PSNR / 压缩比 | QVG: 34.454 dB / 3.75×; QVG-Pro: 35.109 dB / 3.15× | RTN: 33.634 dB; KIVI: 33.768 dB; QuaRot: 33.997 dB(均3.55×) | QVG的PSNR比QuaRot高0.46 dB,压缩比更高 |
| Self-Forcing长视频质量保持(700帧) | VBench Image Quality | QVG: 约69.5 | KIVI: 约57; QuaRot: 约45(急剧退化) | QVG在700帧后仍保持接近BF16(70.56)的质量,基线方法严重退化 |
| Self-Forcing超长视频(1400帧/90秒) | VBench Image Quality | QVG: 67.28 | KIVI: 35.85; QuaRot: 44.45; BF16: 65.87 | QVG甚至略优于BF16基线(67.28 vs 65.87),而KIVI退化超过50% |
| LongCat-Video 720p INT2量化 | PSNR | QVG: 25.99 dB | KIVI: 17.66 dB; QuaRot: 20.88 dB | QVG比QuaRot高5.11 dB,证明方法可扩展到更高分辨率 |
局限与改进
尽管QVG取得了显著成果,但论文也隐含或可推断出若干局限性。首先,QVG目前仅在480p和720p分辨率上验证,对于1080p或更高分辨率的视频生成,KV缓存的token数量会进一步膨胀,k-means聚类的计算开销和质心存储的元数据开销可能会增加,尽管论文在720p实验中展示了良好的可扩展性。其次,Progressive Residual Quantization虽然能进一步降低量化误差,但多个阶段需要存储多套质心和分配向量,在极端压缩场景下这些元数据本身也会占用可观的显存(Figure 7a显示元数据占比可达11%-35%)。第三,k-means聚类引入了额外的计算开销,虽然通过流式质心缓存已将开销降至1.3%-4.3%,但在实时交互场景(如游戏世界模型)中,即使是几毫秒的额外延迟也可能影响用户体验。第四,论文主要在H100和RTX 5090上评估,对于更低端的消费级GPU(如RTX 3060/4060),INT2量化所需的硬件支持和内存带宽约束可能带来额外挑战。最后,QVG的评估主要基于固定的prompt suite,缺乏在多样化的长视频生成场景(如电影级叙事、复杂多人交互)中的全面评估。
独立分析的弱点
QVG的几个具体弱点值得深入分析。第一,k-means聚类的超参数敏感性:聚类中心数 $C$ 的选择直接影响压缩比和质量的权衡,但论文未提供自动选择 $C$ 的方法。在不同视频内容(静态场景vs高速运动)下,最优的 $C$ 可能差异很大,这是一个潜在的实用化障碍。改进方向可以是自适应地根据token分布的复杂度动态调整 $C$。第二,INT2量化的精度天花板:即使经过Semantic-Aware Smoothing,2比特量化仍然只有4个量化级别,在视频内容快速变化的帧中(如剧烈运动、复杂纹理),残差的分布可能仍不够集中,导致局部质量退化。可以考虑混合精度策略,对信息密度高的帧或区域使用更高比特。第三,质心存储的内存开销:每个阶段需要存储 $C$ 个质心(每个 $d$ 维)和 $N$ 个分配索引,在 $T=4$ 阶段、$C=256$ 个中心的配置下,这些元数据可能占据相当比例的显存。改进方向包括对质心本身进行压缩或使用低秩表示。第四,当前方法对Key和Value使用相同的分组策略,但论文Figure 6显示Key和Value的统计特性有显著差异(Key的MSE降低6.9倍,Value仅2.6倍),可能需要针对Key和Value设计不同的平滑策略。
未来方向
基于QVG的成果,有多个值得探索的研究方向。首先,将QVG与其他压缩技术(如稀疏注意力、token剪枝)正交组合,可能实现更大幅度的压缩。论文提到Sparse VideoGen系列的工作已经在探索视频扩散中的稀疏性,与QVG的KV缓存量化结合是自然的延伸。其次,将QVG扩展到更长的生成时长(分钟级甚至小时级),这对世界模型、交互式游戏等应用至关重要。论文提到Google Project Genie等系统仍限制在60秒左右,QVG的技术有望突破这一限制。第三,探索自适应量化策略:根据视频内容的复杂度(如场景变化检测)动态调整量化比特数和聚类粒度,在简单场景用更激进的压缩,在复杂场景保留更多精度。第四,将QVG的思想推广到其他模态的自回归生成模型中,例如长序列音频生成或多模态流式推理中的KV缓存压缩。最后,开发硬件友好的INT2量化kernel和专用加速器支持,进一步降低QVG的实际部署门槛。
复现评估
QVG的复现条件相对友好。论文声称将开源代码(GitHub链接已在论文中标注),这对复现至关重要。方法本身是training-free的,不需要额外的训练数据或微调过程,只需在推理时插入量化模块即可。使用的三个模型(LongCat-Video-13B、HY-WorldPlay-8B、Self-Forcing-Wan-1.3B)均为开源模型。评估使用的是MovieGen benchmark的prompt suite,也是公开数据。算力方面,主要实验在NVIDIA H100上进行,但也展示了在消费级RTX 5090上的可行性,降低了复现的硬件门槛。自定义CUDA和Triton kernel的实现可能需要一定的GPU编程经验,但核心算法逻辑(k-means聚类加质心减除加多阶段残差量化)是清晰的,可以用PyTorch等高层框架快速原型验证。总体而言,复现难度中等偏低,主要挑战在于kernel优化部分。
论文图表
包含三个子图:(a)展示了使用完整KV缓存vs截断滑动窗口的质量差异,完整KV缓存可解决漂移问题但受限于内存;(b)展示了同一空间位置的token在相邻帧间的cosine similarity(接近1),证明时间维度的高度冗余;(c)展示了空间邻近patch的token的cosine similarity(随Manhattan距离增加而降低),证明空间维度的冗余。
该图提供了QVG方法的动机和理论基础:解释了为何需要保留长上下文(动机)、以及为何可以通过聚类和质心减除来压缩KV缓存(可行性)。