arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.00528eess.AS
收藏

使用预训练掩码语言模型的端到端语音识别

End-to-End Speech Recognition with Pre-trained Masked Language Model

Yosuke Higuchi, Tetsuji Ogawa, Tetsunori Kobayashi, Shinji Watanabe

AI总结:

本文提出 BERT-CTC 和 BECTRA 两种模型,将预训练掩码语言模型 BERT 有效融入端到端语音识别,通过迭代细化与适配 ASR 的词汇表弥合文本差异,在多个 ASR 任务上优于 CTC 和 transducer 基线。

摘要:

我们提出了一种新颖的端到端自动语音识别(ASR)方法,该方法利用预训练的掩码语言模型(LMs)来促进语言信息的提取。所提出的模型 BERT-CTC 和 BECTRA 专门设计用于将预训练语言模型(例如 BERT)有效集成到端到端 ASR 模型中。BERT-CTC 通过解决输出标记之间条件独立性假设的约束,使 BERT 适应连接主义时间分类(CTC)。这使得在 ASR 过程中能够显式地以 BERT 的上下文嵌入为条件,并通过迭代细化算法无缝融合音频和语言信息。BECTRA 将 BERT-CTC 扩展到 transducer 框架,并使用适合 ASR 训练的词汇表训练解码器网络。这旨在弥合端到端 ASR 中处理的文本与 BERT 之间的差距,因为这些模型具有不同的词汇表以及不同的文本格式和风格,例如标点符号的存在。在各种 ASR 任务上的实验结果表明,由于引入了 BERT 知识,所提出的模型相对于基于 CTC 和 transducer 的基线均有所改进。此外,我们的深入分析和研究验证了所提出的公式和架构设计的有效性。

英文摘要:

We present a novel approach to end-to-end automatic speech recognition (ASR) that utilizes pre-trained masked language models (LMs) to facilitate the extraction of linguistic information. The proposed models, BERT-CTC and BECTRA, are specifically designed to effectively integrate pre-trained LMs (e.g., BERT) into end-to-end ASR models. BERT-CTC adapts BERT for connectionist temporal classification (CTC) by addressing the constraint of the conditional independence assumption between output tokens. This enables explicit conditioning of BERT's contextualized embeddings in the ASR process, seamlessly merging audio and linguistic information through an iterative refinement algorithm. BECTRA extends BERT-CTC to the transducer framework and trains the decoder network using a vocabulary suitable for ASR training. This aims to bridge the gap between the text processed in end-to-end ASR and BERT, as these models have distinct vocabularies with varying text formats and styles, such as the presence of punctuation. Experimental results on various ASR tasks demonstrate that the proposed models improve over both the CTC and transducer-based baselines, owing to the incorporation of BERT knowledge. Moreover, our in-depth analysis and investigation verify the effectiveness of the proposed formulations and architectural designs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑