English

Knowledge Distillation for mmWave Beam Prediction Using Sub-6 GHz Channels

Signal Processing 2026-05-22 v1 Machine Learning

Abstract

Beamforming in millimeter-wave (mmWave) high-mobility environments typically incurs substantial training overhead. While prior studies suggest that sub-6 GHz channels can be exploited to predict optimal mmWave beams, existing methods depend on large deep learning (DL) models with prohibitive computational and memory requirements. In this paper, we propose a computationally efficient framework for sub-6 GHz channel-mmWave beam mapping based on the knowledge distillation (KD) technique. We develop two compact student DL architectures based on individual and relational distillation strategies, which retain only a few hidden layers yet closely mimic the performance of large teacher DL models. Extensive simulations demonstrate that the proposed student models achieve the teacher's beam prediction accuracy and spectral efficiency while reducing trainable parameters and computational complexity by 99%.

Keywords

Cite

@article{arxiv.2602.04703,
  title  = {Knowledge Distillation for mmWave Beam Prediction Using Sub-6 GHz Channels},
  author = {Sina Tavakolian and Nhan Thanh Nguyen and Ahmed Alkhateeb and Markku Juntti},
  journal= {arXiv preprint arXiv:2602.04703},
  year   = {2026}
}

Comments

5 pages, 4 figures. Accepted for publication at IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP) 2026

R2 v1 2026-07-01T09:36:10.609Z