Federated Learning for Automatic Modulation Classification Under Class Imbalance and Varying Noise Condition

IEEE Transactions on Cognitive Communications and Networking, vol. 8, pp. 86–96

Abstract

Automatic modulation classification (AMC) is a promising technology for identifying modulation types, and deep learning (DL)-based AMC is one of its main research directions. Conventional DL-based AMC methods are centralized solutions (i.e., CentAMC), which are trained on abundant data collected from local clients and stored in the server and generally have advanced performance, but their major problem is the risk of data leakage. Besides, if DL-based AMC is only trained with the data from their corresponding clients, it may exhibit weak performance. Thus, a federated learning (FL)-based AMC (FedeAMC) is proposed under the condition of class imbalance and noise varying. Its advantage is low risk of data leakage without severe performance loss, because data and training are in each local client, while only knowledge (i.e., gradient or model weight), rather than data, is shared with the server. In addition, there is generally class imbalance problem in each local client, and balanced cross entropy is introduced as loss function for solving this problem. Simulation results demonstrated that average accuracy gap between FedeAMC and CentAMC is less than 2%.

Authors 6

1. 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

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. Haris Gacanin Aachen Faculty of Electrical Engineering and Information Technology

   RWTH Aachen University

   Affiliation as printed

   Faculty of Electrical Engineering and Information Technology, RWTH Aachen University, Aachen, Germany

4. Bamidele Adebisi

   Manchester Metropolitan University

   Affiliation as printed

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

5. Hikmet Sari

   Nanjing University of Posts and Telecommunications

   Affiliation as printed

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

6. Fumiyuki Adachi

   Tohoku University

   Affiliation as printed

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