Subspace Representation Learning for Sparse Linear Arrays to Localize More Sources than Sensors: A Deep Learning Methodology

Kuan-Lin Chen,Bhaskar D. Rao
2024-08-29
Abstract:Localizing more sources than sensors with a sparse linear array (SLA) has long relied on minimizing a distance between two covariance matrices and recent algorithms often utilize semidefinite programming (SDP). Although deep neural network (DNN)-based methods offer new alternatives, they still depend on covariance matrix fitting. In this paper, we develop a novel methodology that estimates the co-array subspaces from a sample covariance for SLAs. Our methodology trains a DNN to learn signal and noise subspace representations that are invariant to the selection of bases. To learn such representations, we propose loss functions that gauge the separation between the desired and the estimated subspace. In particular, we propose losses that measure the length of the shortest path between subspaces viewed on a union of Grassmannians, and prove that it is possible for a DNN to approximate signal subspaces. The computation of learning subspaces of different dimensions is accelerated by a new batch sampling strategy called consistent rank sampling. The methodology is robust to array imperfections due to its geometry-agnostic and data-driven nature. In addition, we propose a fully end-to-end gridless approach that directly learns angles to study the possibility of bypassing subspace methods. Numerical results show that learning such subspace representations is more beneficial than learning covariances or angles. It outperforms conventional SDP-based methods such as the sparse and parametric approach (SPA) and existing DNN-based covariance reconstruction methods for a wide range of signal-to-noise ratios (SNRs), snapshots, and source numbers for both perfect and imperfect arrays.
Signal Processing,Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to locate more sources than the number of sensors in a Sparse Linear Array (SLA). Traditional methods usually rely on minimizing the distance between two covariance matrices, and recent algorithms often utilize Semidefinite Programming (SDP). Although methods based on Deep Neural Network (DNN) offer new alternatives, they still rely on the fitting of covariance matrices. The paper proposes a new method. By estimating co - array subspaces from sample covariance, it trains DNN to learn the representations of signal and noise subspaces, which are invariant to the choice of basis. To learn such representations, the authors propose new loss functions to measure the separation between the desired subspaces and the estimated subspaces. In particular, they propose loss functions that measure the shortest path length between subspaces on the union of Grassmann manifolds and prove that DNN can approximate the signal subspaces. Furthermore, the paper also proposes a new batch sampling strategy - consistent rank sampling - to accelerate the learning of subspaces of different dimensions. This method is robust to array imperfections because it is geometry - independent and data - driven. In addition, the authors also propose a fully end - to - end grid - free method to directly learn angles and study the possibility of bypassing the subspace method. In summary, the main goal of this paper is to develop a new subspace representation learning method for locating more sources than the number of sensors in SLA while improving robustness to array imperfections.