Knowledge Distillation for mmWave Beam Prediction Using Sub-6 GHz Channels
AI 摘要
利用知识蒸馏技术,论文提出一种高效的毫米波波束预测框架,显著降低计算和存储需求。
主要贡献
- 提出基于知识蒸馏的毫米波波束预测框架
- 设计两种紧凑的学生模型架构
- 大幅降低计算复杂度和模型参数量
方法论
利用子6 GHz信道信息,通过知识蒸馏训练小型学生模型,模仿大型教师模型的波束预测性能。
原文摘要
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%.