← 返回 2026-02-05

Quant VideoGen:基于2-Bit KV缓存量化的自回归长视频生成 Quant VideoGen: Auto-Regressive Long Video Generation via 2-Bit KV-Cache Quantization

Haocheng Xi, Shuo Yang, Yilong Zhao, Muyang Li, Han Cai, Xingyang Li, Yujun Lin, Zhuoyang Zhang, Jintao Zhang, Xiuyu Li, Zhiying Xu, Jun Wu, Chenfeng Xu, Ion Stoica, Song Han, Kurt Keutzer 📅 2026-02-03 👍 34 2026-07-13 08:35
KV缓存量化 内存优化 自回归模型 视频扩散模型 视频生成

通过语义感知平滑与渐进残差量化,实现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%之间。

Semantic-Aware Smoothing effectively smoothing the KV-cache distribution
Figure 3: Semantic-Aware Smoothing effectively smoothing the KV-cache distribution
Overview of QVG framework
Figure 4: Overview of QVG framework
Semantic-Aware Smoothing effectively reduces quantization error
Figure 6: Semantic-Aware Smoothing effectively reduces quantization error

实验结果

本文在三个自回归视频生成模型上进行了全面评估。在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)。

Quality and Compression results of QVG and baselines
Table 1: Quality and Compression results of QVG and baselines
VBench Image Quality at increasing video lengths
Table 2: VBench Image Quality at increasing video lengths
Quality comparison on LongCat-Video at 720p under INT2 quantization
Table 3: Quality comparison on LongCat-Video at 720p under INT2 quantization
Effect of chunk size on overhead and compression ratio
Table 4: Effect of chunk size on overhead and compression ratio
Effect of codebook size K under INT4 and INT2 quantization
Table 5: Effect of codebook size K under INT4 and INT2 quantization
Latency and QVG overhead on H100 vs. RTX 5090
Table 6: Latency and QVG overhead on H100 vs. RTX 5090
Latency and QVG overhead at different batch sizes
Table 7: Latency and QVG overhead at different batch sizes
QVG makes long video generation extremely memory-efficient and maintains high video quality
Figure 1: QVG makes long video generation extremely memory-efficient and maintains high video quality
Imaging Quality over long-horizon generation and MSE reduction across stages
Figure 5: Imaging Quality over long-horizon generation and MSE reduction across stages
Memory usage decomposition and trade-off curves
Figure 7: Memory usage decomposition and trade-off curves
查看结构化数据
任务指标本文基线提升
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优化部分。