论文:2022,Vol:40,Issue(6):1375-1384
引用本文:
庞伊琼, 许华, 蒋磊, 史蕴豪, 彭翔. 基于混合注意力原型网络的调制识别算法[J]. 西北工业大学学报
PANG Yiqiong, XU Hua, JIANG Lei, SHI Yunhao, PENG Xiang. Modulation recognition algorithm based on mixed attention prototype network[J]. Journal of Northwestern Polytechnical University

基于混合注意力原型网络的调制识别算法
庞伊琼, 许华, 蒋磊, 史蕴豪, 彭翔
空军工程大学 信息与导航学院, 陕西 西安 710077
摘要:
针对极少量带标签样本条件下的通信信号调制识别难题,提出一种基于混合注意力原型网络的调制识别算法。综合元学习和度量学习的思想,在原型网络框架下通过特征提取模块将信号映射至一个新的特征度量空间,并通过比较该空间内各类原型与查询信号之间的距离确定查询信号调制样式。根据通信信号IQ分量的时序特点设计了由卷积神经网络和长短时记忆网络级联的特征提取模块,并引入卷积注意力机制提升关键特征的权重;采用基于Episode的训练策略,使算法可泛化到新的信号识别任务中。仿真结果表明,所提算法在每类信号只有5个带标签样本(5-way 5-shot)时平均识别率可达85.68%。
关键词:    调制识别    原型网络    元学习    度量学习   
Modulation recognition algorithm based on mixed attention prototype network
PANG Yiqiong, XU Hua, JIANG Lei, SHI Yunhao, PENG Xiang
Information and Navigation College, Air Force Engineering University, Xi'an 710077, China
Abstract:
A modulation recognition algorithm based on mixed attention prototype network is proposed to solve the modulation recognition problem of communication signals with very few labeled samples. Combining the ideas of meta learning and metric learning, the algorithm maps the signal to a new feature metric space by feature extraction module in the prototype network framework and determines the modulation pattern of the query signal by comparing the distance between each prototype and the query signal in the space. According to the time sequence characteristics of communication signal IQ components, a feature extraction module is designed which is cascated by the convolutional neural network and long-short term memory network, and the convolutional attention mechanism is introduced to improve the weight of key features. The training strategy based on Episode is used to generalize the algorithm to new signal recognition tasks. The simulation results show that the average recognition rate of the present algorithm can reach 85.68% when there are only 5 labeled samples (5-way 5-shot) for each type of signal.
Key words:    modulation recognition    prototype network    meta learning    metric learning   
收稿日期: 2022-03-11     修回日期:
DOI: 10.1051/jnwpu/20224061375
基金项目: 国家自然科学基金面上项目(61906156)资助
通讯作者: 许华(1976—),空军工程大学教授,主要从事通信信号处理、智能通信对抗研究。e-mail:13720720010@139.com     Email:13720720010@139.com
作者简介: 庞伊琼(1998—),空军工程大学硕士研究生,主要从事深度学习、通信信号识别研究
相关功能
PDF(3010KB) Free
打印本文
把本文推荐给朋友
作者相关文章
庞伊琼  在本刊中的所有文章
许华  在本刊中的所有文章
蒋磊  在本刊中的所有文章
史蕴豪  在本刊中的所有文章
彭翔  在本刊中的所有文章

参考文献:
[1] MILCZAREK H, LESNIK C, DJUROVIC L, et al. Estimating the instantaneous frequency of linear and nonlinear frequency modulated radar signals-a comparative study[J]. Sensors, 2021, 21(8):2840
[2] JIANG Xinrui, CHEN Hui, ZHAO Yaodong, et al. Automatic modulation recognition based on mixed-type features[J]. International Journal of Electronics, 2021, 108(1):105-114
[3] O'SHEA T J, CORGAN J, and CLANCY T C.Convolutional radio modulation recognition networks[C]//17th International Conference on Engineering Applications of Neural Networks, Aberdeen, 2016
[4] O'SHEA T J, ROY T,CLANCY T C. Over-the-air deep learning based radio signal classification[J]IEEE Journal of Selected Topics in Signal Process,2018,12(1):168-179
[5] 白芃远, 许华, 孙莉. 基于卷积神经网络与时频图纹理信息的信号调制方式分类方法[J]. 西北工业大学学报, 2019, 37(4):816-823 BAI Pengyuan, XU Hua, SUN Li. A recognition algorithm for modulation schemes by convolution neural network and spectrum texture[J]. Journal of Northwestern Polytechnical University, 2019, 37(4):816-823 (in Chinese)
[6] KRZYSTON J, BHATTACHARJEA R, STARK A. Complex-valued convolutions for modulation recognition using deep learning[C]//2020 IEEE International Conference on Communications Workshops, Dublin, Ireland, 2020
[7] LIANG Zhi, TAO Mingliang, WANG Ling, et al. Automatic modulation recognition based on adaptive attention mechanism and ResNext WSL model[J]. IEEE Communications Letters, 2021, 25(9):2953-2957
[8] LIU Kai, GAO Wanjun, HUANG Qinghua. Automatic modulation recognition based on a DCN-BiLSTM network[J]. Sensors, 2021, 21(5):1577
[9] NJOKU J N, MOROCHO-CAYAMCELA M E, LIM W. CGDNet:efficient hybrid deep learning model for robust automatic modulation recognition[J].IEEE Networking Letters, 2021, 3(2):47-51
[10] LI Dongjin, YANG Ruijuan, LI Xiaobai, et al. Radar signal modulation recognition based on deep joint learning[J]. IEEE Access, 2020(8):48515-48528
[11] ROYLE J A,DORAZIO R M, LINK W A. Analysis of multinomial models with unknown index using data augmentation[J]. Journal of Computational and Graphical Statistics, 2007, 16(1):67-85
[12] YOSINSKI J, CLUNE J, BENGIO Y, et al. How transferable are features in deep neural networks?[C]//28th Conference on Neural Information Processing Systems, Montreal, Canada, 2014
[13] 史蕴豪, 许华, 刘英辉. 一种基于伪标签半监督学习的小样本调制识别算法[J]. 西北工业大学学报, 2020, 38(5):1074-1083 SHI Yunhao,XU Hua,LIU Yinghui. A few-shot modulation recognition method based on pseudo-label semi-supervised learning[J]. Journal of Northwestern Polytechnical University,2020,38(5):1074-1083 (in Chinese)
[14] MENG Fan, CHEN Peng, WU Lenan, et al. Automatic modulation classification:a deep learning enabled approach[J].IEEE Trans on Vehicular Technology, 2018,67(11):10760-10772
[15] YU Xu, LI Dezhi, WANG Zhenyong, et al. A deep learning method based on convolutional neural network for automatic modulation classification of wireless signals[J]. Wireless Networks, 2019, 25(7):3735-3746
[16] HUISMAN M, VAN RIJN J N, PLAAT A. A survey of deep meta-learning[J]. Artificial Intelligence Review, 2021, 54(6):1-59
[17] CHEN Yutian, HOFFMAN M W, COLMENAREJO S G, et al. Learning to learn by gradient descent by gradient descent[C]//34th International Conference on Machine Learning, Sydney, Australia, 2017
[18] FINN C, ABBEEL P, LEVINE S. Model-agnostic meta-learning for fast adaptation of deep networks[C]//International Conference on Machine Learning, PMLR, 2017
[19] SNELL J, SWERSKY K, ZEMEL R. Prototypical networks for few-shot learning[C]//31st Annual Conference on Neural Information Processing Systems, Long Beach, CA, 2017
[20] HE Kaiming, ZHANG Xiangyu, REN Shaoqing, et al. Deep residual learning for image recognition[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, US, 2016
[21] ZHANG Zilin, LI Yan, GAO Meiguo. Few-shot learning of signal modulation recognition based on attention relation network[C]//2020 28th European Signal Processing Conference, 2021
[22] VINYALS O, BLUNDELL C, LILLICRAP T, et al. Matching networks for one shot learning[C]//30th Conference on Neural Information Processing Systems,Barcelona Spain, 2016
[23] YANG Ning, ZHANG Bangning, DING Guoru, et al. Specific emitter identification with limited samples:a model-agnostic meta-learning approach[J]. IEEE Communications Letters, 2022,26(2):345-349