首页 > AI前沿 > TSS: Target-Side Sparsification for Speculative Decoding in Domain-Specific Large Language Models

TSS: Target-Side Sparsification for Speculative Decoding in Domain-Specific Large Language Models

arXiv自然语言 2026-08-15 15:02 5 阅读 查看原文

Speculative decoding accelerates large language model inference through collaboration between a lightweight draft model and a target verifier.

Existing methods mainly improve the draft side, while the target model is typically kept dense and unchanged.

We show that, under domain-specific inference, full-depth target verification is not always the optimal choice.

Counter-intuitively, skipping selected target layers can reduce verification cost while simultaneously increasing draft acceptance and preserving, or even improving, downstream task performance.

Based on this observation, we propose TSS, a target-side sparsification framework for speculative decoding.

TSS employs an acceptance- and metric-aware breadth search to explore multi-layer skip configurations without imposing a fixed priority between the two objectives.

The selected configurations are stored in a domain-to-configuration mapping and applied by a lightweight skip controller, allowing one complete target model to support multiple sparse verification paths without retraining or permanent parameter pruning.

Experiments on Spec-Bench across multiple domains, model scales, and speculative decoding methods show consistent improvements in draft acceptance and downstream task performance.

In Translation setting, TSS increases the average accept length from 2.70 to 4.53 (+67.8%), improves BLEU from 0.131 to 0.237 (+80.9%), and raises end-to-end throughput from 75.6 to 127.3 tokens/s, corresponding to a 1.68X speedup.