Soft Tail-dropping for Adaptive Visual Tokenization

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Chen, Zeyuan, Zhang, Kai, Tu, Zhuowen, Xiong, Yuanjun
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917212666724352
author Chen, Zeyuan
Zhang, Kai
Tu, Zhuowen
Xiong, Yuanjun
author_facet Chen, Zeyuan
Zhang, Kai
Tu, Zhuowen
Xiong, Yuanjun
contents We present Soft Tail-dropping Adaptive Tokenizer (STAT), a 1D discrete visual tokenizer that adaptively chooses the number of output tokens per image according to its structural complexity and level of detail. STAT encodes an image into a sequence of discrete codes together with per-token keep probabilities. Beyond standard autoencoder objectives, we regularize these keep probabilities to be monotonically decreasing along the sequence and explicitly align their distribution with an image-level complexity measure. As a result, STAT produces length-adaptive 1D visual tokens that are naturally compatible with causal 1D autoregressive (AR) visual generative models. On ImageNet-1k, equipping vanilla causal AR models with STAT yields competitive or superior visual generation quality compared to other probabilistic model families, while also exhibiting favorable scaling behavior that has been elusive in prior vanilla AR visual generation attempts.
format Preprint
id arxiv_https___arxiv_org_abs_2601_14246
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Soft Tail-dropping for Adaptive Visual Tokenization
Chen, Zeyuan
Zhang, Kai
Tu, Zhuowen
Xiong, Yuanjun
Computer Vision and Pattern Recognition
We present Soft Tail-dropping Adaptive Tokenizer (STAT), a 1D discrete visual tokenizer that adaptively chooses the number of output tokens per image according to its structural complexity and level of detail. STAT encodes an image into a sequence of discrete codes together with per-token keep probabilities. Beyond standard autoencoder objectives, we regularize these keep probabilities to be monotonically decreasing along the sequence and explicitly align their distribution with an image-level complexity measure. As a result, STAT produces length-adaptive 1D visual tokens that are naturally compatible with causal 1D autoregressive (AR) visual generative models. On ImageNet-1k, equipping vanilla causal AR models with STAT yields competitive or superior visual generation quality compared to other probabilistic model families, while also exhibiting favorable scaling behavior that has been elusive in prior vanilla AR visual generation attempts.
title Soft Tail-dropping for Adaptive Visual Tokenization
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2601.14246