TABED: Test-Time Adaptive Ensemble Drafting for Robust Speculative Decoding in LVLMs
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2026
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866914286099496960 |
|---|---|
| author | Lee, Minjae Kang, Wonjun Ahn, Byeongkeun Classen, Christian Galim, Kevin Oh, Seunghyuk Yan, Minghao Koo, Hyung Il Lee, Kangwook |
| author_facet | Lee, Minjae Kang, Wonjun Ahn, Byeongkeun Classen, Christian Galim, Kevin Oh, Seunghyuk Yan, Minghao Koo, Hyung Il Lee, Kangwook |
| contents | Speculative decoding (SD) has proven effective for accelerating LLM inference by quickly generating draft tokens and verifying them in parallel. However, SD remains largely unexplored for Large Vision-Language Models (LVLMs), which extend LLMs to process both image and text prompts. To address this gap, we benchmark existing inference methods with small draft models on 11 datasets across diverse input scenarios and observe scenario-specific performance fluctuations. Motivated by these findings, we propose Test-time Adaptive Batched Ensemble Drafting (TABED), which dynamically ensembles multiple drafts obtained via batch inference by leveraging deviations from past ground truths available in the SD setting. The dynamic ensemble method achieves an average robust walltime speedup of 1.74x over autoregressive decoding and a 5% improvement over single drafting methods, while remaining training-free and keeping ensembling costs negligible through parameter sharing. With its plug-and-play compatibility, we further enhance TABED by integrating advanced verification and alternative drafting methods. Code and custom-trained models are available at https://github.com/furiosa-ai/TABED. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2601_20357 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | TABED: Test-Time Adaptive Ensemble Drafting for Robust Speculative Decoding in LVLMs Lee, Minjae Kang, Wonjun Ahn, Byeongkeun Classen, Christian Galim, Kevin Oh, Seunghyuk Yan, Minghao Koo, Hyung Il Lee, Kangwook Machine Learning Computation and Language Speculative decoding (SD) has proven effective for accelerating LLM inference by quickly generating draft tokens and verifying them in parallel. However, SD remains largely unexplored for Large Vision-Language Models (LVLMs), which extend LLMs to process both image and text prompts. To address this gap, we benchmark existing inference methods with small draft models on 11 datasets across diverse input scenarios and observe scenario-specific performance fluctuations. Motivated by these findings, we propose Test-time Adaptive Batched Ensemble Drafting (TABED), which dynamically ensembles multiple drafts obtained via batch inference by leveraging deviations from past ground truths available in the SD setting. The dynamic ensemble method achieves an average robust walltime speedup of 1.74x over autoregressive decoding and a 5% improvement over single drafting methods, while remaining training-free and keeping ensembling costs negligible through parameter sharing. With its plug-and-play compatibility, we further enhance TABED by integrating advanced verification and alternative drafting methods. Code and custom-trained models are available at https://github.com/furiosa-ai/TABED. |
| title | TABED: Test-Time Adaptive Ensemble Drafting for Robust Speculative Decoding in LVLMs |
| topic | Machine Learning Computation and Language |
| url | https://arxiv.org/abs/2601.20357 |