GCEPNet: Graph Convolution-Enhanced Expectation Propagation for Massive MIMO Detection
Abstract: Massive MIMO (multiple-input multiple-output) detection is an important topic in wireless communication and various machine learning based methods have been developed recently for this task. Expectation Propagation (EP) and its variants are widely used for MIMO detection and have achieved the best performance. However, EP-based solvers fail to capture the correlation between unknown variables, leading to a loss of information, and in addition, they are computationally expensive. In this paper, we show that the real-valued system can be modeled as spectral signal convolution on graph, through which the correlation between unknown variables can be captured. Based on such analysis, we propose graph convolution-enhanced expectation propagation (GCEPNet). GCEPNet incorporates data-dependent attention scores into Chebyshev polynomial for powerful graph convolution with better generalization capacity. It enables a better estimation of the cavity distribution for EP and empirically achieves the state-of-the-art (SOTA) MIMO detection performance with much faster inference speed. To our knowledge, we are the first to shed light on the connection between the system model and graph convolution, and the first to design the data-dependent coefficients for graph convolution.
- N. Samuel, T. Diskin, and A. Wiesel, “Learning to detect,” IEEE Transactions on Signal Processing, vol. 67, no. 10, pp. 2554–2564, 2019.
- S. Verdú, “Computational complexity of optimum multiuser detection,” Algorithmica, vol. 4, no. 1, pp. 303–312, 1989.
- N. T. Nguyen, K. Lee, and H. DaiIEEE, “Application of deep learning to sphere decoding for large MIMO systems,” IEEE Transactions on Wireless Communications, vol. 20, no. 10, pp. 6787–6803, 2021.
- L. V. Nguyen, N. T. Nguyen, N. H. Tran, M. Juntti, A. L. Swindlehurst, and D. H. N. Nguyen, “Leverageing deep neural networks for massive MIMO data detection,” IEEE Wireless Communications, pp. 1–7, 2022.
- X.-W. Chang, Z. Chen, and J. Wen, “An extended Babai method for estimating linear model based integer parameters,” ECONOMETRICS AND STATISTICS, 2022.
- J. Wen and X.-W. Chang, “Success probability of the Babai estimators for box-constrained integer linear models,” IEEE Trans. Inf. Theory, vol. 63, pp. 631–648, 2017.
- X.-W. Chang, Z. Chen, and Y. Xu, “On the randomized Babai point,” in IEEE International Symposium on Information Theory, 2020, pp. 1195–1200.
- J. Cespedes, P. M. Olmos, M. Sánchez-Fernández, and F. Perez-Cruz, “Expectation propagation detection for high-order high-dimensional MIMO systems,” IEEE Transactions on Communications, vol. 62, no. 8, pp. 2840–2849, 2014.
- D. L. Donoho, A. Maleki, and A. Montanari, “Message-passing algorithms for compressed sensing,” Proceedings of the National Academy of Sciences, vol. 106, no. 45, pp. 18 914–18 919, 2009.
- A. Scotti, N. N. Moghadam, D. Liu, K. Gafvert, and J. Huang, “Graph neural networks for massive MIMO detection,” arXiv preprint arXiv:2007.05703, 2020.
- K. Pratik, B. D. Rao, and M. Welling, “RE-MIMO: Recurrent and permutation equivariant neural MIMO detection,” IEEE Transactions on Signal Processing, vol. 69, pp. 459–473, 2020.
- H. He, C.-K. Wen, S. Jin, and G. Y. Li, “A model-driven deep learning network for MIMO detection,” in 2018 IEEE Global Conference on Signal and Information Processing (GlobalSIP). IEEE, 2018, pp. 584–588.
- N. Samuel, T. Diskin, and A. Wiesel, “Deep MIMO detection,” in 2017 IEEE 18th International Workshop on Signal Processing Advances in Wireless Communications (SPAWC). IEEE, 2017, pp. 1–5.
- N. T. Nguyen and K. Lee, “Deep learning-aided tabu search detection for large MIMO systems,” IEEE Transactions on Wireless Communications, vol. 19, no. 6, pp. 4262–4275, 2020.
- A. Kosasih, V. Onasis, W. Hardjawana, V. Miloslavskaya, V. Andrean, J.-S. Leu, and B. Vucetic, “Graph neural network aided expectation propagation detector for MU-MIMO systems,” in 2022 IEEE Wireless Communications and Networking Conference (WCNC). IEEE, 2022, pp. 1212–1217.
- A. Kosasih, V. Onasis, V. Miloslavskaya, W. Hardjawana, V. Andrean, and B. Vucetic, “Graph neural network aided MU-MIMO detectors,” IEEE Journal on Selected Areas in Communications, vol. 40, no. 9, pp. 2540–2555, 2022.
- H. He, A. Kosasih, X. Yu, J. Zhang, S. Song, W. Hardjawana, and K. B. Letaief, “GNN-enhanced approximate message passing for massive/ultra-massive MIMO detection,” in 2023 IEEE Wireless Communications and Networking Conference (WCNC). IEEE, 2023, pp. 1–6.
- Z. Liu, D. He, N. Wu, Q. Yan, and Y. Li, “Model-driven IEP-GNN framework for MIMO detection with bayesian optimization,” IEEE Wireless Communications Letters, 2023.
- A. Kosasih, V. Miloslavskaya, W. Hardjawana, C. She, C.-K. Wen, and B. Vucetic, “A bayesian receiver with improved complexity-reliability trade-off in massive MIMO systems,” IEEE Transactions on Communications, vol. 69, no. 9, pp. 6251–6266, 2021.
- K. Yoon, R. Liao, Y. Xiong, L. Zhang, E. Fetaya, R. Urtasun, R. Zemel, and X. Pitkow, “Inference in probabilistic graphical models by graph neural networks,” in 2019 53rd Asilomar Conference on Signals, Systems, and Computers. IEEE, 2019, pp. 868–875.
- M. Defferrard, X. Bresson, and P. Vandergheynst, “Convolutional neural networks on graphs with fast localized spectral filtering,” In Proceedings of the 30th International Conference on Neural Information Processing Systems, 2016.
- S. Luan, C. Hua, Q. Lu, J. Zhu, M. Zhao, S. Zhang, X.-W. Chang, and D. Precup, “Revisiting heterophily for graph neural networks,” Advances in neural information processing systems, vol. 35, pp. 1362–1375, 2022.
- Q. Lu, J. Zhu, S. Luan, and X.-W. Chang, “Representation learning on heterophilic graph with directional neighborhood attention,” arXiv preprint arXiv:2403.01475, 2024.
- S. Luan, M. Zhao, X.-W. Chang, and D. Precup, “Break the ceiling: Stronger multi-scale deep graph convolutional networks,” Advances in neural information processing systems, vol. 32, 2019.
- K. Cho, B. Van Merriënboer, C. Gulcehre, D. Bahdanau, F. Bougares, H. Schwenk, and Y. Bengio, “Learning phrase representations using rnn encoder-decoder for statistical machine translation,” arXiv preprint arXiv:1406.1078, 2014.
Paper Prompts
Sign up for free to create and run prompts on this paper using GPT-5.
Top Community Prompts
Collections
Sign up for free to add this paper to one or more collections.