English

Task-Centric Acceleration of Small-Language Models

Computation and Language 2026-03-02 v1 Artificial Intelligence Information Theory math.IT

Abstract

Small language models (SLMs) have emerged as efficient alternatives to large language models for task-specific applications. However, they are often employed in high-volume, low-latency settings, where efficiency is crucial. We propose TASC, Task-Adaptive Sequence Compression, a framework for SLM acceleration comprising two use-cases: When performing SLM fine-tuning, we propose TASC-ft, which iteratively enriches the tokenizer vocabulary with high-frequency output n-grams and then fine-tunes the model to utilize the expanded vocabulary. Next, we propose an inference-time method, termed TASC-spec. TASC-spec is a lightweight, training-free speculative decoding method that constructs an n-gram draft model from the task's output corpus, mixing task and context n-gram information.TASC-spec avoids any additional training, while bypassing draft-target vocabulary alignment constraints. We demonstrate the effectiveness of both methods across multiple low output-variability generation tasks. Our methods show consistent improvements in inference efficiency while maintaining task performance.

Keywords

Cite

@article{arxiv.2602.24174,
  title  = {Task-Centric Acceleration of Small-Language Models},
  author = {Dor Tsur and Sharon Adar and Ran Levy},
  journal= {arXiv preprint arXiv:2602.24174},
  year   = {2026}
}
R2 v1 2026-07-01T10:55:51.947Z