Beam-Response Contrastive Learning for Transmitter-Side MIMO CSI Representation
Abstract
Self-supervised representation learning from unlabeled channel state information (CSI) can reduce labeling and adaptation overhead in learning-based multiple-input multiple-output (MIMO) systems. Existing CSI pretraining methods typically use reconstruction objectives or contrastive pairs from generic augmentations, which do not explicitly reflect transmission-relevant channel similarity. This paper proposes beam-response contrastive learning (BRCL), a self-supervised CSI pretraining framework based on the transmit-side Gram matrix. For a channel matrix , determines the received power of any unit-norm transmit beam as . BRCL maps each CSI sample to a beam-response profile and uses the induced soft similarity as a label-free relational target for contrastive pretraining. Combined with reconstruction learning, BRCL enforces both sample-level CSI recovery and beam-response-level consistency, yielding transferable CSI representations without task-specific labels or manual positive pairs. Experiments on diverse MIMO channel datasets show that BRCL improves label efficiency and outperforms autoencoder- and channel-charting-based pretraining across beam selection, user selection, and future beam selection tasks.
Cite
@article{arxiv.2607.24872,
title = {Beam-Response Contrastive Learning for Transmitter-Side MIMO CSI Representation},
author = {Sehyun Ryu and Yumin Kim and Minjae Lee and Hyun Jong Yang and John M. Cioffi},
journal= {arXiv preprint arXiv:2607.24872},
year = {2026}
}