arXiv AI

SpecFit: Not All Draft Tokens Cost the Same in Offloaded MoE Inference

Hugging Face Trending Papers
Jul 27

DraftExpert: Expansion-Aware Self-Speculative Decoding for End-Device MoE Inference

Large Mixture-of-Experts (MoE) language models are attractive for end-device deployment because only a small subset of experts is active per token, but their routed expert weights often exceed accelerator memory. We target latency-critical single-user settings where routed experts are staged on demand from CPU memory to a GPU or from Flash to a mobile NPU.

arXiv Machine Learning
Oct 7

APEX: Speculate smarter, not deeper

The paper introduces APEX, a learned controller that improves speculative decoding for large language models by dynamically selecting the best speculation strategy and adjusting draft depth during generation. APEX uses request‑level routing among EAGLE‑3, n‑gram, and draft‑model speculation, and block‑level depth adaptation based on causal decoding signals and verifier feedback. Integrated into vLLM and tested with Qwen3‑8B, APEX achieves up to 5.24× speedup over autoregressive decoding while reducing wasted tokens by 41% compared to fixed‑depth n‑gram speculation.

By Manvi Jha, Zach Zhang, Zhichao Xu, Linbo Liu, Sai Muralidhar Jayanthi, Vinayak Arannil
arXiv Machine Learning
Oct 3

CAST: Cost-Aware Speculative Trees from One-Pass Block Drafters

CAST (Cost‑Aware Speculative Trees) is a method that improves speculative decoding for large language models by packing multiple drafted token candidates into a tree and verifying the entire tree in a single target‑model pass, rather than only the top‑scoring chain. The tree width is adaptively chosen based on a latency measurement, ensuring that each added candidate’s expected gain outweighs its verification cost. Experiments across five domains, three GPU generations, and two model families show that CAST can be up to 43 % faster than the standard chain, while preserving the target model’s output distribution under both greedy and sampled decoding.

By Jungseob Lee, Sugyeong Eo
arXiv Machine Learning
4d ago

HyCo: Hybrid Drafting for Cost-Efficient Mobile-Cloud Speculative Decoding

HyCo is a hybrid drafting method that assigns depth to a server model’s multi‑token prediction head for execution on a mobile NPU, while alternative branches are handled by a small language model in a single pass. By returning only the hidden state, input embedding, and key‑value entries, HyCo reduces server GPU time per token by 38.6% on Spec‑Bench and 29.2% on 32K‑token summarization, and achieves faster delivery and higher savings than an SLM‑only drafter. The approach also incorporates quantization, a restricted vocabulary, a 1K‑token window, and an ε‑threshold rule to relax exact sampling distribution preservation.

By Hanbin Bae, Heejin Choi, Joohyung Lee, Byoung Jin Choi, Sangjun Park, Hoon-Young Cho