TTS-VAR: A Test-Time Scaling Framework for Visual Auto-Regressive Generation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Chen, Zhekai, Chu, Ruihang, Chen, Yukang, Zhang, Shiwei, Wei, Yujie, Zhang, Yingya, Liu, Xihui
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911200838680576
author Chen, Zhekai
Chu, Ruihang
Chen, Yukang
Zhang, Shiwei
Wei, Yujie
Zhang, Yingya
Liu, Xihui
author_facet Chen, Zhekai
Chu, Ruihang
Chen, Yukang
Zhang, Shiwei
Wei, Yujie
Zhang, Yingya
Liu, Xihui
contents Scaling visual generation models is essential for real-world content creation, yet requires substantial training and computational expenses. Alternatively, test-time scaling has garnered growing attention due to resource efficiency and promising performance. In this work, we present TTS-VAR, the first general test-time scaling framework for visual auto-regressive (VAR) models, modeling the generation process as a path searching problem. To dynamically balance computational efficiency with exploration capacity, we first introduce an adaptive descending batch size schedule throughout the causal generation process. Besides, inspired by VAR's hierarchical coarse-to-fine multi-scale generation, our framework integrates two key components: (i) At coarse scales, we observe that generated tokens are hard for evaluation, possibly leading to erroneous acceptance of inferior samples or rejection of superior samples. Noticing that the coarse scales contain sufficient structural information, we propose clustering-based diversity search. It preserves structural variety through semantic feature clustering, enabling later selection on samples with higher potential. (ii) In fine scales, resampling-based potential selection prioritizes promising candidates using potential scores, which are defined as reward functions incorporating multi-scale generation history. Experiments on the powerful VAR model Infinity show a notable 8.7% GenEval score improvement (from 0.69 to 0.75). Key insights reveal that early-stage structural features effectively influence final quality, and resampling efficacy varies across generation scales. Code is available at https://github.com/ali-vilab/TTS-VAR.
format Preprint
id arxiv_https___arxiv_org_abs_2507_18537
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle TTS-VAR: A Test-Time Scaling Framework for Visual Auto-Regressive Generation
Chen, Zhekai
Chu, Ruihang
Chen, Yukang
Zhang, Shiwei
Wei, Yujie
Zhang, Yingya
Liu, Xihui
Computer Vision and Pattern Recognition
Scaling visual generation models is essential for real-world content creation, yet requires substantial training and computational expenses. Alternatively, test-time scaling has garnered growing attention due to resource efficiency and promising performance. In this work, we present TTS-VAR, the first general test-time scaling framework for visual auto-regressive (VAR) models, modeling the generation process as a path searching problem. To dynamically balance computational efficiency with exploration capacity, we first introduce an adaptive descending batch size schedule throughout the causal generation process. Besides, inspired by VAR's hierarchical coarse-to-fine multi-scale generation, our framework integrates two key components: (i) At coarse scales, we observe that generated tokens are hard for evaluation, possibly leading to erroneous acceptance of inferior samples or rejection of superior samples. Noticing that the coarse scales contain sufficient structural information, we propose clustering-based diversity search. It preserves structural variety through semantic feature clustering, enabling later selection on samples with higher potential. (ii) In fine scales, resampling-based potential selection prioritizes promising candidates using potential scores, which are defined as reward functions incorporating multi-scale generation history. Experiments on the powerful VAR model Infinity show a notable 8.7% GenEval score improvement (from 0.69 to 0.75). Key insights reveal that early-stage structural features effectively influence final quality, and resampling efficacy varies across generation scales. Code is available at https://github.com/ali-vilab/TTS-VAR.
title TTS-VAR: A Test-Time Scaling Framework for Visual Auto-Regressive Generation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2507.18537