arXiv:2411.00664eess.AScs.CL2024-11被引 5

用向量量化加速语音识别上下文检索,大幅降低计算开销。

Optimizing Contextual Speech Recognition Using Vector Quantization for Efficient Retrieval

  • 通过向量量化实现高效检索,替代传统注意力机制
  • 支持数千条上下文条目,个人实体识别错误率降71%
  • 相比标准方法,计算时间减20%,内存降85%-95%

神经上下文偏置使语音识别模型能利用相关上下文信息,提升转录准确率。但传统方法依赖音频与偏置条目间的交叉注意力模块,导致计算复杂度高,限制了偏置库规模和准确率提升。本文提出一种基于向量量化的交叉注意力近似方法,实现计算与内存高效的大型偏置库使用。将该技术与基于检索的上下文偏置结合:先用量化检索模块根据音频定位候选偏置条目,再用于模型偏置。该方法不依赖具体偏置方式,可兼容完整交叉注意力、大语言模型提示及二者组合。实验表明,基于检索的短名单策略使系统能有效利用数千条偏置条目,个人实体识别相对错误率降低最高达71%;同时,在百万级条目下,相比标准点积交叉注意力,计算时间减少20%,内存占用降低85%-95%。

原文摘要 · Abstract (English)

Neural contextual biasing allows speech recognition models to leverage contextually relevant information, leading to improved transcription accuracy. However, the biasing mechanism is typically based on a cross-attention module between the audio and a catalogue of biasing entries, which means computational complexity can pose severe practical limitations on the size of the biasing catalogue and consequently on accuracy improvements. This work proposes an approximation to cross-attention scoring based on vector quantization and enables compute- and memory-efficient use of large biasing catalogues. We propose to use this technique jointly with a retrieval based contextual biasing approach. First, we use an efficient quantized retrieval module to shortlist biasing entries by grounding them on audio. Then we use retrieved entries for biasing. Since the proposed approach is agnostic to the biasing method, we investigate using full cross-attention, LLM prompting, and a combination of the two. We show that retrieval based shortlisting allows the system to efficiently leverage biasing catalogues of several thousands of entries, resulting in up to 71% relative error rate reduction in personal entity recognition. At the same time, the proposed approximation algorithm reduces compute time by 20% and memory usage by 85-95%, for lists of up to one million entries, when compared to standard dot-product cross-attention.

语音识别向量量化上下文偏置检索加速

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。