arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.09024cs.CV

基于掩码位建模的自回归图像生成

Autoregressive Image Generation with Masked Bit Modeling

Amazon FAR (Frontier AI \& Robotics)

Qihang Yu, Qihao Liu, Ju He, Xinyang Zhang, Yang Liu, Liang-Chieh Chen, Xi Chen

AI总结:

本文提出基于掩码位建模的自回归框架,通过扩大代码本大小有效缩小离散与连续方法的性能差距,实现更高效的图像生成。

摘要:

本文挑战了连续管道在视觉生成中的主导地位。我们系统地研究了离散方法与连续方法之间的性能差距。与认为离散分词器本质上较差的观点相反,我们证明这种差距主要源于潜在空间中分配的总位数(即压缩比)。我们显示,扩大代码本大小可以有效缩小这一差距,使离散分词器能够匹配或超越连续方法。然而,现有离散生成方法难以利用这一见解,当代码本扩大时,它们会因性能下降或训练成本过高而受到限制。为了解决这个问题,我们提出了掩码位自回归建模(BAR),一个可扩展的框架,支持任意大小的代码本。通过在自回归变压器中添加一个掩码位建模头,BAR通过逐步生成每个离散令牌的组成部分位来预测离散令牌。BAR在ImageNet-256上实现了新的最先进的gFID值0.99,优于连续和离散范式中的领先方法,同时显著降低了采样成本,并比先前的连续方法收敛得更快。项目页面可在https://bar-gen.github.io/上找到。

英文摘要:

This paper challenges the dominance of continuous pipelines in visual generation. We systematically investigate the performance gap between discrete and continuous methods. Contrary to the belief that discrete tokenizers are intrinsically inferior, we demonstrate that the disparity arises primarily from the total number of bits allocated in the latent space (i.e., the compression ratio). We show that scaling up the codebook size effectively bridges this gap, allowing discrete tokenizers to match or surpass their continuous counterparts. However, existing discrete generation methods struggle to capitalize on this insight, suffering from performance degradation or prohibitive training costs with scaled codebook. To address this, we propose masked Bit AutoRegressive modeling (BAR), a scalable framework that supports arbitrary codebook sizes. By equipping an autoregressive transformer with a masked bit modeling head, BAR predicts discrete tokens through progressively generating their constituent bits. BAR achieves a new state-of-the-art gFID of 0.99 on ImageNet-256, outperforming leading methods across both continuous and discrete paradigms, while significantly reducing sampling costs and converging faster than prior continuous approaches. Project page is available at https://bar-gen.github.io/

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

↑