Transformer编码器可内嵌对齐功能,实现更高效语音识别
Aligner-Encoders: Self-Attention Transformers Can Be Self-Transducers
- 让Transformer编码器在前向传播中完成音文对齐,无需解码阶段处理
- 推理速度比RNN-T快2倍,比AED快16倍,性能接近顶尖水平
- 自注意力权重清晰展现对齐过程,适合追求高效语音识别的开发者
当前自动语音识别系统(如RNN-Transducer和基于注意力的编码器-解码器)设计中,编码器不改变音频序列的时间位置信息,对齐由解码阶段完成。我们发现,近年采用的基于Transformer的编码器实际上可在前向传播过程中内部完成对齐。这一新现象催生了更简单高效的模型——Aligner-Encoder。训练时,我们舍弃RNN-T的动态规划,改用AED的帧级交叉熵损失;解码器则采用RNN-T的轻量文本递归结构,无学习的交叉注意力,仅按顺序扫描嵌入帧生成词元直至结束。实验表明,该模型性能接近当前最优,尤其在长语音识别配置下表现突出。在典型对比中,总推理时间仅为RNN-T的1/2、AED的1/16。此外,特定层的自注意力权重清晰显示音文对齐,可视为实现‘自转导’。
原文摘要 · Abstract (English)
Modern systems for automatic speech recognition, including the RNN-Transducer and Attention-based Encoder-Decoder (AED), are designed so that the encoder is not required to alter the time-position of information from the audio sequence into the embedding; alignment to the final text output is processed during decoding. We discover that the transformer-based encoder adopted in recent years is actually capable of performing the alignment internally during the forward pass, prior to decoding. This new phenomenon enables a simpler and more efficient model, the "Aligner-Encoder". To train it, we discard the dynamic programming of RNN-T in favor of the frame-wise cross-entropy loss of AED, while the decoder employs the lighter text-only recurrence of RNN-T without learned cross-attention -- it simply scans embedding frames in order from the beginning, producing one token each until predicting the end-of-message. We conduct experiments demonstrating performance remarkably close to the state of the art, including a special inference configuration enabling long-form recognition. In a representative comparison, we measure the total inference time for our model to be 2x faster than RNN-T and 16x faster than AED. Lastly, we find that the audio-text alignment is clearly visible in the self-attention weights of a certain layer, which could be said to perform "self-transduction".
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。