Enhancing Direction-of-Arrival Estimation with Multi-Task Learning

There are numerous methods in the literature for Direction-of-Arrival (DOA) estimation, including both classical and machine learning-based approaches that jointly estimate the Number of Sources (NOS) and DOA. However, most of these methods do not fully leverage the potential synergies between these two tasks, which could yield valuable shared information. To address this limitation, in this article, we present a multi-task Convolutional Neural Network (CNN) capable of simultaneously estimating both the NOS and the DOA of the signal. Through experiments on simulated data, we demonstrate that our proposed model surpasses the performance of state-of-the-art methods, especially in challenging environments characterized by high noise levels and dynamic conditions.