A Closer Look at Wav2Vec2 Embeddings for On-Device Single-Channel Speech Enhancement

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Shankar, Ravi, Tan, Ke, Xu, Buye, Kumar, Anurag
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913251766304768
author Shankar, Ravi
Tan, Ke
Xu, Buye
Kumar, Anurag
author_facet Shankar, Ravi
Tan, Ke
Xu, Buye
Kumar, Anurag
contents Self-supervised learned models have been found to be very effective for certain speech tasks such as automatic speech recognition, speaker identification, keyword spotting and others. While the features are undeniably useful in speech recognition and associated tasks, their utility in speech enhancement systems is yet to be firmly established, and perhaps not properly understood. In this paper, we investigate the uses of SSL representations for single-channel speech enhancement in challenging conditions and find that they add very little value for the enhancement task. Our constraints are designed around on-device real-time speech enhancement -- model is causal, the compute footprint is small. Additionally, we focus on low SNR conditions where such models struggle to provide good enhancement. In order to systematically examine how SSL representations impact performance of such enhancement models, we propose a variety of techniques to utilize these embeddings which include different forms of knowledge-distillation and pre-training.
format Preprint
id arxiv_https___arxiv_org_abs_2403_01369
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle A Closer Look at Wav2Vec2 Embeddings for On-Device Single-Channel Speech Enhancement
Shankar, Ravi
Tan, Ke
Xu, Buye
Kumar, Anurag
Audio and Speech Processing
Artificial Intelligence
Machine Learning
Self-supervised learned models have been found to be very effective for certain speech tasks such as automatic speech recognition, speaker identification, keyword spotting and others. While the features are undeniably useful in speech recognition and associated tasks, their utility in speech enhancement systems is yet to be firmly established, and perhaps not properly understood. In this paper, we investigate the uses of SSL representations for single-channel speech enhancement in challenging conditions and find that they add very little value for the enhancement task. Our constraints are designed around on-device real-time speech enhancement -- model is causal, the compute footprint is small. Additionally, we focus on low SNR conditions where such models struggle to provide good enhancement. In order to systematically examine how SSL representations impact performance of such enhancement models, we propose a variety of techniques to utilize these embeddings which include different forms of knowledge-distillation and pre-training.
title A Closer Look at Wav2Vec2 Embeddings for On-Device Single-Channel Speech Enhancement
topic Audio and Speech Processing
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2403.01369