Lightweight Automatic Modulation Classification Based on Decentralized Learning

IEEE Transactions on Cognitive Communications and Networking, vol. 8, pp. 57–70

Abstract

Due to the implementation and performance limitations of centralized learning automatic modulation classification (CentAMC) method, this paper proposes a decentralized learning AMC (DecentAMC) method using model consolidation and lightweight design. Specifically, the model consolidation is realized by a central device (CD) for edge device (ED) model averaging (MA) and multiple EDs for ED model training. The lightweight is designed by separable convolutional neural network (S-CNN), in which the separable convolutional layer is utilized to replace the standard convolution layer and most of fully connected layers are cut off. Simulation results show that the proposed method substantially reduces the storage and computational capacity requirements of the EDs and communication overhead. The training efficiency also shows remarkable improvement. Compared with convolutional neural network (CNN), the space complexity (i.e., model parameters and output feature map) is decreased by about 94% and the time complexity (i.e., floating point operations) of S-CNN is decreased by about 96% while degrading the average correct classification probability by less than 1%. Compared with S-CNN-based CentAMC, without considering model weights uploading and downloading, the training efficiency of our proposed method is about${N}$times of it, where${N}$is the number of EDs. Considering the model weights uploading and downloading, the training efficiency of our proposed method can still be maintained at a high level (e.g., when the number of EDs is 12, the training efficency of the proposed AMC method is about 4 times that of S-CNN-based CentAMC in dataset$D_{1} = \{2{\mathrm {FSK, 4FSK, 8FSK, BPSK, QPSK, 8PSK, 16QAM}}\}$and about 5 times that of S-CNN-based CentAMC in dataset$D_{2} = \{2 {\mathrm {FSK, 4FSK, 8FSK, BPSK, QPSK, 8PSK, PAM2, PAM4, PAM8, 16QAM}}\}$), while the communication overhead is reduced more than 35%.

Authors 7

1. Xue Gang Fu

   Nanjing University of Posts and Telecommunications

   Affiliation as printed

   College of Telecommunications and Information Engineering, Nanjing University of Posts and Telecommunications, Nanjing, China

2. Guan Gui

   Nanjing University of Posts and Telecommunications

   Affiliation as printed

   College of Telecommunications and Information Engineering, Nanjing University of Posts and Telecommunications, Nanjing, China

3. Yu Wang

   Nanjing University of Posts and Telecommunications

   Affiliation as printed

   College of Telecommunications and Information Engineering, Nanjing University of Posts and Telecommunications, Nanjing, China

4. Tomoaki Otsuki Ohtsuki

   Keio University

   Affiliation as printed

   Department of Information and Computer Science, Keio University, Yokohama, Japan

5. Bamidele Adebisi

   Manchester Metropolitan University

   Affiliation as printed

   Department of Engineering, Faculty of Science and Engineering, Manchester Metropolitan University, Manchester, U.K

6. Haris Gacanin Aachen Institute for Communication Technologies and Embedded Systems

   RWTH Aachen University

   Affiliation as printed

   Institute for Communication Technologies and Embedded Systems, RWTH Aachen University, Aachen, Germany

7. Fumiyuki Adachi

   Tohoku University

   Affiliation as printed

   Research Organization of Electrical Communication, Tohoku University, Sendai, Japan