Multi-Task Learning-Based CSI Feedback Design in Multiple Scenarios

Xiangyi Li

doi:10.1109/TCOMM.2023.3317924

Multi-Task Learning-Based CSI Feedback Design in Multiple Scenarios

2023-12-01

4

OA

AI

DOI：10.1109/TCOMM.2023.3317924

Original

Original request for help

Save

Abstract

En 中文

For frequency division duplex (FDD) systems, downlink channel state information (CSI) feedback is essential. Deep learning-based auto-encoder (AE) structures have shown promise in reducing feedback overhead. However, designing a super-large AE network to handle the CSI of all scenarios is not practical. A more practical approach is to divide the CSI dataset by region/scenario and use multiple simple AE networks. However, this method requires high memory capacity, making it unsuitable for low-end user equipment (UE). In this paper, we propose a new UE-friendly framework based on multi-tasking mode. Our framework, called single-encoder-to-multiple-decoders (S-to-M), uses multi-task-learning to design multiple independent AEs into a joint architecture with a shared encoder that corresponds to multiple task-specific decoders. We also integrate GateNet as a classifier to enable the base station to autonomously select the right task-specific decoder for the subregion. Our experiments on a simulated multi-scenario CSI dataset show that our proposed S-to-M framework outperforms other benchmark modes by significantly reducing model complexity and UE memory consumption.

Keywords:

Massive MIMO

CSI feedback

deep learning

multitask learning

multi-scenario

Multi-Task Learning-Based CSI Feedback Design in Multiple Scenarios

Abstract

Journal

Organization