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

UnitY:基于离散单元的两遍直接语音到语音翻译

UnitY: Two-pass Direct Speech-to-speech Translation with Discrete Units

FAIR(脸书人工智能研究);Meta AI;Carnegie Mellon University(卡内基梅隆大学)

Hirofumi Inaguma, Sravya Popuri, Ilia Kulikov, Peng-Jen Chen, Changhan Wang, Yu-An Chung, Yun Tang, Ann Lee, Shinji Watanabe, Juan Pino

AI总结:

本文提出两遍直接语音到语音翻译架构UnitY,先预测文本再预测离散声学单元,并通过子词预测、解码器设计与预训练提升性能,在ASR-BLEU和解码速度上显著优于单遍模型。

摘要:

直接语音到语音翻译(S2ST)中所有组件均可联合优化,相较于级联方法具有推理速度快、流程简化的优势。我们提出了一种新颖的两遍直接S2ST架构UnitY,它首先生成文本表示,随后预测离散声学单元。我们通过第一遍解码器中的子词预测、先进的两遍解码器架构设计与搜索策略以及更好的训练正则化来提升模型性能。为了利用大量无标注文本数据,我们基于自监督去噪自编码任务预训练第一遍文本解码器。在多种数据规模的基准数据集上的实验评估表明,UnitY比单遍语音到单元翻译模型在ASR-BLEU上高出2.5-4.2,同时解码速度提升2.83倍。我们表明,即使第二遍预测频谱图,所提出的方法也能提升性能。然而,与该情况相比,预测离散单元可实现2.51倍的解码加速。

英文摘要:

Direct speech-to-speech translation (S2ST), in which all components can be optimized jointly, is advantageous over cascaded approaches to achieve fast inference with a simplified pipeline. We present a novel two-pass direct S2ST architecture, UnitY, which first generates textual representations and predicts discrete acoustic units subsequently. We enhance the model performance by subword prediction in the first-pass decoder, advanced two-pass decoder architecture design and search strategy, and better training regularization. To leverage large amounts of unlabeled text data, we pre-train the first-pass text decoder based on the self-supervised denoising auto-encoding task. Experimental evaluations on benchmark datasets at various data scales demonstrate that UnitY outperforms a single-pass speech-to-unit translation model by 2.5-4.2 ASR-BLEU with 2.83x decoding speed-up. We show that the proposed methods boost the performance even when predicting spectrogram in the second pass. However, predicting discrete units achieves 2.51x decoding speed-up compared to that case.

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

↑