English

Fast Inference of Visual Autoregressive Model with Adjacency-Adaptive Dynamical Draft Trees

Computer Vision and Pattern Recognition 2025-12-29 v1

Abstract

Autoregressive (AR) image models achieve diffusion-level quality but suffer from sequential inference, requiring approximately 2,000 steps for a 576x576 image. Speculative decoding with draft trees accelerates LLMs yet underperforms on visual AR models due to spatially varying token prediction difficulty. We identify a key obstacle in applying speculative decoding to visual AR models: inconsistent acceptance rates across draft trees due to varying prediction difficulties in different image regions. We propose Adjacency-Adaptive Dynamical Draft Trees (ADT-Tree), an adjacency-adaptive dynamic draft tree that dynamically adjusts draft tree depth and width by leveraging adjacent token states and prior acceptance rates. ADT-Tree initializes via horizontal adjacency, then refines depth/width via bisectional adaptation, yielding deeper trees in simple regions and wider trees in complex ones. The empirical evaluations on MS-COCO 2017 and PartiPrompts demonstrate that ADT-Tree achieves speedups of 3.13xand 3.05x, respectively. Moreover, it integrates seamlessly with relaxed sampling methods such as LANTERN, enabling further acceleration. Code is available at https://github.com/Haodong-Lei-Ray/ADT-Tree.

Keywords

Cite

@article{arxiv.2512.21857,
  title  = {Fast Inference of Visual Autoregressive Model with Adjacency-Adaptive Dynamical Draft Trees},
  author = {Haodong Lei and Hongsong Wang and Xin Geng and Liang Wang and Pan Zhou},
  journal= {arXiv preprint arXiv:2512.21857},
  year   = {2025}
}
R2 v1 2026-07-01T08:41:12.328Z