arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2311.04534cs.CLcs.SDeess.AS

损失掩蔽在基于离散词元的解码器专用Transformer自动语音识别中并非必需

Loss Masking Is Not Needed in Decoder-only Transformer for Discrete-token-based ASR

Alibaba Group(阿里巴巴集团)

Qian Chen, Wen Wang, Qinglin Zhang, Siqi Zheng, Shiliang Zhang, Chong Deng, Yukun Ma, Hai Yu, Jiaqing Liu, Chong Zhang

AI总结:

针对解码器专用Transformer在离散词元ASR中依赖损失掩蔽的问题,提出平滑标签蒸馏(SLD)方法,以自回归方式建模语音词元,实验证明SLD优于损失掩蔽。

摘要:

近期,诸如SpeechGPT、VioLA和AudioPaLM等统一语音-文本模型在各种语音任务上取得了显著性能。这些模型将语音信号离散化为词元(语音离散化),并为文本和语音词元共享词汇表,然后在混合语音任务上训练单一的解码器专用Transformer。然而,这些模型在自动语音识别(ASR)任务中依赖损失掩蔽策略,该策略忽略了语音词元之间的依赖关系。本文提出以自回归方式对语音词元建模,类似于文本处理。我们发现,对输入语音词元应用传统交叉熵损失并不总能比损失掩蔽方法持续提升ASR性能。为解决此问题,我们提出一种新方法,称为平滑标签蒸馏(SLD),该方法在语音词元上应用带平滑标签的KL散度损失。实验表明,SLD能有效建模语音词元,并在不同语音离散化方法的ASR任务中,对解码器专用Transformer优于损失掩蔽。源代码可在以下网址找到:https://github.com/alibaba-damo-academy/SpokenNLP/tree/main/sld

英文摘要:

Recently, unified speech-text models, such as SpeechGPT, VioLA, and AudioPaLM, have achieved remarkable performance on various speech tasks. These models discretize speech signals into tokens (speech discretization) and use a shared vocabulary for both text and speech tokens. Then they train a single decoder-only Transformer on a mixture of speech tasks. However, these models rely on the Loss Masking strategy for the ASR task, which ignores the dependency among speech tokens. In this paper, we propose to model speech tokens in an autoregressive way, similar to text. We find that applying the conventional cross-entropy loss on input speech tokens does not consistently improve the ASR performance over the Loss Masking approach. To address this issue, we propose a novel approach denoted Smoothed Label Distillation (SLD), which applies a KL divergence loss with smoothed labels on speech tokens. Our experiments show that SLD effectively models speech tokens and outperforms Loss Masking for decoder-only Transformers in ASR tasks with different speech discretization methods. The source code can be found here: https://github.com/alibaba-damo-academy/SpokenNLP/tree/main/sld

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

↑