arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2502.08758eess.SPcs.LG

面向大规模MIMO预编码的特定站点深度神经网络压缩

Compression of Site-Specific Deep Neural Networks for Massive MIMO Precoding

Polytechnique Montréal(蒙特利尔理工学院)

Ghazal Kasalaee, Ali Hasanzadeh Karkan, Jean-François Frigon, François Leduc-Primeau

AI总结:

针对大规模MIMO预编码中DL模型高能耗问题,本文提出融合混合精度量化感知训练与神经架构搜索的压缩框架,经射线追踪数据集验证,压缩模型在性能相当下能效较WMMSE最高提升35倍,为高能效DL预编码器提供基准。

摘要:

在大规模多输入多输出(mMIMO)系统中部署深度学习(DL)模型进行预编码时,往往受到高计算需求和能耗的限制。本文研究了基于DL方法的mMIMO预编码器的计算能效,并将其与迫零(zero forcing)、加权最小均方误差(WMMSE)等传统方法进行比较。我们的能耗模型同时考虑了DL加速器中的内存访问和计算能耗。我们提出了一个融合混合精度量化感知训练和神经架构搜索的框架,在不降低精度的前提下降低能耗。利用覆盖不同基站站点的射线追踪数据集,我们分析了特定站点条件对压缩模型能效的影响。结果表明,根据场景和期望速率的不同,深度神经网络压缩生成的预编码器在性能相当的情况下,能效比WMMSE高出多达35倍。这些结果为开发高能效的基于DL的mMIMO预编码器奠定了基础并提供了基准。

英文摘要:

The deployment of deep learning (DL) models for precoding in massive multiple-input multiple-output (mMIMO) systems is often constrained by high computational demands and energy consumption. In this paper, we investigate the compute energy efficiency of mMIMO precoders using DL-based approaches, comparing them to conventional methods such as zero forcing and weighted minimum mean square error (WMMSE). Our energy consumption model accounts for both memory access and calculation energy within DL accelerators. We propose a framework that incorporates mixed-precision quantization-aware training and neural architecture search to reduce energy usage without compromising accuracy. Using a ray-tracing dataset covering various base station sites, we analyze how site-specific conditions affect the energy efficiency of compressed models. Our results show that deep neural network compression generates precoders with up to 35 times higher energy efficiency than WMMSE at equal performance, depending on the scenario and the desired rate. These results establish a foundation and a benchmark for the development of energy-efficient DL-based mMIMO precoders.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑