MDTCNet:Multi-Task Classifications Network and TCNN for Direction of Arrival Estimation  

作　　者：Yu Jiarun Wang Yafeng 

机构地区：[1]Department of Information and Communication Engineering,Beijing University of Posts and Telecommunications,Beijing 100876,China [2]Key Laboratory of Universal Wireless Communications(Ministry of Education),Beijing University of Posts and Telecommunications,Beijing 100876,China

出　　处：《China Communications》2024年第10期148-166,共19页中国通信（英文版）

基　　金：funded by Beijing University of Posts and Telecommunications-China Mobile Research Institute Joint Innovation Center。

摘　　要：The direction-of-arrival(DoA) estimation is one of the hot research areas in signal processing. To overcome the DoA estimation challenge without the prior information about signal sources number and multipath number in millimeter wave system,the multi-task deep residual shrinkage network(MTDRSN) and transfer learning-based convolutional neural network(TCNN), namely MDTCNet, are proposed. The sampling covariance matrix based on the received signal is used as the input to the proposed network. A DRSN-based multi-task classifications model is first introduced to estimate signal sources number and multipath number simultaneously. Then, the DoAs with multi-signal and multipath are estimated by the regression model. The proposed CNN is applied for DoAs estimation with the predicted number of signal sources and paths. Furthermore, the modelbased transfer learning is also introduced into the regression model. The TCNN inherits the partial network parameters of the already formed optimization model obtained by the CNN. A series of experimental results show that the MDTCNet-based DoAs estimation method can accurately predict the signal sources number and multipath number under a range of signal-to-noise ratios. Remarkably, the proposed method achieves the lower root mean square error compared with some existing deep learning-based and traditional methods.

关 键 词：DoA estimation MDTCNet millimeter wave system multi-task classifications model regression model 

分 类 号：TN911.7[电子电信—通信与信息系统] TP183[电子电信—信息与通信工程]