arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.03497eess.AScs.AIcs.CLcs.LGcs.SD
收藏

EAT:基于高效音频Transformer的自监督预训练

EAT: Self-Supervised Pre-Training with Efficient Audio Transformer

MoE Key Lab of Artificial Intelligence, AI Institute(教育部人工智能重点实验室,人工智能研究院);X-LANCE Lab, Department of Computer Science and Engineering(X-LANCE实验室,计算机科学与工程系);Shanghai Jiao Tong University(上海交通大学)

Wenxi Chen, Yuzhe Liang, Ziyang Ma, Zhisheng Zheng, Xie Chen

AI总结:

本文提出高效音频Transformer(EAT),通过bootstrap自监督训练范式和Utterance-Frame Objective(UFO)及大型逆块掩码策略,在多项音频任务上实现SOTA性能,并将预训练速度提升约15倍。

摘要:

音频自监督学习(SSL)预训练旨在从无标签音频中学习良好的表示,已取得显著进展。然而,预训练期间的大量计算需求对音频SSL模型的潜在应用和优化构成了重大障碍。本文受data2vec 2.0在图像模态和Audio-MAE在音频模态中成功的启发,引入了Efficient Audio Transformer(EAT),以进一步提升音频SSL的有效性和效率。所提出的EAT将bootstrap自监督训练范式应用于音频领域。设计了一种新颖的Utterance-Frame Objective(UFO)以增强声学事件的建模能力。此外,我们揭示了掩码策略在音频SSL预训练中的关键作用,并发现使用大型逆块掩码可获得更优的音频表示。实验结果表明,EAT在包括AudioSet(AS-2M、AS-20K)、ESC-50和SPC-2在内的一系列音频相关任务上取得了最先进(SOTA)的性能,且与现有音频SSL模型相比,预训练速度最高提升约15倍。

英文摘要:

Audio self-supervised learning (SSL) pre-training, which aims to learn good representations from unlabeled audio, has made remarkable progress. However, the extensive computational demands during pre-training pose a significant barrier to the potential application and optimization of audio SSL models. In this paper, inspired by the success of data2vec 2.0 in image modality and Audio-MAE in audio modality, we introduce Efficient Audio Transformer (EAT) to further improve the effectiveness and efficiency in audio SSL. The proposed EAT adopts the bootstrap self-supervised training paradigm to the audio domain. A novel Utterance-Frame Objective (UFO) is designed to enhance the modeling capability of acoustic events. Furthermore, we reveal that the masking strategy is critical in audio SSL pre-training, and superior audio representations can be obtained with large inverse block masks. Experiment results demonstrate that EAT achieves state-of-the-art (SOTA) performance on a range of audio-related tasks, including AudioSet (AS-2M, AS-20K), ESC-50, and SPC-2, along with a significant pre-training speedup up to ~15x compared to existing audio SSL models.

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

↑