CCA-Net: A Lightweight Network Using Criss-Cross Attention for CSI Feedback

Abstract

To enhance the performance of a massive multiple-input multiple-output (MIMO) system, the downlink channel state information (CSI) should be compressed and fed back to the base station (BS) effectively in a frequency division duplex (FDD) mode. In contrast to traditional compression methods that rely on the particular prior assumptions, deep learning (DL) based CSI feedback has the potential to learn informatively from the hidden complex models behind the data. However, most of the current DL based methods stay in the early stage that uses DL modules directly from image processing, while the distinctive channel semantic information has been largely ignored. In this letter, considering the spatial-temporal correlation property of CSI matrices, we propose an architecture based on criss-cross attention, called CCA-Net, which extracts the channel features more precisely by leveraging the contextual information along the delay and angle domain. Meanwhile, a real and imaginary part combination module is used to assist in extracting the internal features of CSI. In addition, to reap the computational resources at BSs, a scalable decoder is designed to extend the reconstruction abilities. Experiments demonstrate that our network achieves better normalized mean square error with lower flops than existing state-of-the-art algorithms.