Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

ChannelTok: Efficient Flexible-Length Vision Tokenization

About

Leading flexible vision tokenizers achieve SOTA quality at an extreme cost, relying on parameter-heavy backbones and slow, multi-step generative decoders. We depart from this complex, spatial-token paradigm and introduce a simple, lightweight, and fast channel-wise flexible-length tokenizer. Our method treats each latent channel as a visual token, enabling a parameter-efficient CNN-Transformer hybrid backbone. Furthermore, employing a stochastic tail-dropping paradigm during training naturally forces channels to organize by semantic importance. This allows for flexible compression at inference by simply retaining the first $k$ channels, and naturally enables variable-length autoregressive image generation. We validate our approach through extensive experiments on ImageNet, demonstrating consistent quality across diverse token budgets. The results establish a new quality-efficiency frontier: our model achieves state-of-the-art perceptual quality (rFID 2.92) while being $8.6\times$ faster in decoding and $2.1\times$ smaller (159M params) than the next-best alternative. Our work establishes channel-wise tokenization as a powerful and practical paradigm for efficient visual representation. Project page: https://channeltok.github.io

Sukriti Paul, Arpit Bansal, Tom Goldstein• 2026

Related benchmarks

TaskDatasetResultRank
Image ReconstructionImageNet1K (val)--
124
Image ReconstructionImageNet-1K 1.0 (val)
rFID2.92
42
Showing 2 of 2 rows

Other info

Follow for update