# CSI-4CAST: A Hybrid Deep Learning Model for CSI Prediction with Comprehensive Robustness and Generalization Testing

arXiv: https://arxiv.org/abs/2510.12996 · HTML: https://mlsanity.com/paper/2510.12996/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 32 of its 75 references that are in the map.
- First posted on arXiv 14 Oct 2025.

## Builds on (most cited first, 30 of 32)

- [Attention Is All You Need](https://mlsanity.com/paper/1706.03762/) (2017)
- [MobileNetV2: Inverted Residuals and Linear Bottlenecks](https://mlsanity.com/paper/1801.04381/) (2018)
- [MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications](https://mlsanity.com/paper/1704.04861/) (2017)
- [Optuna: A Next-generation Hyperparameter Optimization Framework](https://mlsanity.com/paper/1907.10902/) (2019)
- [Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising](https://mlsanity.com/paper/1608.03981/) (2016)
- [ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices](https://mlsanity.com/paper/1707.01083/) (2017)
- [ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design](https://mlsanity.com/paper/1807.11164/) (2018)
- [Are Language Models Actually Useful for Time Series Forecasting?](https://mlsanity.com/paper/2406.16964/) (2024)
- [LLM4CP: Adapting Large Language Models for Channel Prediction](https://mlsanity.com/paper/2406.14440/) (2024)
- [Multi-resolution CSI Feedback with deep learning in Massive MIMO System](https://mlsanity.com/paper/1910.14322/) (2019)
- [Massive MIMO Channel Prediction: Kalman Filtering vs. Machine Learning](https://mlsanity.com/paper/2009.09967/) (2020)
- [Deep Learning based Downlink Channel Prediction for FDD Massive MIMO System](https://mlsanity.com/paper/1908.03360/) (2019)
- [CSI-BERT2: A BERT-inspired Framework for Efficient CSI Prediction and Classification in Wireless Communication and Sensing](https://mlsanity.com/paper/2412.06861/) (2024)
- [Addressing the curse of mobility in massive MIMO with Prony-based angular-delay domain channel predictions](https://mlsanity.com/paper/1912.11330/) (2019)
- [A Comparison of Neural Networks for Wireless Channel Prediction](https://mlsanity.com/paper/2308.14020/) (2023)
- [Optimal Design of Energy-Efficient Multi-User MIMO Systems: Is Massive MIMO the Answer?](https://mlsanity.com/paper/1403.6150/) (2014)
- [A Spatially Separable Attention Mechanism for massive MIMO CSI Feedback](https://mlsanity.com/paper/2208.03369/) (2022)
- [Deep UL2DL: Channel Knowledge Transfer from Uplink to Downlink](https://mlsanity.com/paper/1812.07518/) (2018)
- [Enabling FDD Massive MIMO through Deep Learning-based Channel Prediction](https://mlsanity.com/paper/1901.03664/) (2019)
- [Estimating Doubly-Selective Channels for Hybrid mmWave Massive MIMO Systems: A Doubly-Sparse Approach](https://mlsanity.com/paper/1912.08380/) (2019)
- [Machine Learning for Future Wireless Communications: Channel Prediction Perspectives](https://mlsanity.com/paper/2502.18196/) (2025)
- [Spectral Temporal Graph Neural Network for massive MIMO CSI Prediction](https://mlsanity.com/paper/2312.02159/) (2023)
- [Modelling the 5G Energy Consumption using Real-world Data: Energy Fingerprint is All You Need](https://mlsanity.com/paper/2406.16929/) (2024)
- [AI for CSI Prediction in 5G-Advanced and Beyond](https://mlsanity.com/paper/2504.12571/) (2025)
- [FDD massive MIMO channel spatial covariance conversion using projection methods](https://mlsanity.com/paper/1804.04850/) (2018)
- [Enhancing Deep Learning Performance of Massive MIMO CSI Feedback](https://mlsanity.com/paper/2208.11333/) (2022)
- [Hybrid Data-Driven SSM for Interpretable and Label-Free mmWave Channel Prediction](https://mlsanity.com/paper/2411.11576/) (2024)
- [Phase noise in modular millimeter wave massive MIMO](https://mlsanity.com/paper/1910.09095/) (2019)
- [A Multi-Dimensional Matrix Pencil-Based Channel Prediction Method for Massive MIMO with Mobility](https://mlsanity.com/paper/2208.01935/) (2022)
- [Complex CNN CSI Enhancer for Integrated Sensing and Communications](https://mlsanity.com/paper/2305.17938/) (2023)

## Abstract (arXiv)

Channel state information (CSI) prediction is a promising strategy for ensuring reliable and efficient operation of massive multiple-input multiple-output (mMIMO) systems by providing timely downlink (DL) CSI. While deep learning-based methods have advanced beyond conventional model-driven and statistical approaches, they remain limited in robustness to practical non-Gaussian noise, generalization across diverse channel conditions, and computational efficiency. This paper introduces CSI-4CAST, a hybrid deep learning architecture that integrates 4 key components, i.e., Convolutional neural network residuals, Adaptive correction layers, ShuffleNet blocks, and Transformers, to efficiently capture both local and long-range dependencies in CSI prediction. To enable rigorous evaluation, this work further presents a comprehensive benchmark, CSI-RRG for Regular, Robustness and Generalization testing, which includes more than 300,000 samples across 3,060 realistic scenarios for both TDD and FDD systems. The dataset spans multiple channel models, a wide range of delay spreads and user velocities, and diverse noise types and intensity degrees. Experimental results show that CSI-4CAST achieves superior prediction accuracy with substantially lower computational cost, outperforming baselines in 81.5% of TDD scenarios and 44.4% of FDD scenario, the best performance among all evaluated models, while reducing FLOPs by 5x and 3x compared to LLM4CP, the strongest baseline. In addition, evaluation over CSI-RRG provides valuable insights into how different channel factors affect the performance and generalization capability of deep learning models. Both the dataset (https://huggingface.co/CSI-4CAST) and evaluation protocols (https://github.com/AI4OPT/CSI-4CAST) are publicly released to establish a standardized benchmark and to encourage further research on robust and efficient CSI prediction.