Direction of Arrival Estimation for Coherent Signals’ Method Based on LSTM Neural Network
Radio direction finding system is a system that determines the direction or coordinates of radio signal sources. The main function of this system is to determine the direction of arrival (DOA) of an incident radio wave. DOA information plays an important role in array signal processing and has many applications in communications, radar, seismic survey, etc. In this study, we propose a method to estimate the DOA by using the simulated signal dataset obtained at the linear antenna array (ULA) and the suitable Long Short-Term Memory (LSTM) network model. The performance of the method is evaluated based on the root mean square error (RMSE) parameter and then is compared with 2 other algorithms, multiple signal classification (MUSIC) and deep neural network (DNN) in different cases such as deviation of incoming signals, variation of signal-to-noise ratio (SNR), and coherent incoming signals. The obtained results have shown that the proposed method has significantly improved accuracy compared to other methods.
For a long time, the problem of determining DOA has been a common problem in radio communication systems, radar systems , and navigation systems in air and waterway traffic . These systems often use antenna arrays such as uniform linear antenna array (ULA), uniform circular antenna array (UCA), and uniform rectangular antenna array (URA) . Many methods and algorithms have been researched and deployed to calculate DOA such as MUSIC [4–7], ESPRIT , total forward-backward matrix pencil [9, 10], and acoustic vector sensor . They are also continuously developed to improve performance in DOA estimation for accuracy, resolution, and adaptability in the case of a limited number of snapshots, low signal-to-noise ratio (SNR), signal-to-noise correlation, etc.
In recent years, the application of artificial intelligence techniques in the DOA estimation problem has been concerned. Network models have been applied to improve accuracy and speed in DOA calculations [12, 13]. The deep learning methods do not need to calculate the signal characteristics during the prediction process, so the real-time estimation process will be shortened thereby providing higher real-time applicability such as support vector regression (SVR) [14, 15] and support vector machine (SVM) [16, 17]. In deep neural network (DNN) [12, 18], convolution neural network (CNN) [13, 19–21] and Adam optimal function were used to estimate DOA with satisfactorily accurate results. Also, radial basis function neural network (RBFNN)  can estimate the DOA with good accuracy under favorable environmental conditions.
This study focuses on the research and development of a simulation database of the signals received from the ULA antenna array. From the obtained dataset, the suitable long-short term memory (LSTM) algorithm is proposed to be applied to calculate the DOA of incoming signals which are coherent. The received results will be evaluated and compared with other typical methods to assess the performance of the proposed method.
The study is organized as follows. Section 2 introduces the summary of research results on DOA calculation that have been done previously. Section 3 presents the antenna array model, the method of simulating the signal received at the antenna array, and the applied algorithm model. Section 4 shows the experimental results and gives evaluation for each algorithm. Conclusions and future work are in Section 5.
2. Related Work
Table 1 summarizes several methods of estimating the DOA based on the signal spectrum. These included both correlated and uncorrelated signals. They are divided into two categories: using machine learning algorithms and using classical algorithms. The classical algorithm based on multisignal classification can predict accurately, but the computational complexity is high. One of the most commonly used classical algorithms is MUSIC which was discovered by independent studies of Schmidt  and Bienvenu . The music algorithm has been shown to work well when the signals are uncorrelated, the incoming signal sources are far apart and the SNR is large enough. Specifically, with the antenna array ULA, MUSIC can estimate the DOA of 2 signals at and with for and . However, when the signal sources are correlated, the performance of the MUSIC algorithm is not good. Since then, the improved MUSIC algorithms have been studied to determine the DOA of correlated incoming signals with the reduced number of calculations, such as IMMUSIC and MMUSIC  or using the covariance matrix with the transpose elements . However, if there are more three correlated incoming signals, classical algorithms still need to improve and develop. In addition, algorithms are often developed with the assumption that the number of incoming sources is known [5, 8, 9], which reduces the generality of the problem and is no longer true when applied in practice because the signal received at the antenna array is the total signal of many unknown and unstable incoming sources.
Several recent publications have shown that machine learning methods have gradually been applied to solve the problem of DOA estimation. Usually, convolutional neural network algorithms can extract the basic nonlinear structures of the input data. Therefore, CNN with a simple layer structure can estimate the DOA of uncorrelated sources [13, 19]. Because of its simple structure, the CNN network performs DOA estimation with large bandwidth quickly and efficiently, thereby the DOA information can be calculated in real time.
CNNs are often trained with large amounts of data where suitable data are fed into the network when they have almost the same distributions, including both training and testing data [25, 26]. In fact, besides DOA information, the received signal model at the antenna array includes many unknown parameters such as the number of incoming sources, frequency, and signal-to-noise ratio. Signals received from the antenna array will be preprocessed to reduce their distribution divergence before becoming input data of the DNN [12, 18]. The output of a DNN network [18, 27] is usually as an angular grid, corresponding to each position in the angular grid representing the spectral value of the signal. If the angle of incidence coincides with the angle present in the mesh, then the DOA problem can be estimated correctly. However, the angles present in the mesh cannot match the actual DOA angle completely. The authors in  describe the construction of the network in 2 stages, in which the first stage performs the estimation with the grid of angles. The second stage corrects the difference between the DOA and the discrete angle in the nearest mesh, thus resolving the disparity caused by the discrete angle. Specifically, this study estimates 2 narrowband and uncorrelated incoming signals at the ULA antenna array. The number of antennas in the array is 8 with the number of snapshots being 256. The DOA was estimated with at and at . However, the research results of this study still need to be improved to be able to estimate DOA in cases where there are more than 2 incoming signals as well as in cases correlated incoming signals.
Besides using RNNs for applications related to natural language processing such as speech recognition, RNN networks are also used for DOA estimation [28–30]. In , the RNN is created based on bidirectional long-short term memory (BiLSTM). RNNs do not directly estimate DOA but classify them based on classes.
In those classes, the incident angles are in the range with angular deviations of the incoming sources being , respectively. This study for the estimation of DOA has quite good results. However, similar to the previously presented DNN algorithm, the RNN is dependent on the angular grid for classification and the performance of this method is still degraded at low SNR.
In , the LSTM network is also used to determine the DOA. With LSTM networks, it is suitable for nonstationary targets because it can be generalized to learn sequential patterns. The LSTM network is presented in more detail in . In the unknown multipath environment, it is necessary to estimate the DOA for a moving target using the LSTM-based “New Multi-frame Phase Enhancement” technique, in which the recommended number of frames is 3, 5, and 7. In most cases, when conducting surveys under the same environmental conditions, the larger the number of frames, the better the accuracy. For example, the signal received at the ULA antenna array-comprising 21 elements with distanced , , and wavelength , using the estimated network got RMSE values of , , and which correspond to k = 7, 5, 3. The increase in frame weight increases the accuracy of DOA estimation but reduces the performance of the problem. Therefore, depending on the environmental conditions, the system must choose the appropriate number of frames. Although the results are quite good, the problem still encounters obstacles when the incoming sources are correlated. To improve on the DOA problem when the incoming sources are correlated signals and the angular deviations of the signal sources are small, this study proposes an LSTM network with a simpler model consisting of fully connected layers, excluding frame, to estimate the direction of the incoming signal. The obtained model will be compared with DNN and LSTM algorithms through the RMSE parameter presented in Section 3.
3. Materials and Methods
3.1. Uniform Linear Antenna Array and Signal Model
This study uses a uniform linear antenna array (ULA) with elements. The structure of the ULA antenna array is shown in Figure 1. The element in the antenna system acts as an omnidirectional source. These antenna elements operate in phase with each other to create a unique radiation direction so that the signal sent to the processor is kept in phase and amplitude in comparison with the incoming signal. Furthermore, the ULA antenna array has several advantages over other types of antenna arrays as shown in [3, 9].
Assuming that the incoming signal is in the same azimuth plane as the antenna array, the signal transmitted to the antenna array is illustrate as Figure 1.
The antenna array (ULA) used in this study has M elements, equally spaced with a distance of . Assume that the system has incoming signal sources with wavelength . The received signal at each antenna element is the sum of all incoming signals at the same time. The mathematical representation of the signal received at the antenna element is described as in Equation.(1), where and are the complex amplitude and the DOA of the source , respectively:where and is the noise received at the antenna element of the array.
Define is a matrix of size including the elements represented as follows:
Therefore, Equation (1) can be rewritten aswhere are defined bywhere is the transpose of the matrix.
In [16, 33], the signals received at the antenna array will be passed through a preprocessor before being processed to calculate DOA information. Therefore, the correlation matrix of size of the received signals at the antenna array can be represented as follows:where and are the expectation and the Hermitian transpose, respectively, and and are correlation matrices of size of signal and noise, respectively, and are represented as follows:
From there, Equation (2) can be rewritten aswhere is an identity matrix of size and is the noise power. The correlation matrix is also called as the Hermitian matrix. This matrix is used as input for the DOA estimation models.
3.2. DOA Estimation
3.2.1. Recurrent Neural Network
Deep learning has two major models: convolutional neural network (CNN) for problems with image input and recurrent neural network (RNN) for sequence data problems.
Recurrent neural network is a model that uses memory to store information from previous computation steps and makes an accurate prediction for the current prediction step. Consider the “many to many” RNN model, as shown in Figure 2.
Figure 2 shows that the input will be combined with the previously hidden layer using the function to compute the hidden layer and the output . is the set of weights added to all activation functions. Loss functions are to calculate the difference in the output from the actual value. The smaller the value of the loss function, the more accurate the result.
Recurrent means that the model will perform identical calculations for each element of the input data series, and the output will depend on the results of the previous calculations. Here, the RNN only uses a single neural network (usually a layer) to calculate the output value in each time step. Therefore, the outputs converted to inputs will be multiplied by the same weight matrix (here, as shown in Figure 2). It is also why there is the word Recurrent in the name of the RNN.
3.2.2. Long Short-Term Memory Networks
Long short-term memory (LSTM) is an artificial recurrent neural network that takes the form of a sequence of repeating modules and contains feedback connections. This network is often used in problems where the input is a data string such as speech or video. Figure 3 shows the structure of multilayer LSTM networks.
With the network depicted in Figure 3, the LSTM network nodes in the same layer connect in a chain form and connect to the corresponding nodes in the next layer. An LSTM unit consists of an input and an output port. They have a 4-layer structure that interacts with each other in a very specific way as depicted in Figure 4.
The general parameters of the network model have been described in . Specifically, in the state of the LSTM model, the classic LSTM module structure includes input , are the outputs in the state, and is the input in the state of the model. Output includes , where is called cell state and is hidden state at state:where is the forget gate, is input gate, is output gate, , are the bias coefficients, and is the weight matrix.
This study proposes an LSTM model for the DOA problem, as shown in Figure 5. This network is designed with one input layer, three LSTM layers, three fully connected layers, and one output layer. Relu activation function is used at the output of each layer fully connected. In the output layer, we use the linear activation function. The signal received at the antenna array is processed at the preprocessing unit to obtain the correlation vector. That correlation vector is the input to the LSTM network. The output layer with elements is used to estimate the DOA. A detailed description of the classes is shown in the following section.
For the fully connected layer, the nodes in the former layer are connected to all the nodes in the following layer with their coefficients . Each node has its bias coefficient . Each node in the fully connected, and output layer performs two steps: linear summation and applying the activation function, as shown in equations (15) and (16). Assume the number of nodes in the fully connected is . The matrix of size is the coefficient matrix between layer and layer , where is the connection coefficient from the node of layer to the node of layer . Vector of size is the bias coefficient of the nodes in layer , where is the bias of the node in layer . Step 1 (linear summation): this is the sum of all nodes in the previous layer multiplied by the corresponding factor plus the bias : Step 2: applying activation function,where is the activation function, vector of size is the value of nodes in layer after the linear summation step. Vector of size is the value of nodes in layer after applying the activation function.
3.2.3. Data Preprocessing
The LSTM network is trained with a large amount of data. In order to reduce the input bias and variation of the signal, the signal preprocessing is carried out with the input signal received at the antenna array and the output as a correlation matrix of size :
Since is a Hermitanian matrix, the upper triangular matrix and the lower triangular matrix carry the same information. According to the methods published in [12, 34], the upper triangular matrix has enough information to compute the DOA. Therefore, to reduce the amount of information for the input of the network, this study only uses the upper triangular matrix of and then transforms it into a vector of length :where is the element in the matrix . Since is a complex number, it cannot be put directly into the network for calculation. Therefore, before putting into the training network, each element will be represented into 2 components, the real part and the imaginary part .
3.2.4. Data Generation
In this section, a general method to generate data for the training model for the case of multiple incoming sources is proposed. Suppose there are sources to the elements ULA antenna array-(). The angles correspond to incoming sources in the range . The values are defined, respectively, as the difference between the incoming sources, where is the difference between the first and the second incoming signals, is the difference between the , and the incoming signals with . Then, the DOA of the first source is , then the DOA of source, 3rd… and , respectively, is , and then the signal received at the antenna array is calculated as in Equations (1) and (4).
For data generation process, when the DOA of the first signal is sampled in the range  with a jump of , the DOA of signal … and , respectively, are . From there, the total amount of generated data is calculated by the following formula:
3.2.5. Data Labeling
The input is defined as vector as Equation (18), where is the number of incoming signals, corresponding to the incoming signals at the angles .
In this study, a labeling method called one hot encoding with multiple labels is used to label the data. is the corresponding label of the incoming signals. With 121 outputs corresponding to incoming angles in the range with a jump of , is defined as
Therefore, the output of the LSTM network corresponding to the input is
3.2.6. Evaluation Parameters
To evaluate the accuracy of two proposed models, this research uses two error functions: MSE and RMSE.(a)MSE function: to evaluate the model during training, the network uses the mean square error loss (MSE) function. MSE is defined as(b)RMSE function: this study uses the root mean square error function (RMSE) to evaluate the performance of the model and algorithm: where is the number of incident sources, is the number of trials, and and are the incident angle of the source and the estimated angle of source at the trial.
4. Experiments and Results
In this section, the results obtained from the LSTM method are presented in different cases and compared with some other DOA methods, such as Music and DNN.
4.1. Simulation Establishment
This study uses a 10-element ULA antenna array with . The incoming signal has a frequency of . The received signals must be narrowband in both correlated and noncorrelated cases; simulation data are generated according to Equation. (1) with and .
For the LSTM network, the size of each layer is shown in Table 2. Assume that the incoming signals are in the range [, with a jump of . Therefore, the number of nodes of the output layer will be 121. The spatial spectrum is constructed with a grid of , so there is a total of 121 grids with . In this study, we use the spectral reconstruction method according to .
The training samples are generated by considering signals separated by . In this study, we assume that the number of incoming signals is one, two, and three, respectively. The details are shown in Table 3. The data generation process is mentioned in Section 3.2.4.
For the case of two incoming signals, when the DOA of the first signal is created by sampling in the range with a jump of , then the DOA of the second signal will be . In the case of three incoming signals, when the DOA of the first signal is created by sampling in the range with a jump of , the DOA of the second signal and are and , respectively. In the last case, with an incoming signal, the DOA of the signal is created by sampling in the range with a jump of . The covariance vectors are computed from the samples as input to the LSTM network according to Equation. (18) and the corresponding labels, as in Equation. (20). The SNR for incoming signals in all cases is . For the training process, the learning rate is 0.001, the batch size is 1024, and the number of Epochs is 400. During the training, the network parameters are continuously updated to optimize the MSE loss function, as mentioned in Equation. (21). In addition, the network uses the ADAM optimization algorithm  to optimize the time and predictability of the algorithm.
4.2. Simulation Results
4.2.1. Uncorrelated Signal
In the first test, with , we assume that the incoming signals are uncorrelated. All three algorithms with the same simulation setup conditions will be executed. With the LSTM network, the results shown in Figure 6 include the cases of 1 source , two sources (DOA of and ), and three incoming sources (DOA of , 36°, and 42°). In the above cases, the DOA can be well estimated. Specifically, in Figure 6(a), it is easy to see that the signal spectrum is quite simple because there is only one incoming source, LSTM gives the result . In the remaining two cases, when the angular distance is , the LSTM network can all estimate well with the signal spectrum separated by . With the DNN network and the MUSIC algorithms, it can be seen that the proposed LSTM network estimates the DOAs more accurately than the other two algorithms as well as the previous studies (CNN and LSTM). The signal spectrum is shown in Figures 7 and 8. It shows that the signal sources have not been separated.
In the next test, to evaluate the influence of SNRs on the accuracy of the algorithm, we apply the LSTM model to estimate the DOA when there are three incoming sources with angular difference . Considering 2 cases of and with 31 test samples, the results are shown in Figure 9. At , the estimated angle is almost close to the actual angle. When , although the estimated result differs slightly from reality, the LSTM model still gives good results with error .
To clarify more clearly the influence of SNR on the performance of the LSTM network model in the case of one incoming signal, consider the SNR in the range . The results are shown in Figure 10. It is easy to see that, in the case of 1 source to the antenna array, LSTM still gives better results than the other two methods with .
For the case of 2 incoming sources, consider the SNR in the range . The results are shown in Figure 11. It can be seen that, with two incoming sources, the LSTM method still gives better results than the DNN and MUSIC networks.
When investigating the angle resolution of the proposed algorithm, we consider the case that there are two uncorrelated incoming signals with at . The results are shown in Tables 4–6, respectively, with LSTM, DNN, and MUSIC algorithms. From the results’ tables, we can see that the LSTM model still works effectively and gives better results than the other two methods in both cases, where the signals are close to each other and far apart with .
4.2.2. Correlated Signal
With the correlated incoming signals, the MUSIC algorithm no longer works correctly [4, 5]. Therefore, the MUSIC-IMPROVE algorithm (according to the covariance matrix transpose method) and DNN are used to compare with the results obtained from the LSTM algorithm.
Test on two incoming signals with the DOA of and, the same with three incoming signals at the DOA of , and The results are presented in Figures 12–14 for LSTM, MUSIC-IMPROVE, and DNN algorithms, respectively. Since the signal spectrum conspicuous to the incoming signal is two correlated signals, all three algorithms work well with a well-separated signal spectrum. When the incoming signal is over two, both MUSIC-IMPROVE and DNN give worse results, while the LSTM algorithm still works well.
To evaluate the resolution in this case, we assume that the three correlated incoming signals differ by in turn amount degrees. Simulation results are shown in Table 7, while with two algorithms, MUSIC-IMPROVE and DNN, the estimated results are shown in Tables 8 and 9. Those tables show that the LSTM model still correctly predicts the DOA of many correlated incoming signals, while the MUSIC algorithm proves to be less efficient in this case. This can also be seen in Figure 13. Some results published in  have also shown that some other improved MUSIC methods also give poor results in the case of correlated incoming signals. For the DNN model, with the difference among the incoming signals being small, the DNN method also gives poor results. The algorithm only works better when the difference among the incoming signals is large enough.
The next experiment will evaluate the effect of SNRs on the accuracy of the algorithm. In the case of two correlated incoming signals, consider the SNR in the range of ; the results are shown in Figure 15. With the three correlated incoming signals and the SNR in the range of , the result depicted in Figure 16 indicates that, in both cases, the LSTM model gives the best results compared to the other two methods (in the case of , ).
Figure 17 shows the RMSE comparison results of the algorithms at the DOA , with the SNR in the range of in the case that the first and second signals are correlated, but they are uncorrelated with the third signal. The received results plotted in this figure shows that the proposed method works better than the other methods. From the above comparison cases, it is easy to see that the LSTM model works effectively and gives good results in most cases, especially in the case of many correlated incoming signals.
Figures 18 and 19 plot the results of the LSTM model in two cases: all three incoming signals are correlated and the case of 2 correlated signals with one uncorrelated signal at and with 31 samples and . It can be easily seen that, in the case of multiple incoming signals, the LSTM model still works well, giving almost accurate results at different SNR values. The comparison results with the MUSIC IMPROVE algorithm in the case of correlated signals are also summarized in Table 10.
This study proposed the modified LSTM network to estimate the DOA of coherent incoming signals with the ULA antenna system. Two keys contributions of this work are(i)Create a simulation database of the signal received at the ULA antenna array in the case of multiple incoming sources, which are narrowband signals, in the two cases, where the incoming signals are correlated and uncorrelated(ii)Propose to apply the modified LSTM algorithm with an architecture that combines network nodes with fully connected layers using Adam’s optimization function in the DOA estimation problem in both cases of uncorrelated and correlated incoming signals
The obtained simulation results show that the model works more accurately than typical algorithms such as MUSIC and DNN algorithms in cases of low SNR, multiple incoming signals, and uncorrelated and correlated incoming signals, as well as when the radiation source is quite close. However, the LSTM algorithm is still limited, where the deviation between the angles is not in the training set and the error is still quite high. In the future, it can be developed to work with other antenna systems, such as UCA, or increase accuracy.
The data used to support the findings of this study are available from the corresponding author upon request.
This study was approved by Hanoi University of Science and Technology and Vietnam Maritime University (Vietnam).
Conflicts of Interest
The authors declare no conflicts of interest.
The authors appreciate the support from the two universities, Ha Noi University of Science and Technology and Vietnam Maritime University.
D. E. Dudgeon and D. H. Johnson, Array Signal Processing: Concepts and Techniques, Englewood Cliffs, Bergen County, New Jersey, USA, 1993.
L. Vu Van Yem, Application of The Music Algorithms to Localisation of Small and Medium-Sized Fishing Boats in Vietnam Sea, 2007.
Y. Gao, W. Chang, Z. Pei, and Z. Wu, “An improved music algorithm for DOA estimation,” Sensors and Transducers, vol. 175, no. 7, pp. 75–82, 2014.View at: Google Scholar
D.-S. Kim and V.-S. Doan, “DOA estimation of multiple non-coherent and coherent signals using element,” ICT Express, vol. 6, pp. 68–76, 2020.View at: Google Scholar
M. Ahmad and X. Zhang, “Performance of music algorithm for DOA estimation,” in Proceedings of the International Conference in Aerospace for Young Scientists, China, Beijingpp, 2016.View at: Google Scholar
H. T. Thanh, V. V. Yem, N. D. Minh, and H. D Thang, Direction of Arrival Estimation using the Total Forward-Backward Matrix Pencil Method, ICCE, Virar, India, 2014.
M. Wajid, A. Kumar, and R. Bahl, “Direction estimation and tracking of coherent sources using a single acoustic vector sensor,” Archives of Acoustics, vol. 45, no. 2, pp. 209–219, 2020.View at: Google Scholar
W. Zhu and M. Zhang, “A deep learning architecture for broadband DOA estimation,” in Proceedings of the 19th IEEE International Conference on Communication Technology, Xi’an, China, 2019.View at: Google Scholar
M. Wajid, F. Alam, S. Yadav, M. A. Khan, and M. Usman, “Support vector regression based direction of arrival estimation of an acoustic source,” in Proceedings of the International Conference on Innovation and Intelligence for Informatics, Sakheer, Bahrain, 2020.View at: Publisher Site | Google Scholar
M. Pastorino and A. Randazzo, “Real-time SVM-based approach for localization of sources,” in Proceedings of the IEEE International Workshop Imaging Systems and Technology (IST), Stresa, Italy, 2004.View at: Google Scholar
M. Wajid, F. Alam, S. Yadav, M. A. Khan, and M. Usman, “Support vector machine-based direction of arrival estimation with uniform linear array,” Advances in Computational Intelligence Techniques, pp. 253–264, 2020.View at: Google Scholar
H. Chung, H. Seo, J. Joo, D. Lee, and S. Kim, “Off-grid DOA estimation via two-stage cascaded neural network,” Energies, vol. 14, pp. 6–10, 2021.View at: Google Scholar
R. O. Schmidt, “Multiple emitter location and signal parameter estimation,” in Proceedings of the RADC Spectrum Estimation Workshop, pp. 243–258, Griffiss AFB, NY, USA, 1979.View at: Google Scholar
G. Bienvenu and L. Kopp, “Principe de la goniometrie passive adaptive,” in Proceedings of the the 7 è me Colloque GRETSI, 106/1-106/10, Nice, France, 1979.View at: Google Scholar
M. Sugiyama, A. Schwaighofer, N. D. Lawrence, and J. Q. Candela, Dataset Shift in Machine Learning, The MIT Press, Cambridge, Massachusetts, USA, 2009.
M. Sugiyama and M. Kawanabe, Machine Learning in Non-Stationary Environments: Introduction to Covariate Shift Adaptation, MIT press, Cambridge, Massachusetts, USA, 2012.
M. Wajid, B. Kumar, A. Goel, A. Kumar, and R. Bahl, in Proceedings of the 5th IEEE International Conference on Signal Processing, Computing and Control, Solan, India, 2019.
A. H. El Zooghby, G. C. Christos, and M. Georgiopoulos, “A neural network-based smart antenna for multiple source tracking,” IEEE Transactions on Antennas and Propagation, vol. 48, pp. 1–4, 2000.View at: Google Scholar