arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.04410cs.CVcs.AI

Open-MAGVIT2:一个旨在实现自回归视觉生成民主化的开源项目

Open-MAGVIT2: An Open-Source Project Toward Democratizing Auto-regressive Visual Generation

ARC Lab, Tencent PCG(腾讯平台与内容事业群ARC实验室);Tsinghua University(清华大学);Nanjing University(南京大学)

Zhuoyan Luo, Fengyuan Shi, Yixiao Ge, Yujiu Yang, Limin Wang, Ying Shan

AI总结:

本文开源复现谷歌MAGVIT-v2分词器,提出非对称分词分解与下一子token预测方法,构建300M至1.5B的自回归图像生成模型,并在重建与零样本基准上取得领先性能。

摘要:

Open-MAGVIT2项目对谷歌的MAGVIT-v2分词器进行了开源复现,该分词器拥有超大型码本(即$2^{18}$个码),并在ImageNet和UCF基准上取得了最先进的重建性能。我们还提供了一个在大规模数据上预训练的分词器,在零样本基准上显著优于Cosmos(在ImageNet原始分辨率上rFID为1.93对0.78)。此外,我们探索了其在普通自回归模型中的应用以验证可扩展性,产出了一系列参数量从300M到1.5B的自回归图像生成模型。为帮助自回归模型在超大词汇量下进行预测,我们通过非对称分词分解将其分解为两个不同大小的子词汇表,并进一步引入“下一子token预测”来增强子token间的交互,从而提升生成质量。我们发布所有模型和代码,以促进自回归视觉生成领域的创新与创造力。

英文摘要:

The Open-MAGVIT2 project produces an open-source replication of Google's MAGVIT-v2 tokenizer, a tokenizer with a super-large codebook (i.e., $2^{18}$ codes), and achieves the state-of-the-art reconstruction performance on ImageNet and UCF benchmarks. We also provide a tokenizer pre-trained on large-scale data, significantly outperforming Cosmos on zero-shot benchmarks (1.93 vs. 0.78 rFID on ImageNet original resolution). Furthermore, we explore its application in plain auto-regressive models to validate scalability properties, producing a family of auto-regressive image generation models ranging from 300M to 1.5B. To assist auto-regressive models in predicting with a super-large vocabulary, we factorize it into two sub-vocabulary of different sizes by asymmetric token factorization, and further introduce ``next sub-token prediction'' to enhance sub-token interaction for better generation quality. We release all models and codes to foster innovation and creativity in the field of auto-regressive visual generation.

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

↑