用微积分方法确定端到端语音识别最佳词表大小。
A Calculus-Based Framework for Determining Vocabulary Size in End-to-End ASR

- 通过拟合训练数据曲线,用导数判断最优词表大小。
- 在LibriSpeech上实验显示,优化词表可提升识别性能。
- 适合研究或调参时需要科学设定词表大小的开发者。
在混合式语音识别系统中,词表规模明确,通常由音素、双音素或三音素数量决定。而在端到端语音识别系统中,词表(称为token)由训练文本语料生成。词表大小是训练中的关键超参数。如BPE、WordPiece和Unigram语言模型等分词算法均以词表大小为输入,生成训练用子词。主流工具包ESPNet在训练脚本中固定词表大小,但文献中缺乏对这些值如何选定的讨论。近期工作[1]提出一种成本函数框架,将分词过程视为黑箱,用于识别适配端到端语音识别的词表大小。本文在此基础上,通过曲线拟合训练数据,并运用微积分的一阶与二阶导数检验,正式推导词表大小超参数。我们在标准LibriSpeech语料上验证该方法,结果表明,合理选择词表大小能显著提升语音识别性能。本文的主要贡献在于形式化了一种确定端到端语音识别最优词表大小的方法。
原文摘要 · Abstract (English)
In hybrid automatic speech recognition (ASR) systems, the vocabulary size is unambiguous, typically determined by the number of phones, bi-phones, or tri-phones present in the language. In contrast, end-to-end ASR systems derive their vocabulary, often referred to as tokens from the text corpus used for training. The choice and, more importantly, the size of this vocabulary is a critical hyper-parameter in training end-to-end ASR systems. Tokenization algorithms such as Byte Pair Encoding (BPE), WordPiece, and Unigram Language Model (ULM) use the vocabulary size as an input hyper-parameter to generate the sub-words employed during ASR training. Popular toolkits like ESPNet provide a fixed vocabulary size in their training recipes, but there is little documentation or discussion in the literature regarding how these values are determined. Recent work [1] has formalized an approach to identify the vocabulary size best suited for end-to-end ASR, introducing a cost function framework that treats the tokenization process as a black box. In this paper, we build upon that foundation by curve fitting the training data and using the principle of first and second derivative tests in calculus to formally estimate the vocabulary size hyper-parameter. We demonstrate the utility and usefulness of our approach by applying it on a standard Librispeech corpus and show that the optimal choice of vocabulary size hyper-parameter improves the performance of the ASR. The main contribution of this paper in formalizing an approach to identify the vocabulary size best suited for training an end-to-end ASR system.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。