Deep Transfer Learning for 5G Massive MIMO Downlink CSI Feedback

IEEE Wireless Communications and Networking Conference, pp. 1–5

Abstract

Acquisition of downlink channel state information (CSI) is an important procedure performed at the base station (BS) for high quality wireless communication in frequency division duplexing (FDD) communication system. Generally, the downlink CSI is fed back to the BS through the user equipment (UE). Compared with traditional methods, neural network (NN) can effectively compress the downlink CSI, thus greatly reducing the feedback overhead. However, the generalization of the NN is poor, hence it is necessary to train a NN from scratch whenever there is a change in the wireless channel environment. Nevertheless, training a NN this way requires huge data and time cost in 5G massive MIMO systems. In this paper, deep transfer learning (DTL) is proposed to solve the problem of high training cost of the downlink CSI feedback NN. In a new wireless environment, our proposed technique utilises relatively small number of samples to fine-tune a pre-trained model, in order to obtain a new model with low training cost. The performance of this model is shown to be comparable with that of the NN trained with large samples. Experiment results demonstrate the effectiveness and superiority of the proposed method.

Authors 7

1. Jun Zeng

   Nanjing University of Posts and Telecommunications

   Affiliation as printed

   College of Telecommunications and Information Engineering, NJUPT, Nanjing, China

2. Zhengran He

   Nanjing University of Posts and Telecommunications

   Affiliation as printed

   College of Telecommunications and Information Engineering, NJUPT, Nanjing, China

3. Jinlong Sun

   Nanjing University of Posts and Telecommunications

   Affiliation as printed

   College of Telecommunications and Information Engineering, NJUPT, Nanjing, China

4. Bamidele Adebisi

   Manchester Metropolitan University

   Affiliation as printed

   Faculty of Science and Engineering, Manchester Metropolitan University, UK

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

6. Guan Gui

   Nanjing University of Posts and Telecommunications

   Affiliation as printed

   College of Telecommunications and Information Engineering, NJUPT, Nanjing, China

7. Fumiyuki Adachi

   Tohoku University

   Affiliation as printed

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