Towards speech enhancement using a variational U-Net architecture

TL;DR

This paper explores a variational U-Net architecture for speech enhancement, demonstrating improved spectral reconstruction and noise suppression capabilities over traditional methods, especially in reverberant and impulsive noise conditions.

Contribution

It introduces a probabilistic bottleneck into the U-Net for direct spectral reconstruction, showing its effectiveness without filter mask estimation.

Findings

Variational U-Net outperforms classic U-Net in PESQ and STOI scores.

Residual connections are essential for spectral reconstruction.

Improved suppression of impulsive noise sources.

Abstract

We investigate the viability of a variational U-Net architecture for denoising of single-channel audio data. Deep network speech enhancement systems commonly aim to estimate filter masks, or opt to work on the waveform signal, potentially neglecting relationships across higher dimensional spectro-temporal features. We study the adoption of a probabilistic bottleneck into the classic U-Net architecture for direct spectral reconstruction. Evaluation of several ablation network variants is carried out using signal-to-distortion ratio and perceptual measures, on audio data that includes known and unknown noise types as well as reverberation. Our experiments show that the residual (skip) connections in the proposed system are a prerequisite for successful spectral reconstruction, i.e., without filter mask estimation. Results show, on average, an advantage of the proposed variational U-Net architecture over its classic, non-variational version in signal enhancement performance under reverberant conditions of 0.31 and 6.98 in PESQ and STOI scores, respectively. Anecdotal evidence points to improved suppression of impulsive noise sources with the variational U-Net compared to the recurrent mask estimation network baseline.