论文标题
在常规和RIS辅助大规模MIMO中进行频道估计的联合学习
Federated Learning for Channel Estimation in Conventional and RIS-Assisted Massive MIMO
论文作者
论文摘要
机器学习(ML)由于其复杂性低和鲁棒性而引起了物理层设计问题(例如渠道估计)的极大研究兴趣。通过ML的频道估计需要数据集上的模型培训,该模型通常包括接收到的试点信号作为输入和频道数据作为输出。在以前的工作中,模型培训主要是通过集中学习(CL)进行的,其中整个培训数据集都是从基站(BS)的用户那里收集的。这种方法引入了巨大的通信开销,以供数据收集。在本文中,为了应对这一挑战,我们提出了一个联合学习(FL)框架以进行渠道估计。我们设计了一个在用户本地数据集中训练的卷积神经网络(CNN),而无需将其发送到BS。我们为常规和RI(智能反射表面)辅助MIMO(多输入多输出)系统开发了基于FL的通道估计方案,其中为两种情况培训了两个不同数据集的单个CNN。我们评估了嘈杂和量化模型传输的性能,并表明所提出的方法提供的开销比CL低约16倍,同时保持接近CL的令人满意的性能。此外,所提出的体系结构比最新基于ML的方案显示出较低的估计误差。
Machine learning (ML) has attracted a great research interest for physical layer design problems, such as channel estimation, thanks to its low complexity and robustness. Channel estimation via ML requires model training on a dataset, which usually includes the received pilot signals as input and channel data as output. In previous works, model training is mostly done via centralized learning (CL), where the whole training dataset is collected from the users at the base station (BS). This approach introduces huge communication overhead for data collection. In this paper, to address this challenge, we propose a federated learning (FL) framework for channel estimation. We design a convolutional neural network (CNN) trained on the local datasets of the users without sending them to the BS. We develop FL-based channel estimation schemes for both conventional and RIS (intelligent reflecting surface) assisted massive MIMO (multiple-input multiple-output) systems, where a single CNN is trained for two different datasets for both scenarios. We evaluate the performance for noisy and quantized model transmission and show that the proposed approach provides approximately 16 times lower overhead than CL, while maintaining satisfactory performance close to CL. Furthermore, the proposed architecture exhibits lower estimation error than the state-of-the-art ML-based schemes.