An automated multi-layer perceptron discriminative neural network based on Bayesian optimization achieves high-precision one-source single-snapshot direction-of-arrival estimation

Bin Zhang,Jiawen He,Peishun Liu,Liang Wang,Ruichun Tang
DOI: https://doi.org/10.1038/s41598-024-60798-w
IF: 4.6
2024-05-07
Scientific Reports
Abstract:This paper proposes an innovative global solution which is a pioneering work applying automated machine learning algorithms to remarkable precision sparse underwater direction-of-arrival (DOA) estimation that views the subaquatic sparse-sampling DOA estimation problem as a classification prediction task. The proposed solution, termed automated multi-layer perceptron discriminative neural network (AutoMPDNN), is built upon a Bayesian optimization framework. AutoMPDNN transforms sparsely sampled time-domain signals into the complex domain, preserving essential components in a one-source single-snapshot scenario. Leveraging Bayesian optimization principles, the algorithm embeds necessary hyperparameters into the loss function, effectively defining it as a maximum likelihood problem using the upper confidence bound function and incorporating sparse signal features. We also explore the model space architecture and introduce variants of AutoMPDNN, denoted as AutoMPDNNs_ln (n = 2,3,4). Through a series of plane wave simulation experiments, it is demonstrated that AutoMPDNN achieves the highest prediction performance for one-source single-snapshot scenarios compared to classical DOA estimation algorithms that incorporate sparse representation approaches, as well as contemporary deep learning DOA methods under varying conditions.
multidisciplinary sciences
What problem does this paper attempt to address?
The paper primarily addresses the problem of underwater Direction-of-Arrival (DOA) estimation, especially in scenarios involving one-source single-snapshot and sparse sampling conditions for high-precision DOA estimation. Traditional DOA estimation methods often perform poorly under complex interference environments, rapidly moving or non-static targets, and spatial sampling limitations. To solve the aforementioned issues, the researchers propose an innovative global solution, namely the Automated Multi-Layer Perceptron Discriminative Neural Network (AutoMPDNN) based on a Bayesian optimization framework. This method treats the DOA estimation problem with sparse underwater sampling as a classification prediction task and utilizes machine learning algorithms to achieve high-precision DOA estimation. Specifically, the main contributions of AutoMPDNN include: 1. **Signal Reconstruction and Feature Extraction**: Reconstructing the sparsely sampled time-domain signals through signal processing techniques to extract key feature information, adapting to the one-source single-snapshot scenario. 2. **Design of Multi-Layer Perceptron Discriminative Neural Network (MPDNN)**: Constructing a lightweight multi-layer perceptron network model based on supervised learning for end-to-end DOA estimation. 3. **Automated Hyperparameter Optimization**: Using Bayesian optimization strategies to automatically adjust the hyperparameters in MPDNN, thereby achieving overall model optimization. This method incorporates hyperparameters as penalty terms in the loss function, defining the problem through maximum likelihood estimation. Through a series of plane wave simulation experiments, the paper demonstrates that AutoMPDNN exhibits higher predictive performance compared to traditional DOA estimation algorithms and modern deep learning DOA methods in one-source single-snapshot scenarios under different signal-to-noise ratio (SNR) conditions. This indicates that AutoMPDNN can effectively address the challenges of sparse data and shows high accuracy and real-time performance in practical applications.