arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2411.10433cs.CV

M-VAR:用于高质量图像生成的解耦尺度级自回归建模

M-VAR: Decoupled Scale-wise Autoregressive Modeling for High-Quality Image Generation

Johns Hopkins University(约翰斯·霍普金斯大学);University of California, Berkeley(加州大学伯克利分校);Google(谷歌公司);UC Santa Cruz(加州大学圣克鲁兹分校)

Sucheng Ren, Yaodong Yu, Nataniel Ruiz, Feng Wang, Alan Yuille, Cihang Xie

AI总结:

本文提出M-VAR框架,将尺度级自回归图像生成解耦为尺度内双向自注意力建模与尺度间基于Mamba的线性复杂度建模,在提升计算效率的同时,以更少参数实现了更优的图像质量与生成速度。

摘要:

近期计算机视觉领域有一项名为VAR的研究,提出了一种用于图像生成的新自回归范式。与普通的下一词元预测不同,VAR在结构上将图像生成重新表述为从粗到细的下一尺度预测。在本文中,我们展示了这种尺度级自回归框架可以有效地解耦为捕获每个尺度内局部空间依赖的尺度内建模,以及从粗到细逐步建模跨尺度关系的尺度间建模。这种解耦结构允许以计算更高效的方式重建VAR。具体而言,对于对生成高保真图像至关重要的尺度内建模,我们保留原始的双向自注意力设计以确保全面建模;对于在语义上连接不同尺度但计算量大的尺度间建模,我们应用Mamba等线性复杂度机制来大幅减少计算开销。我们将这个新框架称为M-VAR。大量实验表明,我们的方法在图像质量和生成速度上均优于现有模型。例如,我们的1.5B模型以更少的参数和更快的推理速度,超越了最大的VAR-d30-2B。此外,我们最大的模型M-VAR-d32在ImageNet 256×256上取得了1.78的FID,分别优于先前先进的自回归模型LlamaGen/VAR 0.4/0.19,以及流行的扩散模型LDM/DiT 1.82/0.49。代码可在https://github.com/OliverRensu/MVAR获取。

英文摘要:

There exists recent work in computer vision, named VAR, that proposes a new autoregressive paradigm for image generation. Diverging from the vanilla next-token prediction, VAR structurally reformulates the image generation into a coarse to fine next-scale prediction. In this paper, we show that this scale-wise autoregressive framework can be effectively decoupled into \textit{intra-scale modeling}, which captures local spatial dependencies within each scale, and \textit{inter-scale modeling}, which models cross-scale relationships progressively from coarse-to-fine scales. This decoupling structure allows to rebuild VAR in a more computationally efficient manner. Specifically, for intra-scale modeling -- crucial for generating high-fidelity images -- we retain the original bidirectional self-attention design to ensure comprehensive modeling; for inter-scale modeling, which semantically connects different scales but is computationally intensive, we apply linear-complexity mechanisms like Mamba to substantially reduce computational overhead. We term this new framework M-VAR. Extensive experiments demonstrate that our method outperforms existing models in both image quality and generation speed. For example, our 1.5B model, with fewer parameters and faster inference speed, outperforms the largest VAR-d30-2B. Moreover, our largest model M-VAR-d32 impressively registers 1.78 FID on ImageNet 256$\times$256 and outperforms the prior-art autoregressive models LlamaGen/VAR by 0.4/0.19 and popular diffusion models LDM/DiT by 1.82/0.49, respectively. Code is avaiable at \url{https://github.com/OliverRensu/MVAR}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑