Publication Details
Comparison of wav2vec 2.0 models on three speech processing tasks
Zajíc Zbyněk, Ing., Ph.D. (WBU in Pilsen)
Šmídl Luboš, Ing., Ph.D. (WBU in Pilsen)
Karafiát Martin, Ing., Ph.D. (DCGM FIT BUT)
Speaker change detection Voice activity detection Overlapped speech detection Wav2vec 2.0
The current state-of-the-art for various speech processing problems is a sequence-to-sequence model based on a self-attention mechanism known as transformer. The widely used wav2vec 2.0 is a self-supervised transformer model pre-trained on large amounts of unlabeled speech and then fine-tuned for a specific task. The data used for training and fine-tuning, along with the size of the transformer model, play a crucial role in both of these training steps. The most commonly used wav2vec 2.0 models are trained on relatively "clean" data from sources such as the LibriSpeech dataset, but we can expect there to be a benefit in using more realistic data gathered from a variety of acoustic conditions. However, it is not entirely clear how big the difference would be. Investigating this is the main goal of our article. To this end, we utilize wav2vec 2.0 models in three fundamental speech processing tasks: speaker change detection, voice activity detection, and overlapped speech detection, and test them on four real conversation datasets. We compare four wav2vec 2.0 models with different sizes and different data used for pre-training, and we fine-tune them either on in-domain data from the same dataset or on artificial training data created from the LibriSpeech corpus. Our results suggest that richer data that are more similar to the task domain bring better performance than a larger model.
@ARTICLE{FITPUB13316, author = "Marie Kune\v{s}ov\'{a} and Zbyn\v{e}k Zaj\'{i}c and Lubo\v{s} \v{S}m\'{i}dl and Martin Karafi\'{a}t", title = "Comparison of wav2vec 2.0 models on three speech processing tasks", pages = "1--13", journal = "International Journal of Speech Technology", volume = 27, number = 4, year = 2024, ISSN = "1572-8110", doi = "10.1007/s10772-024-10140-6", language = "english", url = "https://www.fit.vut.cz/research/publication/13316" }