基于CNN-NAFNet级联网络的低信噪比DOA估计方法研究

Estimating DOA Using a Cascaded CNN-NAFNet Architecture in Environments with Low SNR

  • 摘要: 低信噪比(Signal-to-Noise Ratio, SNR)环境下的波达方向(Direction of Arrival, DOA)估计一直是阵列信号处理领域的挑战性问题,阵列信号处理中的协方差矩阵包含了信号的空间相关性和噪声分布信息,它可以被视作一个特殊的“图像”,而无论是图像去噪还是低信噪比DOA估计,核心任务都是从混杂的观测数据中提取出有用的结构化信息。现有的卷积神经网络(Convolutional Neural Network, CNN)等深度学习方法虽然能够提取协方差矩阵的空间特征,但在强噪声干扰下容易丢失细节,导致估计性能下降。受图像去噪领域前沿成果NAFNet(Nonlinear Activation Free Network)启发,本文提出了一种基于CNN-NAFNet级联混合架构的深度学习网络,通过“粗特征提取——精细化增强——全局聚合”三阶段策略来解决低信噪比下的DOA估计难题。首先,网络采用多尺度卷积对输入协方差矩阵数据进行粗粒度特征提取,通过逐层降采样和通道递进式扩张,实现从局部细节到全局整体的多尺度特征聚合,形成对信号空间的初步鲁棒表示;其次,设计NAFNet增强模块对粗特征进行精细化处理,利用SimpleGate机制和简化通道注意力(Simplified Channel Attention, SCA)机制有效抑制噪声干扰,设置多处可学习残差缩放因子,用于动态调节残差连接贡献度,防止深层网络中的梯度消失问题;最后,经过全局特征融合和全连接分类器,输出目标角度的概率分布,实现端到端的DOA估计。仿真结果表明,在低信噪比情况下,所提方法的估计性能优于现有方法,具有一定的鲁棒性和泛化性,同时可满足实际阵列信号处理系统的实时性要求。

     

    Abstract: Estimating the direction of arrival (DOA) of a given object in environments with a low signal-to-noise ratio (SNR) remains a challenging problem in array signal processing. The covariance matrix contains information about the spatial correlation of signals and the distribution of noise and can thus be regarded as a special type of “image”. Both image denoising and low-SNR DOA estimation share the core task of extracting useful structured information from contaminated data. Although existing deep learning methods such as convolutional neural network (CNN) models can extract spatial features from the covariance matrix, they often lose fine details in conditions with strong noise interference, which leads to a notable degradation in performance. To address these challenges in estimating DOA in environments with low SNR, a novel deep learning network based on a CNN-NAFNet cascaded architecture is proposed. The proposed approach was inspired by advances in image denoising, especially the nonlinear activation-free network (NAFNet), and adopts a three-stage strategy including coarse feature extraction, refined enhancement, and global aggregation. First, a multi-scale CNN is applied to perform coarse-grained feature extraction from the input covariance matrix. Multi-scale features are aggregated from local details to global context to form a robust preliminary representation through progressive downsampling and channel expansion. Subsequently, these coarse features are refined using an NAFNet enhancement module designed for the purpose, which suppresses noise interference by leveraging SimpleGate and simplified channel attention (SCA) mechanisms. Multiple learnable residual scaling factors are incorporated to dynamically adjust the contribution of residual connections and mitigate gradient vanishing. Finally, global feature fusion is performed and a fully connected classifier outputs the probability distribution of target angles to provide an end-to-end estimation of the DOA of a given object or target. The simulation results demonstrate that the proposed method achieved superior estimation performance under low-SNR conditions compared to existing approaches. Furthermore, they also show that the method exhibited excellent robustness and generalization capability while meeting the real-time requirements of practical array signal processing systems.

     

/

返回文章
返回