Towards speech enhancement using a variational U-Net architecture
This work offers an incremental improvement in speech enhancement performance for users in noisy and reverberant environments.
This paper explores a variational U-Net architecture for single-channel audio denoising, focusing on direct spectral reconstruction rather than filter mask estimation. The proposed variational U-Net shows an average advantage over its non-variational counterpart, improving PESQ by 0.31 and STOI by 6.98 under reverberant conditions.
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.