English

TABED: Test-Time Adaptive Ensemble Drafting for Robust Speculative Decoding in LVLMs

Machine Learning 2026-01-29 v1 Computation and Language

Abstract

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.

Keywords

Cite

@article{arxiv.2601.20357,
  title  = {TABED: Test-Time Adaptive Ensemble Drafting for Robust Speculative Decoding in LVLMs},
  author = {Minjae Lee and Wonjun Kang and Byeongkeun Ahn and Christian Classen and Kevin Galim and Seunghyuk Oh and Minghao Yan and Hyung Il Koo and Kangwook Lee},
  journal= {arXiv preprint arXiv:2601.20357},
  year   = {2026}
}

Comments

Accepted to Findings of EACL 2026

R2 v1 2026-07-01T09:23:28.291Z