Deep Networks for Direction-of-Arrival Estimation in Low SNR

In this work, we consider direction-of-arrival (DoA) estimation in the presence of extreme noise using Deep Learning (DL). In particular, we introduce a Convolutional Neural Network (CNN) that is trained from mutli-channel data of the true array manifold matrix and is able to predict angular directions using the sample covariance estimate. We model the problem as a multi-label classification task and train a CNN in the low-SNR regime to predict DoAs across all SNRs. The proposed architecture demonstrates enhanced robustness in the presence of noise, and resilience to a small number of snapshots. Moreover, it is able to resolve angles within the grid resolution. Experimental results demonstrate significant performance gains in the low-SNR regime compared to state-of-the-art methods and without the requirement of any parameter tuning. We relax the assumption that the number of sources is known a priori and present a training method, where the CNN learns to infer the number of sources jointly with the DoAs. Simulation results demonstrate that the proposed CNN can accurately estimate off-grid angles in low SNR, while at the same time the number of sources is successfully inferred for a sufficient number of snapshots. Our robust solution can be applied in several fields, ranging from wireless array sensors to acoustic microphones or sonars.

[1]  Peter Gerstoft,et al.  Source localization in an ocean waveguide using supervised machine learning , 2017, The Journal of the Acoustical Society of America.

[2]  Ralph Otto Schmidt,et al.  A signal subspace approach to multiple emitter location and spectral estimation , 1981 .

[3]  Harry L. Van Trees,et al.  Optimum Array Processing: Part IV of Detection, Estimation, and Modulation Theory , 2002 .

[4]  Sharon Gannot,et al.  Machine learning in acoustics: Theory and applications. , 2019, The Journal of the Acoustical Society of America.

[5]  Lihua Xie,et al.  Enhancing Sparsity and Resolution via Reweighted Atomic Norm Minimization , 2014, IEEE Transactions on Signal Processing.

[6]  P. P. Vaidyanathan,et al.  Remarks on the Spatial Smoothing Step in Coarray MUSIC , 2015, IEEE Signal Processing Letters.

[7]  M. Haardt,et al.  Unitary ESPRIT: how to exploit additional information inherent in the relational invariance structure , 1994, Proceedings of ICASSP '94. IEEE International Conference on Acoustics, Speech and Signal Processing.

[8]  Kurt Hornik,et al.  Multilayer feedforward networks are universal approximators , 1989, Neural Networks.

[9]  Toshihiko Nishimura,et al.  DOA Estimation of Two Targets with Deep Learning , 2018, 2018 15th Workshop on Positioning, Navigation and Communications (WPNC).

[10]  Stephen P. Boyd,et al.  Convex Optimization , 2004, Algorithms and Theory of Computation Handbook.

[11]  Yoshua Bengio,et al.  Gradient-based learning applied to document recognition , 1998, Proc. IEEE.

[12]  G. G. Stokes "J." , 1890, The New Yale Book of Quotations.

[13]  Emanuel A. P. Habets,et al.  Broadband doa estimation using convolutional neural networks trained with noise signals , 2017, 2017 IEEE Workshop on Applications of Signal Processing to Audio and Acoustics (WASPAA).

[14]  Carlos Fernandez-Granda,et al.  A Learning-based Framework for Line-spectra Super-resolution , 2018, ICASSP 2019 - 2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[15]  Yonina C. Eldar,et al.  Robust Recovery of Signals From a Structured Union of Subspaces , 2008, IEEE Transactions on Information Theory.

[16]  Philip S. Yu,et al.  Direction-of-Arrival Estimation Based on Deep Neural Networks With Robustness to Array Imperfections , 2018, IEEE Transactions on Antennas and Propagation.

[17]  Thomas Kailath,et al.  ESPRIT-estimation of signal parameters via rotational invariance techniques , 1989, IEEE Trans. Acoust. Speech Signal Process..

[18]  Arthur Jay Barabell,et al.  Improving the resolution performance of eigenstructure-based direction-finding algorithms , 1983, ICASSP.

[19]  Soumitro Chakrabarty,et al.  Multi-Speaker DOA Estimation Using Deep Convolutional Networks Trained With Noise Signals , 2018, IEEE Journal of Selected Topics in Signal Processing.

[20]  Sergey Ioffe,et al.  Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift , 2015, ICML.

[21]  Guan Gui,et al.  Deep Learning for Super-Resolution Channel Estimation and DOA Estimation Based Massive MIMO System , 2018, IEEE Transactions on Vehicular Technology.

[22]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[23]  Emma Ozanich,et al.  A Deep Network for Single-Snapshot Direction of Arrival Estimation , 2019, 2019 IEEE 29th International Workshop on Machine Learning for Signal Processing (MLSP).

[24]  Geoffrey E. Hinton,et al.  Deep Learning , 2015, Nature.

[25]  Arye Nehorai,et al.  Coarrays, MUSIC, and the Cramér–Rao Bound , 2016, IEEE Transactions on Signal Processing.

[26]  Petre Stoica,et al.  Performance study of conditional and unconditional direction-of-arrival estimation , 1990, IEEE Trans. Acoust. Speech Signal Process..

[27]  Cishen Zhang,et al.  A Discretization-Free Sparse and Parametric Approach for Linear Array Signal Processing , 2013, IEEE Transactions on Signal Processing.

[28]  Zhang-Meng Liu,et al.  Deep Convolution Network for Direction of Arrival Estimation With Sparse Prior , 2019, IEEE Signal Processing Letters.

[29]  Jian Li,et al.  Sparse Methods for Direction-of-Arrival Estimation , 2016, ArXiv.

[30]  Jie Chen,et al.  Theoretical Results on Sparse Representations of Multiple-Measurement Vectors , 2006, IEEE Transactions on Signal Processing.

[31]  E. Candès The restricted isometry property and its implications for compressed sensing , 2008 .

[32]  S. Frick,et al.  Compressed Sensing , 2014, Computer Vision, A Reference Guide.

[33]  Emma Ozanich,et al.  A feedforward neural network for direction-of-arrival estimation. , 2020, The Journal of the Acoustical Society of America.

[34]  A. Robert Calderbank,et al.  Sensitivity to Basis Mismatch in Compressed Sensing , 2011, IEEE Trans. Signal Process..

[35]  Peter Gerstoft,et al.  Ship localization in Santa Barbara Channel using machine learning classifiers. , 2017, The Journal of the Acoustical Society of America.

[36]  Josef A. Nossek,et al.  Unitary ESPRIT: how to obtain increased estimation accuracy with a reduced computational burden , 1995, IEEE Trans. Signal Process..

[37]  Gitta Kutyniok,et al.  1 . 2 Sparsity : A Reasonable Assumption ? , 2012 .

[38]  P. P. Vaidyanathan,et al.  Nested Arrays: A Novel Approach to Array Processing With Enhanced Degrees of Freedom , 2010, IEEE Transactions on Signal Processing.

[39]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

[40]  Ting Liu,et al.  Recent advances in convolutional neural networks , 2015, Pattern Recognit..

[41]  Sergios Theodoridis,et al.  Academic Press Library in Signal Processing, Volume 3: Array and Statistical Signal Processing , 2013 .

[42]  Björn E. Ottersten,et al.  A subspace method for direction of arrival estimation of uncorrelated emitter signals , 1999, IEEE Trans. Signal Process..