Speech Enhancement with Fullband-Subband Cross-Attention Network

SUBMITTED TO InterSpeech 2022

Download this project as a .zip file Download this project as a tar.gz file

Speech Enhancement with Fullband-Subband Cross-Attention Network

Jun Chen, Wei Rao, Zilin Wang, Zhiyong Wu, Yannan Wang, Tao Yu, Shidong Shang, Helen Meng

Abstract

FullSubNet has shown its promising performance on speech enhancement by utilizing both fullband and subband information. However, the relationship between fullband and subband in FullSubNet is achieved by simply concatenating the output of fullband model and subband units. It has not considered the interaction between fullband and subband. This paper proposes a fullband-subband cross-attention (FSCA) module to interactively fuse the global and local spectral information and applies it to FullSubNet. This new framework is called as FS-CANet. Moreover, different from FullSubNet, the proposed FS-CANet optimize the fullband extractor by temporal convolutional network (TCN) blocks to further reduce the model size. Experimental results on DNS Challenge - Interspeech 2021 dataset show that the proposed FS-CANet outperforms other state-of-the-art speech enhancement approaches, and demonstrate the effectiveness of fullband-subband cross-attention.

With Reverberation

case 1

case 1  
Noisy
FullSubNet
flowtron_0.0 flowtron_0.0
FS-CANet
Clean
flowtron_0.0 flowtron_0.0

case 2

case 2  
Noisy
FullSubNet
flowtron_0.0 flowtron_0.0
FS-CANet
Clean
flowtron_0.0 flowtron_0.0

case 3

case 3  
Noisy
FullSubNet
flowtron_0.0 flowtron_0.0
FS-CANet
Clean
flowtron_0.0 flowtron_0.0

case 4

case 4  
Noisy
FullSubNet
flowtron_0.0 flowtron_0.0
FS-CANet
Clean
flowtron_0.0 flowtron_0.0

Without Reverberation

case 1

case 1  
Noisy
FullSubNet
flowtron_0.0 flowtron_0.0
FS-CANet
Clean
flowtron_0.0 flowtron_0.0

case 2

case 2  
Noisy
FullSubNet
flowtron_0.0 flowtron_0.0
FS-CANet
Clean
flowtron_0.0 flowtron_0.0

case 3

case 3  
Noisy
FullSubNet
flowtron_0.0 flowtron_0.0
FS-CANet
Clean
flowtron_0.0 flowtron_0.0

case 4

case 4  
Noisy
FullSubNet
flowtron_0.0 flowtron_0.0
FS-CANet
FS-CANet
flowtron_0.0 flowtron_0.0