English

Beam-Response Contrastive Learning for Transmitter-Side MIMO CSI Representation

Information Theory 2026-07-27 v1 Systems and Control

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 H\mathbf{H}, R=HHH\mathbf{R}=\mathbf{H}^{\mathrm{H}}\mathbf{H} determines the received power of any unit-norm transmit beam w\mathbf{w} as Hw22=wHRw|\mathbf{H}\mathbf{w}|_2^2=\mathbf{w}^{\mathrm{H}}\mathbf{R}\mathbf{w}. 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}
}