Skip to main content
QUICK REVIEW

[论文解读] A Model-Driven Deep Learning Network for MIMO Detection

Hengtao He, Chao-Kai Wen|arXiv (Cornell University)|Sep 25, 2018
Wireless Signal Modulation Classification参考文献 16被引用 4
一句话总结

该论文提出OAMP-Net,一种用于MIMO检测的模型驱动深度学习网络,通过将正交近似消息传递(OAMP)算法展开为每层具有可训练参数的神经网络,实现性能显著提升。通过反向传播优化这些参数,该网络在瑞利衰落和相关MIMO信道中均表现出色,尤其在8×8天线MIMO系统中,误比特率(BER)性能最高提升达2.97 dB。

ABSTRACT

In this paper, we propose a model-driven deep learning network for multiple-input multiple-output (MIMO) detection. The structure of the network is specially designed by unfolding the iterative algorithm. Some trainable parameters are optimized through deep learning techniques to improve the detection performance. Since the number of trainable variables of the network is equal to that of the layers, the network can be easily trained within a very short time. Furthermore, the network can handle time-varying channel with only a single training. Numerical results show that the proposed approach can improve the performance of the iterative algorithm significantly under Rayleigh and correlated MIMO channels.

研究动机与目标

  • 通过将深度学习与迭代检测算法结合,解决MIMO检测中的性能-复杂度权衡问题。
  • 开发一种可训练的、基于模型的神经网络,继承OAMP算法的结构以提升检测精度。
  • 实现快速且稳定的训练,仅需极少可训练参数,适用于时变信道。
  • 实现软输出检测,以兼容现代无线系统。
  • 在独立和相关MIMO衰落信道中,证明其性能优于传统OAMP和现有基于深度学习的检测器。

提出的方法

  • 通过将迭代OAMP算法展开为T层,构建OAMP-Net架构,每层引入可训练参数(γt, θt)以优化估计性能。
  • 采用复MIMO系统的实值表示,使标准深度学习训练可在实数域中进行。
  • 利用QPSK调制的合成训练数据,通过L2损失函数和Adam优化器,采用反向传播优化可训练参数。
  • 网络仅需一次训练即可泛化至时变信道,得益于可训练变量数量极少。
  • 通过利用每层的后验均值估计,实现软判决输出。
  • 网络结构设计在保持OAMP贝叶斯最优性的同时,通过端到端学习提升性能。

实验结果

研究问题

  • RQ1深度学习能否在保持低复杂度的前提下提升OAMP算法在MIMO检测中的性能?
  • RQ2在瑞利衰落MIMO信道中,OAMP-Net的性能与原始OAMP及其他基于深度学习的检测器相比如何?
  • RQ3OAMP-Net是否仅通过一次训练即可泛化至时变信道?
  • RQ4在相关MIMO信道中,OAMP-Net相较于OAMP的性能增益是多少?
  • RQ5可训练参数的数量如何影响训练速度和收敛稳定性?

主要发现

  • 在瑞利衰落信道下,对于4×4 MIMO系统,OAMP-Net在BER = 10⁻³时相比OAMP算法实现约1.37 dB的BER增益。
  • 对于8×8 MIMO系统,OAMP-Net在相同BER水平下相比OAMP实现约2.97 dB的性能增益。
  • 在64×64 MIMO系统中,OAMP-Net仍实现相比OAMP 0.82 dB的增益,证明其可扩展性。
  • 在相关MIMO信道(ρ = 0.5)下,OAMP-Net实现超过2.15 dB的增益,高于独立瑞利信道中的1.86 dB增益。
  • 即使在相关信道下,OAMP-Net仍保持高性能,BER = 10⁻²时仅损失6.05 dB SNR,而OAMP算法损失为6.34 dB。
  • 网络仅用10,000个周期即可完成训练,且可训练参数极少,实现快速且稳定的收敛。

更好的研究,从现在开始

从阅读论文到最终审阅,大幅缩短您的研究时间。

无需绑定信用卡

本解读由 AI 生成,并经人工编辑审核。