arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.07548cs.CVcs.ITcs.LGeess.IVmath.IT
收藏

使用二值球面量化的图像与视频分词

Image and Video Tokenization with Binary Spherical Quantization

UT Austin(德克萨斯大学奥斯汀分校);MThreads AI(MThreads人工智能公司)

Yue Zhao, Yuanjun Xiong, Philipp Krähenbühl

AI总结:

提出基于 transformer 与二值球面量化的图像/视频分词器 BSQ-ViT,无需码本、支持可变长视频,在重建、压缩和图像合成任务上达到优异性能。

摘要:

我们提出了一种基于 transformer 的新型图像与视频分词器,采用二值球面量化(Binary Spherical Quantization,BSQ)。BSQ 将高维视觉嵌入投影到低维超球面,然后施加二值量化。BSQ 具有以下特点:(1)参数高效,无需显式码本;(2)可扩展到任意 token 维度;(3)紧凑:能够以最小失真将视觉数据压缩至多 100 倍。我们的分词器使用带有简单分块因果掩码的 transformer 编码器和解码器,以支持可变长度视频作为输入。所得 BSQ-ViT 在图像和视频重建基准上达到最先进的视觉重建质量,并且与此前最佳方法相比吞吐量为其 2.4 倍。此外,通过为自适应算术编码学习自回归先验,BSQ-ViT 在视频压缩上取得了与最先进视频压缩标准相当的结果。BSQ-ViT 还使掩码语言模型能够实现与基于 GAN 和扩散的方法相竞争的图像合成质量。

英文摘要:

We propose a new transformer-based image and video tokenizer with Binary Spherical Quantization (BSQ). BSQ projects the high-dimensional visual embedding to a lower-dimensional hypersphere and then applies binary quantization. BSQ is (1) parameter-efficient without an explicit codebook, (2) scalable to arbitrary token dimensions, and (3) compact: compressing visual data by up to 100$\times$ with minimal distortion. Our tokenizer uses a transformer encoder and decoder with simple block-wise causal masking to support variable-length videos as input. The resulting BSQ-ViT achieves state-of-the-art visual reconstruction quality on image and video reconstruction benchmarks with 2.4$\times$ throughput compared to the best prior methods. Furthermore, by learning an autoregressive prior for adaptive arithmetic coding, BSQ-ViT achieves comparable results on video compression with state-of-the-art video compression standards. BSQ-ViT also enables masked language models to achieve competitive image synthesis quality to GAN- and diffusion-based methods.

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

↑