跳到正文
Hugging Face Blog·· 2023-12-20精选AI 评分62

用投机解码让 Whisper 推理提速 2 倍

Speculative Decoding for 2x Faster Whisper Inference

AI 导读

Hugging Face 博客介绍如何用投机解码将 Whisper 推理速度提升约 2 倍,且数学上保证输出与单独使用主模型完全一致。在 LibriSpeech 基准上,Whisper large-v2 的 73 个样本推理时间从 73 秒降至 33 秒,WER 保持 3.5%;荷兰语 VoxPopuli 上从 117 秒降至 62 秒,WER 保持 12.8%。

推荐理由

原文给出可复用的投机解码实现与基准数据,读者可据此在现有 Whisper 流程中获得约 2 倍加速。

来源:Hugging Face Blog · huggingface.co