arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.02884eess.AScs.LGcs.SDeess.SP
收藏

探索用于语音分离的自注意力机制

Exploring Self-Attention Mechanisms for Speech Separation

Université Laval(拉瓦尔大学);Concordia University(康考迪亚大学);Mila-Quebec AI Institute(米拉-魁北克人工智能研究所);Università Politecnica delle Marche(马尔凯理工大学);Université de Sherbrooke(舍布鲁克大学)

Cem Subakan, Mirco Ravanelli, Samuele Cornell, Francois Grondin, Mirko Bronzi

AI总结:

本文深入研究基于 Transformer 的语音分离方法,扩展 SepFormer 至带噪及混响场景,并首次探索 Linformer、Lonformer、ReFormer 等高效自注意力机制,在降低内存需求的同时取得优于 Conv-TasNet 的性能。

摘要:

Transformer 已在深度学习中带来了显著的改进。它们在利用并行处理的同时,在许多任务中往往优于循环模型和卷积模型。最近,我们提出了 SepFormer,它在 WSJ0-2/3 Mix 数据集上取得了语音分离的最先进性能。本文深入研究用于语音分离的 Transformer。特别是,我们通过在更具挑战性的带噪和带噪混响数据集(如 LibriMix、WHAM! 和 WHAMR!)上提供结果,扩展了我们先前关于 SepFormer 的发现。此外,我们扩展了模型以执行语音增强,并提供了去噪和去混响任务的实验证据。最后,我们首次在语音分离中研究了高效自注意力机制的使用,例如 Linformer、Lonformer 和 ReFormer。我们发现它们显著降低了内存需求。例如,我们表明基于 Reformer 的注意力在 WSJ0-2Mix 数据集上优于流行的 Conv-TasNet 模型,同时在推理速度上更快,并且在内存消耗方面具有可比性。

英文摘要:

Transformers have enabled impressive improvements in deep learning. They often outperform recurrent and convolutional models in many tasks while taking advantage of parallel processing. Recently, we proposed the SepFormer, which obtains state-of-the-art performance in speech separation with the WSJ0-2/3 Mix datasets. This paper studies in-depth Transformers for speech separation. In particular, we extend our previous findings on the SepFormer by providing results on more challenging noisy and noisy-reverberant datasets, such as LibriMix, WHAM!, and WHAMR!. Moreover, we extend our model to perform speech enhancement and provide experimental evidence on denoising and dereverberation tasks. Finally, we investigate, for the first time in speech separation, the use of efficient self-attention mechanisms such as Linformers, Lonformers, and ReFormers. We found that they reduce memory requirements significantly. For example, we show that the Reformer-based attention outperforms the popular Conv-TasNet model on the WSJ0-2Mix dataset while being faster at inference and comparable in terms of memory consumption.

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

↑