Tokenize Image as a Set
Not on the map. Why?
Abstract (arXiv)
This paper proposes a fundamentally new paradigm for image generation through set-based tokenization and distribution modeling. Unlike conventional methods that serialize images into fixed-position latent codes with a uniform compression ratio, we introduce an unordered token set representation to dynamically allocate coding capacity based on regional semantic complexity. This TokenSet enhances global context aggregation and improves robustness against local perturbations. To address the critical challenge of modeling discrete sets, we devise a dual transformation mechanism that bijectively converts sets into fixed-length integer sequences with summation constraints. Further, we propose Fixed-Sum Discrete Diffusion--the first framework to simultaneously handle discrete values, fixed sequence length, and summation invariance--enabling effective set distribution modeling. Experiments demonstrate our method's superiority in semantic-aware representation and generation quality. Our innovations, spanning novel representation and modeling strategies, advance visual generation beyond traditional sequential token paradigms. Our code and models are publicly available at https://github.com/Gengzigang/TokenSet.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 35 of its 46 references that are in the map. First posted on arXiv 20 Mar 2025.
Builds on
35 of 46 references.
References · 35
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2020End-to-End Object Detection with Transformerscited by 7,414
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2017Neural Discrete Representation Learningcited by 4,712
- 2020Taming Transformers for High-Resolution Image Synthesiscited by 3,253
- 2017Deep Setscited by 1,421
- 2019Generating Diverse High-Fidelity Images with VQ-VAE-2cited by 1,387
- 2022MaskGIT: Masked Generative Image Transformercited by 1,099
23 more
- 2024Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Predictioncited by 1,023
- 2024Autoregressive Model Beats Diffusion: Llama for Scalable Image Generationcited by 777
- 2021Vector Quantized Diffusion Model for Text-to-Image Synthesiscited by 774
- 2022Autoregressive Image Generation using Residual Quantizationcited by 722
- 2021Vector-quantized Image Modeling with Improved VQGANcited by 658
- 2024Autoregressive Image Generation without Vector Quantizationcited by 651
- 2023Finite Scalar Quantization: VQ-VAE Made Simplecited by 440
- 2022Analog Bits: Generating Discrete Data using Diffusion Models with Self-Conditioningcited by 373
- 2024Discrete Flow Matchingcited by 317
- 2023Common Diffusion Noise Schedules and Sample Steps are Flawedcited by 310
- 2024An Image is Worth 32 Tokens for Reconstruction and Generationcited by 302
- 2018Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networkscited by 132
- 2022MoVQ: Modulating Quantized Vectors for High-Fidelity Image Generationcited by 125
- 2016Towards a Neural Statisticiancited by 121
- 2024ImageFolder: Autoregressive Image Generation with Folded Tokenscited by 94
- 2024Addressing Representation Collapse in Vector Quantized Models with One Linear Layercited by 80
- 2018Point Cloud GANcited by 79
- 2019Deep Set Prediction Networkscited by 71
- 2024Scaling the Codebook Size of VQGAN to 100,000 with a Utilization Rate of 99%cited by 68
- 2023Towards Accurate Image Coding: Improved Autoregressive Image Generation with Dynamic Vector Quantizationcited by 60
- 2023Image as Set of Pointscited by 43
- 2023Efficient-VQGAN: Towards High-Resolution Image Generation with Efficient Vision Transformerscited by 33
- 2020Conditional Set Generation with Transformerscited by 32
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Semantic-Aware Prefix Learning for Token-Efficient Image Generation
- 2025Bridging Continuous and Discrete Tokens for Autoregressive Visual Generation
- 2024Language-Guided Image Tokenization for Generation
- 2025Highly Compressed Tokenizer Can Generate Without Training
- 2024Image Understanding Makes for A Good Tokenizer for Image Generation
- 2024Factorized Visual Tokenization and Generation
- 2026AdaTok: Self-Budgeting Image Tokenization with Quality-Preserving Dynamic Tokens
- 2026MergeTok: Unified Continuous and Discrete Visual Tokenization via Token Merging