logo
0
0
WeChat Login
zhuangshaobin<zhuangshaobin@bytedance.com>
Update UniWeTok readme

UniWeTok: An Unified Binary Tokenizer with Codebook Size 2128\mathit{2^{128}} for Unified Multimodal Large Language Model

UniWeTok Paper on arXiv UniWeTok Models

Shaobin Zhuang*, Yuang Ai*, Jiaming Han*, Weijia Mao, Xiaohui Li, Fangyikang Wang, Xiao Wang, Yan Li, Shanchuan Lin, Kun Xu, Zhenheng Yang, Huaibo Huang, Xiangyu Yue, Hao Chen*†‡, Yali Wang†

* Equal Contribution   Corresponding Author   Project Lead

Unified Multimodal Large Language Models (MLLMs) require a visual representation that simultaneously supports high-fidelity reconstruction, complex semantic extraction, and generative suitability. However, existing visual tokenizers typically struggle to satisfy these conflicting objectives within a single framework. In this paper, we introduce UniWeTok, a unified discrete tokenizer designed to bridge this gap using a massive binary codebook (2128\mathit{2^{128}}). For training framework, we introduce Pre-Post Distillation and a Generative-Aware Prior to enhance the semantic extraction and generative prior of the discrete tokens. In terms of model architecture, we propose a convolution-attention hybrid architecture with the SigLu activation function. SigLu activation not only bounds the encoder output and stabilizes the semantic distillation process but also effectively addresses the optimization conflict between token entropy loss and commitment loss. We further propose a three-stage training framework designed to enhance UniWeTok's adaptability cross various image resolutions and perception-sensitive scenarios, such as those involving human faces and textual content. On ImageNet, UniWeTok achieves state-of-the-art image generation performance (FID: UniWeTok 1.38 vs. REPA 1.42) while requiring a remarkably low training compute (Training Tokens: UniWeTok 33B vs. REPA 262B). On general-domain, UniWeTok demonstrates highly competitive capabilities across a broad range of tasks, including multimodal understanding, image generation (DPG Score: UniWeTok 86.63 vs. FLUX.1 [Dev] 83.84), and editing (GEdit Overall Score: UniWeTok 5.09 vs. OmniGen 5.06). We release code and models to facilitate community exploration of unified tokenizer and MLLM.

⚡ Highlights

  • Extreme Compression: Compresses 256x256 images into just 64 tokens (32x downsampling), reducing sequence length by 75% compared to standard tokenizers.
  • Massive Codebook: Uses a binary codebook of size 21282^{128} via Group-Wise Lookup-Free Quantization.
  • Unified Capability: Excels at both Generation (DPG Score 86.63 vs FLUX.1 [Dev] 83.84) and Understanding (Competitive on MMMU, MME).
  • Efficiency: Requires remarkably low training compute (33B tokens vs 262B for comparable methods).

🪪 License

UniWeTok is licensed under the Apache 2.0 license.

📖 Citation

If you find our work useful for your research, please consider citing our paper:

@article{zhuang2026uniwetok, title = {UniWeTok: An Unified Binary Tokenizer with Codebook Size $\mathit{2^{128}}$ for Unified Multimodal Large Language Model}, author = {Zhuang, Shaobin and Ai, Yuang and Han, Jiaming and Mao, Weijia and Li, Xiaohui and Wang, Fangyikang and Wang, Xiao and Li, Yan and Lin, Shanchuan and Xu, Kun and Yang, Zhenheng and Huang, Huaibo and Yue, Xiangyu and Chen, Hao and Wang, Yali}, journal = {arXiv preprint arXiv:2602.14178}, year = {2026} }