mlsanityData to 13 Oct 2026

HeterCSI: Channel-Adaptive Heterogeneous CSI Pretraining Framework for Generalized Wireless Foundation Models

arXiv 2601.18200 ↗First posted 26 Jan 2026cs.LG cs.AIcited by 3

Not on the map. Why?

Abstract (arXiv)

Wireless foundation models promise transformative capabilities for channel state information (CSI) processing across diverse 6G network applications, yet face fundamental challenges due to the inherent dual heterogeneity of CSI across both scale and scenario dimensions. However, current pretraining approaches either constrain inputs to fixed dimensions or isolate training by scale, limiting the generalization and scalability of wireless foundation models. In this paper, we propose HeterCSI, a channel-adaptive pretraining framework that reconciles training efficiency with robust cross-scenario generalization via a new understanding of gradient dynamics in heterogeneous CSI pretraining. Our key insight reveals that CSI scale heterogeneity primarily causes destructive gradient interference, while scenario diversity actually promotes constructive gradient alignment when properly managed. Specifically, we formulate heterogeneous CSI batch construction as a partitioning optimization problem that minimizes zero-padding overhead while preserving scenario diversity. To solve this, we develop a scale-aware adaptive batching strategy that aligns CSI samples of similar scales, and design a double-masking mechanism to isolate valid signals from padding artifacts. Extensive experiments on 12 datasets demonstrate that HeterCSI establishes a generalized foundation model without scenario-specific finetuning, achieving superior average performance over full-shot baselines. Compared to the state-of-the-art zero-shot benchmark WiFo, it reduces NMSE by 7.19 dB, 4.08 dB, and 5.27 dB for CSI reconstruction, time-domain, and frequency-domain prediction, respectively. The proposed HeterCSI framework also reduces training latency by 53% compared to existing approaches while improving generalization performance by 1.53 dB on average.

Key facts. Cited by 3 papers in the map (data through 13 Oct 2026). Builds on 30 of its 49 references that are in the map. First posted on arXiv 26 Jan 2026.

Builds on

30 of 49 references.

References · 30

  1. 2017Attention Is All You Needcited by 60,106
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,304
  3. 2020Language Models are Few-Shot Learnerscited by 32,320
  4. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,043
  5. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,223
  6. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,364
  7. 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,344
  8. 2013An Empirical Investigation of Catastrophic Forgetting in Gradient-Based Neural Networkscited by 718
  9. 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 549
  10. 2023From Quantity to Quality: Boosting LLM Performance with Self-Guided Data Selection for Instruction Tuningcited by 303
  11. 2023SpectralGPT: Spectral Remote Sensing Foundation Modelcited by 251
  12. 2024LLM4CP: Adapting Large Language Models for Channel Predictioncited by 42
18 more
  1. 2024Large Wireless Model (LWM): A Foundation Model for Wireless Channelscited by 40
  2. 2025WirelessGPT: A Generative Pre-trained Multi-task Learning Framework for Wireless Communicationcited by 40
  3. 2024WiFo: Wireless Foundation Model for Channel Predictioncited by 28
  4. 2020Massive MIMO Channel Prediction: Kalman Filtering vs. Machine Learningcited by 21
  5. 2025A Multi-Task Foundation Model for Wireless Channel Representation Using Contrastive and Masked Autoencoder Learningcited by 16
  6. 2025LLM4WM: Adapting LLM for Wireless Multi-Taskingcited by 15
  7. 2023Domain Generalization in Machine Learning Models for Wireless Communications: Concepts, State-of-the-Art, and Open Issuescited by 12
  8. 2019Addressing the curse of mobility in massive MIMO with Prony-based angular-delay domain channel predictionscited by 10
  9. 2025Large Wireless Localization Model (LWLM): A Foundation Model for Positioning in 6G Networkscited by 10
  10. 2024ChannelGPT: A Large Model to Generate Digital Twin Channel for 6G Environment Intelligencecited by 8
  11. 2025BERT4MIMO: A Foundation Model using BERT Architecture for Massive MIMO Channel State Information Predictioncited by 8
  12. 2025LVM4CSI: Enabling Direct Application of Pre-Trained Large Vision Models for Wireless Channel Taskscited by 7
  13. 2025Enabling 6G Through Multi-Domain Channel Extrapolation: Opportunities and Challenges of Generative Artificial Intelligencecited by 4
  14. 2025LLM4PG: Adapting Large Language Model for Pathloss Map Generation via Synesthesia of Machinescited by 3
  15. 2025Sharp Minima Can Generalize: A Loss Landscape Perspective On Datacited by 3
  16. 2025Bridging the Modality Gap: Enhancing Channel Prediction with Semantically Aligned LLMs and Knowledge Distillationcited by 2
  17. 2024MMSE Channel Estimation in Large-Scale MIMO: Improved Robustness with Reduced Complexitycited by 1
  18. 2026WiFo-M2: Empower Wireless Communications With Plug-and-Play Environment Sensing via Foundation Modelcited by 1

Built on by

Cited by year (2026 to 13 Oct)

Citing papers per year: 2026: 3.

3 papers.