Towards a Foundation Model for Communication Systems

Artificial Intelligence (AI) has demonstrated unprecedented performance across various domains, and its application to communication systems is an active area of research. While current methods focus on task-specific solutions, the broader trend in AI is shifting toward large general models capable of supporting multiple applications. In this work, we take a step toward a foundation model for communication data--a transformer-based, multi-modal model designed to operate directly on communication data. We propose methodologies to address key challenges, including tokenization, positional embedding, multimodality, variable feature sizes, and normalization. Furthermore, we empirically demonstrate that such a model can successfully estimate multiple features, including transmission rank, selected precoder, Doppler spread, and delay profile.

@article{buffelli2025_2505.14603,
  title={ Towards a Foundation Model for Communication Systems },
  author={ Davide Buffelli and Sowmen Das and Yu-Wei Lin and Sattar Vakili and Chien-Yi Wang and Masoud Attarifar and Pritthijit Nath and Da-shan Shiu },
  journal={arXiv preprint arXiv:2505.14603},
  year={ 2025 }
}