WiFo-M$^2$: Plug-and-Play Multi-Modal Sensing via Foundation Model to Empower Wireless Communications
Haotian Zhang, Shijian Gao, Xiang Cheng
TL;DR
WiFo-M2 introduces a foundation-model approach to sensing-aided wireless communications, aiming to extract universal out-of-band channel features from multi-modal data (images and LiDAR) and apply them across diverse transceiver modules. The ContraSoM pre-training strategy aligns sensing representations with CSI using modality-specific augmentations, including diffusion-based LiDAR feature augmentation and temporal feature extrapolation, enabling future CSI-aligned feature generation. Across beam prediction, channel estimation, interpolation, and prediction, WiFo-M2 demonstrates universal gains and strong cross-scenario generalization, while remaining lightweight enough for real-time deployment. This work shifts the design paradigm from task-specific pipelines to a plug-and-play, universal representation that broadens the impact of multi-modal sensing on core wireless tasks and demonstrates practical deployability with modest parameter overhead.
Abstract
The growing adoption of sensor-rich intelligent systems has boosted the use of multi-modal sensing to improve wireless communications. However, traditional methods require extensive manual design of data preprocessing, network architecture, and task-specific fine-tuning, which limits both development scalability and real-world deployment. To address this, we propose WiFo-M$^2$, a foundation model that can be easily plugged into existing deep learning-based transceivers for universal performance gains. To extract generalizable out-of-band (OOB) channel features from multi-modal sensing, we introduce ContraSoM, a contrastive pre-training strategy. Once pre-trained, WiFo-M$^2$ infers future OOB channel features from historical sensor data and strengthens feature robustness via modality-specific data augmentation. Experiments show that WiFo-M$^2$ improves performance across multiple transceiver designs and demonstrates strong generalization to unseen scenarios.
