TensorX
返回文献探索

Paper · arXiv 2406.16758

Towards Fast Multilingual LLM Inference: Speculative Decoding and Specialized Drafters

Euiin Yi, Taehyeon Kim, Hongseok Jeung, Du-Seong Chang, Se-Young Yun

20 upvotesJune 24, 2024arXiv 预印本
AI 摘要

A speculative decoding approach with language-specific draft models speeds up inference time for large language models in multilingual settings.

large language modelsspeculative decodingdraft modelspretrain-and-finetune strategyGPT-4o evaluation

Abstract

Large language models (LLMs) have revolutionized natural language processing and broadened their applicability across diverse commercial applications. However, the deployment of these models is constrained by high inference time in multilingual settings. To mitigate this challenge, this paper explores a training recipe of an assistant model in speculative decoding, which are leveraged to draft and-then its future tokens are verified by the target LLM. We show that language-specific draft models, optimized through a targeted pretrain-and-finetune strategy, substantially brings a speedup of inference time compared to the previous methods. We validate these models across various languages in inference time, out-of-domain speedup, and GPT-4o evaluation.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号