用图神经网络改进深度干扰消除,减少参数和样本需求。
Data-Driven Deep MIMO Detection:Network Architectures and Generalization Analysis
- 构建基于图的MLP网络,实现用户间共享参数的消息传递。
- 相比DeepSIC,参数量减少且训练样本更少,性能相当或更好。
- 理论证明迭代次数对泛化能力的影响被消除,适合小样本场景。
在实际多用户多输入多输出(MU-MIMO)系统中,符号检测因严重的用户间干扰和信道状态信息(CSI)不确定性而面临挑战。与计算复杂度高的基于信念传播的模型驱动方法不同,软干扰抵消(SIC)在性能与复杂度之间取得良好平衡。为进一步应对CSI失配和非线性效应,近期提出的数据驱动深度神经接收机(如DeepSIC)利用深度神经网络进行干扰抵消与符号检测,展现出优异的实证性能。然而,目前尚缺乏关于DeepSIC在训练样本数量有限时泛化能力的理论依据。本文提出将全数据驱动的DeepSIC检测置于由外层和内层有向无环图(DAG)连接的多层MLP组成的网络架构中,将DeepSIC升级为基于图神经网络(GNN)的消息传递过程,称为GNNSIC,其用户与迭代间共享模型参数。值得注意的是,GNNSIC在可比表达能力下显著减少可训练参数,提升样本效率并增强用户泛化能力。通过基于雷达马彻复杂度的范数泛化分析,我们揭示由于参数共享,GNNSIC可消除DeepSIC对迭代次数的指数依赖关系。仿真结果表明,GNNSIC在显著减少参数和训练样本的前提下,达到与DeepSIC相当或更优的符号误码率(SER)性能。
原文摘要 · Abstract (English)
In practical Multiuser Multiple-Input Multiple-Output (MU-MIMO) systems, symbol detection remains challenging due to severe inter-user interference and sensitivity to Channel State Information (CSI) uncertainty. In contrast to the mostly studied belief propagation-type model-driven methods, which incur high computational complexity, Soft Interference Cancellation (SIC) strikes a good balance between performance and complexity. To further address CSI mismatch and nonlinear effects, the recently proposed data-driven deep neural receivers, such as DeepSIC, leverage the advantages of deep neural networks for interference cancellation and symbol detection, demonstrating strong empirical performance. However, there is still a lack of theoretical underpinning for why and to what extent DeepSIC could generalize with the number of training samples. This paper proposes inspecting the fully data-driven DeepSIC detection within a Network-of-MLPs architecture, which is composed of multiple interconnected MLPs via outer and inner Directed Acyclic Graphs (DAGs). Within such an architecture, DeepSIC can be upgraded as a graph-based message-passing process using Graph Neural Networks (GNNs), termed GNNSIC, with shared model parameters across users and iterations. Notably, GNNSIC achieves excellent expressivity comparable to DeepSIC with substantially fewer trainable parameters, resulting in improved sample efficiency and enhanced user generalization. By conducting a norm-based generalization analysis using Rademacher complexity, we reveal that an exponential dependence on the number of iterations for DeepSIC can be eliminated in GNNSIC due to parameter sharing. Simulation results demonstrate that GNNSIC attains comparable or improved Symbol Error Rate (SER) performance to DeepSIC with significantly fewer parameters and training samples.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。