Deep Transfer Learning Based Downlink Channel Prediction for FDD Massive MIMO Systems

Abstract

Artificial intelligence (AI) based downlink channel state information (CSI) prediction for frequency division duplexing (FDD) massive multiple-input multiple-output (MIMO) systems has attracted growing attention recently. However, existing works focus on the downlink CSI prediction for the users under a given environment and is hard to adapt to users in new environment especially when labeled data is limited. To address this issue, we formulate the downlink channel prediction as a deep transfer learning (DTL) problem, where each learning task aims to predict the downlink CSI from the uplink CSI for one single environment. Specifically, we develop the direct-transfer algorithm based on the fully-connected neural network architecture, where the network is trained on the data from all previous environments in the manner of classical deep learning and is then fine-tuned for new environments. To further improve the transfer efficiency, we propose the meta-learning algorithm that trains the network by alternating inner-task and across-task updates and then adapts to a new environment with a small number of labeled data. Simulation results show that the direct-transfer algorithm achieves better performance than the deep learning algorithm, which implies that the transfer learning benefits the downlink channel prediction in new environments. Moreover, the meta-learning algorithm significantly outperforms the direct-transfer algorithm in terms of both prediction accuracy and stability, which validates its effectiveness and superiority.

arXiv commentAccepted by IEEE Transactions on Communications

References in corpus (11)

• Adam: A Method for Stochastic Optimization
• Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks
• Channel Estimation and Hybrid Precoding for Millimeter Wave Cellular Systems
• Distributed Compressive CSIT Estimation and Feedback for FDD Multi-user Massive MIMO Systems
• On First-Order Meta-Learning Algorithms
• Deep CNN-Based Channel Estimation for mmWave Massive MIMO Systems
• DeepMIMO: A Generic Deep Learning Dataset for Millimeter Wave and Massive MIMO Applications
• Understanding the Loss Surface of Neural Networks for Binary Classification
• Deep UL2DL: Channel Knowledge Transfer from Uplink to Downlink
• Learning to Transfer: Unsupervised Meta Domain Translation
• 3D Scene Based Beam Selection for mmWave Communications

Cited by in corpus (5)

• Learning to Demodulate from Few Pilots via Offline and Online Meta-Learning
• From Learning to Meta-Learning: Reduced Training Overhead and Complexity for Communication Systems
• Transfer Learning-based Channel Estimation in Orthogonal Frequency Division Multiplexing Systems Using Data-nulling Superimposed Pilots
• Transfer Learning and Meta Learning Based Fast Downlink Beamforming Adaptation
• FusionNet: Enhanced Beam Prediction for mmWave Communications Using Sub-6GHz Channel and A Few Pilots