张大妈

世界模型深入浅出 | VQ家族论文整理(VQ-VAE/VQ-GAN/RQ-VAE等)

源自知乎:极市平台

01-19 11:13

视觉生成领域为何痴迷离散化?本文系统梳理了VQ家族从VQ-VAE到FlexVAR的技术演进,深入剖析了“先压缩后生成”路线下的信息瓶颈、残差预测与并行解码等核心思想。对于想理解当前图像生成模型底层逻辑的读者,这是一份清晰的技术路线图。

世界模型深入浅出 | VQ家族论文整理(VQ-VAE/VQ-GAN/RQ-VAE等)智能速览

  • 神经网络是函数拟合器而非概率密度拟合器,离散化是图像生成的关键选择。

  • VQ-VAE/VQ-GAN开创“先压缩后生成”范式,但高压缩比导致细节损失。

  • RQ-VAE采用多级残差量化,用多个码本分层表达图像的框架与纹理。

  • VAR引入“下一尺度预测”,通过并行解码大幅提升生成速度,但步长设计刚性。

  • FlexVAR改进VAR,直接预测真值而非残差,实现语义连续与生成步长灵活。

  • LADM探索无VQ的自回归方案,结合扩散模型在连续空间进行生成。

世界模型深入浅出 | VQ家族论文整理(VQ-VAE/VQ-GAN/RQ-VAE等)精华内容

要理解这些模型的演进,关键在于看它们如何应对“信息损失”和“生成效率”两大核心挑战。从离散化思想的开端到对残差预测的反思,每一步都充满巧思。

离散化的必然

神经网络是函数的万能拟合器,但并非概率密度的万能拟合器,这给连续型变量的生成建模带来本质困难。离散型变量的概率约束(求和为1)则能通过Softmax轻松实现,这便是离散化的根本动因。

然而,直接在像素空间进行自回归(AR)生成,步数过于庞大(例如256×256图像约需20万步),计算成本高昂。因此,“先压缩后生成”成为主流思路,即先将图像压缩到低维离散空间,再进行生成,最后解码恢复。但强压缩必然带来信息损失,信息论估算显示,若要在16×16或32×32的网格上无损承载ImageNet图像信息,所需词表规模远超现有能力,信息损失问题亟待解决。

VQ-GAN的突破

VQ-VAE是这一路线的基石,它通过一个码本将编码器输出的连续特征映射为离散的索引,并利用梯度直通估计器解决离散操作的不可导问题。其损失函数包含重建损失和码本更新承诺两部分,以保证码本稳定和编码器不漂移。

VQ-GAN则在此基础上进行了三大关键升级,显著提升了高压缩比下的生成质量。首先,它采用LPIPS或VGG等感知损失替代像素级L2损失,使重建图像在感知上更逼真。其次,引入了基于Patch的对抗判别器,通过最小最大博弈进一步提升生成细节的真实感。最后,提出滑动注意力窗口机制,将Transformer的注意力复杂度从O(L^2)降低至O(L*k),有效解决了高分辨率生成的计算瓶颈。

RQ-VAE的深化

为缓解单一码本带来的信息损失,RQ-VAE(Residual Quantized VAE)提出了多层级残差量化的思想。与VQ-VAE中一个patch仅对应一个码本索引不同,RQ-VAE为每个patch特征分配D个码本词表。

其操作流程是:先在第一个码本中查找最接近的code,计算残差;然后在第二个码本中为该残差查找最接近的code,再计算新的残差,如此循环D次。这使得不同层级的码本能够分别学习图像的框架、纹理等不同层次的信息。在推理时,模型先进行空间维度的自回归,再在每个空间位置上进行深度维度的自回归,逐步重建出完整的细节。

VAR的并行加速

VAR(Visual Autoregressive Modeling)改变了传统逐token预测的范式,提出了“下一尺度预测”的方法。它不再预测单个token,而是并行地预测整个下一尺度的token map,大大加快了推理速度。

具体而言,VAR预测的是当前尺度与上一尺度上采样结果之间的残差。在采样时,它借鉴MaskGIT的并行解码策略,采用基于置信度的重掩码和多步精炼,通过余弦调度等方式,逐步补全图像。实验结果显示,VAR的推理步数极少,速度远超传统AR模型,并且呈现出明显的Scaling Law,性能随模型规模和计算量的增加而持续提升。但其缺点在于步长设计刚性,必须严格按顺序从低分辨率到高分辨率生成。

FlexVAR的灵活性

FlexVAR(Flexible Visual Autoregressive Modeling)针对VAR的刚性步长问题进行了改进。其核心变革是将VAR的“下一尺度残差预测”改为“直接预测下一尺度的真值”(Ground-Truth latent token map)。

这一改变带来了两大优势:第一,它确保了相邻尺度之间的语义连续性,因为每一步都直接对齐真实值。第二,也是最关键的,它打破了VAR的刚性步骤设计,使得在推理的任何一步都能独立输出一幅可用的图像。此外,FlexVAR还提供了可缩放的2D位置编码与随机步采样,让推理时的步数、分辨率和宽高比都可以灵活调整,极大地增强了模型的应用灵活性与扩展性。

内容由AI生成
0
扫一下,分享更方便,购买更轻松
0评论

当前文章无评论,是时候发表评论了
提示信息

取消
确认
评论举报

最新文章 热门文章