An Effective Radio Frequency Signal Classification Method Based on Multi-Task Learning Mechanism
Hongwei Liu, Chengyao Hao, Peng Yang, Yu Wang, Tomoaki Ohtsuki, Guan Gui
Abstract
With the increasing popularity of Internet of things (IoT), the emergence of many IoT devices has led to security vulnerabilities. The classification of wireless signals is very important for secure communications. Most of existing signal classification tasks only focus on single signal classification task, while ignoring the relationship between radio frequency fingerprinting identification (RFFI) and automatic modulation classification (AMC). To solve the multi-task classification problem, this paper designs a multi-task learning convolutional neural networks (MTL-CNN). Real-radio datasets are generated by Signal Hound VSG60A and collected by Signal Hound BB60C to solve the lack of RFF samples with numerous modulation types. Experimental results confirm that the MTL-CNN method can work well by using the generated dataset. The MTL network designed in this paper improves the accuracy of RFFI by 1xs% relative to the single-task learning (STL) network. The keras code is released at https://github.comLiuK1288/1hw-000.