WiMamba: Linear-Scale Wireless Foundation Model
On the map: State-space sequence models (S4, Mamba) → Core SSM architectures for language
Abstract (arXiv)
Foundation models learn transferable representations, motivating growing interest in their application to wireless systems. Existing wireless foundation models are predominantly based on transformer architectures, whose quadratic computational and memory complexity can hinder practical deployment for large-scale channels. In this work, we introduce WiMamba, a wireless foundation model built upon the recently proposed Mamba architecture, which replaces attention mechanisms with selective state-space models and enables linear-time sequence modeling. Leveraging this architectural advantage combined with adaptive preprocessing, WiMamba achieves scalable and low-latency inference while maintaining strong representational expressivity. We further develop a dedicated task-agnostic, self-supervised pre-training framework tailored to wireless channels, resulting in a genuine foundation model that learns transferable channel representations. Evaluations across four downstream tasks demonstrate that WiMamba matches or outperforms transformer-based wireless foundation models, while offering dramatic latency and memory reductions.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Core SSM architectures for language, one of 10 directions in State-space sequence models (S4, Mamba). First posted on arXiv 27 Mar 2026.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2026ComHymba: Low-Complexity Domain-Informed Foundation Model for Wireless Communications
- 2026MambaCSP: Hybrid-Attention State Space Models for Hardware-Efficient Channel State Prediction
- 2026Fast Wireless Foundation Models with Early-Exits
- 2026Structured Latent Dynamics in Wireless CSI via Homomorphic World Models
- 2025TransMamba: Fast Universal Architecture Adaption from Transformers to Mamba
- 2025SCA-LLM: Spectral-Attentive LLM-Based Wireless World Modeling for Agentic Communications
- 2025Sensing and Understanding the World over Air: A Large Multimodal Model for Mobile Networks
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spaces