A Comparative Study of Invariance-Aware Loss Functions for Deep Learning-based Gridless Direction-of-Arrival Estimation
Kuan-Lin Chen, Bhaskar D. Rao
TL;DR
This work tackles gridless direction-of-arrival estimation for sparse linear arrays by moving beyond covariance-matrix reconstruction toward invariance-aware loss functions. It introduces a scale-invariant loss based on the scale-invariant SDR and analyzes its invariance properties, comparing it to affine- and geodesic-distance-based approaches and to a recently proposed subspace loss that operates on Grassmann manifolds. Experiments on minimum-redundancy and nested MRAs demonstrate that scale-invariant losses outperform non-invariant alternatives, while the subspace loss achieves the best performance due to its maximal invariance to basis changes. The findings indicate that increasing invariance in loss design expands the effective solution space and improves learning, providing practical guidance for loss design in gridless DoA estimation with DL.
Abstract
Covariance matrix reconstruction has been the most widely used guiding objective in gridless direction-of-arrival (DoA) estimation for sparse linear arrays. Many semidefinite programming (SDP)-based methods fall under this category. Although deep learning-based approaches enable the construction of more sophisticated objective functions, most methods still rely on covariance matrix reconstruction. In this paper, we propose new loss functions that are invariant to the scaling of the matrices and provide a comparative study of losses with varying degrees of invariance. The proposed loss functions are formulated based on the scale-invariant signal-to-distortion ratio between the target matrix and the Gram matrix of the prediction. Numerical results show that a scale-invariant loss outperforms its non-invariant counterpart but is inferior to the recently proposed subspace loss that is invariant to the change of basis. These results provide evidence that designing loss functions with greater degrees of invariance is advantageous in deep learning-based gridless DoA estimation.
