基于信息提炼与残差特征聚合网络的单通道语音增强

Single-channel Speech Enhancement Method Based on Hierarchical Refinement and Residual Feature Aggregation Network

  • 摘要: 针对语音增强的深层神经网络中对丰富的全局语音相关信息提取困难、未充分利用中间层次特征的问题,本文以尽可能小的参数为前提,基于注意力U型网络,设计了一种基于信息提炼和残差特征聚合的新型卷积编解码网络来进行语音增强。本文在编解码部分提出一种2维的层次细化残差(HRR,Hierarchical Refinement Residual)模块,该模块能显著降低训练参数并扩大感受野,对多尺度上下文信息进行不同层次的提取;传输层提出一种轻量级的1维通道自适应注意力(1D-CAA,One-Dimensional Channel Dimension Adaptive Attention)模块,结合门控机制和范数归一化,选择性地传递特征并提高网络表达能力,并联合门控残差线性单元搭建了一种门控残差特征聚合(GRFA,Gating Residual Feature Aggregation)网络,增强了层间信息流动并充分利用中间层次特征细节,获取更多时序相关信息。实验部分,本文在21种噪声环境下训练和测试,最终以1.23×106的参数相比于其他方法取得更优的客观与主观指标,具备较强的增强效果与泛化能力,并在模型复杂度与精度上取得良好平衡。

     

    Abstract: ‍ ‍To address the problem of difficult extraction of rich global speech-related information and underutilization of intermediate level features in deep neural networks for speech enhancement, this paper designed a novel convolutional codec network based on information refinement and aggregation of residual features for speech enhancement based on attention U-Net with the smallest possible parameters. The mentioned network proposed a Two-Dimensional Hierarchical Refined Residual (HRR) module in the codec part, which could significantly reduce the training parameters and expanded the perceptual field to extract multi-scale contextual information at different levels; A lightweight One-Dimensional Channel Dimension Adaptive Attention (1D-CAA) module was proposed in the transmission layer, combining gating mechanism and parametric normalization to selectively deliver features and improve network expression capability, and a Gating Residual Feature Aggregation (GRFA) network was built jointly with gating residual linear units to enhance inter-layer information flow and make full use of intermediate level feature details. Residual feature aggregation network, which enhanced the information flow between layers and made full use of the intermediate level feature details to obtain more time-series relevant information. In the experimental part, this paper was trained and tested in 21 noisy environments, and finally achieved better objective and subjective indexes with 1.23×106 parameters compared with other methods, with strong enhancement effect and generalization ability, and a good balance of model complexity and accuracy.

     

/

返回文章
返回