Next-slot OFDM-CSI Prediction: Multi-head Self-attention or State Space Model?

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Akrout, Mohamed, Bellili, Faouzi, Mezghani, Amine, Heath, Robert W.
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929348158685184
author Akrout, Mohamed
Bellili, Faouzi
Mezghani, Amine
Heath, Robert W.
author_facet Akrout, Mohamed
Bellili, Faouzi
Mezghani, Amine
Heath, Robert W.
contents The ongoing fifth-generation (5G) standardization is exploring the use of deep learning (DL) methods to enhance the new radio (NR) interface. Both in academia and industry, researchers are investigating the performance and complexity of multiple DL architecture candidates for specific one-sided and two-sided use cases such as channel state estimation (CSI) feedback, CSI prediction, beam management, and positioning. In this paper, we set focus on the CSI prediction task and study the performance and generalization of the two main DL layers that are being extensively benchmarked within the DL community, namely, multi-head self-attention (MSA) and state-space model (SSM). We train and evaluate MSA and SSM layers to predict the next slot for uplink and downlink communication scenarios over urban microcell (UMi) and urban macrocell (UMa) OFDM 5G channel models. Our numerical results demonstrate that SSMs exhibit better prediction and generalization capabilities than MSAs only for SISO cases. For MIMO scenarios, however, the MSA layer outperforms the SSM one. While both layers represent potential DL architectures for future DL-enabled 5G use cases, the overall investigation of this paper favors MSAs over SSMs.
format Preprint
id arxiv_https___arxiv_org_abs_2405_11072
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Next-slot OFDM-CSI Prediction: Multi-head Self-attention or State Space Model?
Akrout, Mohamed
Bellili, Faouzi
Mezghani, Amine
Heath, Robert W.
Information Theory
Signal Processing
The ongoing fifth-generation (5G) standardization is exploring the use of deep learning (DL) methods to enhance the new radio (NR) interface. Both in academia and industry, researchers are investigating the performance and complexity of multiple DL architecture candidates for specific one-sided and two-sided use cases such as channel state estimation (CSI) feedback, CSI prediction, beam management, and positioning. In this paper, we set focus on the CSI prediction task and study the performance and generalization of the two main DL layers that are being extensively benchmarked within the DL community, namely, multi-head self-attention (MSA) and state-space model (SSM). We train and evaluate MSA and SSM layers to predict the next slot for uplink and downlink communication scenarios over urban microcell (UMi) and urban macrocell (UMa) OFDM 5G channel models. Our numerical results demonstrate that SSMs exhibit better prediction and generalization capabilities than MSAs only for SISO cases. For MIMO scenarios, however, the MSA layer outperforms the SSM one. While both layers represent potential DL architectures for future DL-enabled 5G use cases, the overall investigation of this paper favors MSAs over SSMs.
title Next-slot OFDM-CSI Prediction: Multi-head Self-attention or State Space Model?
topic Information Theory
Signal Processing
url https://arxiv.org/abs/2405.11072