CASCADE: Context-Aware Relaxation for Speculative Image Decoding

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Yildirim, Selin, Chowdhury, Subhajit Dutta, Kamani, Mohammad Mahdi, Appia, Vikram, Chen, Deming
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911661545226240
author Yildirim, Selin
Chowdhury, Subhajit Dutta
Kamani, Mohammad Mahdi
Appia, Vikram
Chen, Deming
author_facet Yildirim, Selin
Chowdhury, Subhajit Dutta
Kamani, Mohammad Mahdi
Appia, Vikram
Chen, Deming
contents Autoregressive generation is a powerful approach for high-fidelity image synthesis, but it remains computationally demanding and slow even on the most advanced accelerators. While speculative decoding has been explored to mitigate this bottleneck, existing approaches fail to achieve efficiency gains comparable to those observed in text generation. A key limitation is the target model's high uncertainty during image generation, which leads to high draft token rejection rates. In this work, we identify previously overlooked patterns in the target model's behavior that emerge naturally in tree-based speculative decoding. Specifically, we formalize two properties, semantic interchangeability and convergence, arising from the redundancies in the target model's hidden state representations. By capturing these redundancies across the depth and breadth of the predicted token tree, our method identifies principled opportunities for acceptance relaxation without requiring additional training. Additionally, we enhance standalone drafter performance by injecting the redundancy signals from the target model into drafter training with minimal modification. We evaluate our approach across multiple text-to-image models and drafter architectures. Results show that CASCADE achieves state-of-the-art speedups for drafter-based speculative decoding, with up to 3.6x acceleration, while maintaining image quality and text-prompt fidelity.
format Preprint
id arxiv_https___arxiv_org_abs_2605_07230
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle CASCADE: Context-Aware Relaxation for Speculative Image Decoding
Yildirim, Selin
Chowdhury, Subhajit Dutta
Kamani, Mohammad Mahdi
Appia, Vikram
Chen, Deming
Computer Vision and Pattern Recognition
Artificial Intelligence
Autoregressive generation is a powerful approach for high-fidelity image synthesis, but it remains computationally demanding and slow even on the most advanced accelerators. While speculative decoding has been explored to mitigate this bottleneck, existing approaches fail to achieve efficiency gains comparable to those observed in text generation. A key limitation is the target model's high uncertainty during image generation, which leads to high draft token rejection rates. In this work, we identify previously overlooked patterns in the target model's behavior that emerge naturally in tree-based speculative decoding. Specifically, we formalize two properties, semantic interchangeability and convergence, arising from the redundancies in the target model's hidden state representations. By capturing these redundancies across the depth and breadth of the predicted token tree, our method identifies principled opportunities for acceptance relaxation without requiring additional training. Additionally, we enhance standalone drafter performance by injecting the redundancy signals from the target model into drafter training with minimal modification. We evaluate our approach across multiple text-to-image models and drafter architectures. Results show that CASCADE achieves state-of-the-art speedups for drafter-based speculative decoding, with up to 3.6x acceleration, while maintaining image quality and text-prompt fidelity.
title CASCADE: Context-Aware Relaxation for Speculative Image Decoding
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2605.07230