eess.SPOct 27, 2026

RAG-CSI: Global Local Retrieval via Joint State and Dynamics Matching for CSI Prediction

Authors: Xinyue Jiang, Tong Li, Shiceng Zhang, Ke Chen, Zhu Xiao

Organizations: College of Computer Science and Electronic Engineering, Hunan University, Changsha, China. · Xidian University, Xian, China. · Peng Cheng Laboratory, Shenzhen 518066, China.

Abstract

Channel state information (CSI) prediction forecasts future channel states from historical observations, supporting transceiver design and resource allocation with reduced measurement overhead. However, existing deep learning predictors mainly encode channel knowledge in model parameters, which can limit their ability to represent diverse patterns of channel dynamics across scenes. In this paper, we propose RAG-CSI, a retrieval-enhanced Transformer framework that augments prediction with relevant historical channel knowledge. Specifically, we construct a global--local CSI knowledge store containing representative observations across scenes and historical observations from the current serving scene. We further develop joint state--dynamics retrieval (JSDR), which jointly characterizes channel states, phase dynamics, and amplitude dynamics to retrieve relevant CSI sequences. A Transformer predictor then uses cross-attention to fuse the retrieved knowledge with the current CSI history and predict future CSI with limited additional model parameters. Experiments on a multi-scene CSI dataset show overall improvements exceeding 10% across all five evaluation metrics over the best baseline for each metric, including a 15.42% reduction in normalized mean squared error.

Figures & tables

Explore similar work

CardsList
  1. JEPA-kMMVAE: CSI Prediction for Time-Varying Wireless Channels

    Oct 9, 2026Efstathios Chatziloizos, Konstantinos Vandikas, Aneta Vulgarakis Feljan +2Channel State PredictionVariational Autoencoders

  2. TRACE-CRC: Trajectory-Adaptive Conformal Risk Control for Multi-Step Channel State Information Prediction

    Aug 27, 2026Kiarash Rezaei, Mehdi Sattari, Javad Aliakbari +3Channel State PredictionUncertainty Quantification

  3. Lightweight PCGAE-Net: Parallel CrossGate Attention and Bottleneck AutoEncoder for Efficient 5G Channel Prediction

    Jun 24, 2026Uma Kishore Godavarti, K. Giridhar, Vanani Prince Dharmendrabhai +2AutoencodersChannel State Prediction