GCEPNet: Graph Convolution-Enhanced Expectation Propagation for Massive MIMO Detection

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Lu, Qincheng, Luan, Sitao, Chang, Xiao-Wen
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909305516589056
author Lu, Qincheng
Luan, Sitao
Chang, Xiao-Wen
author_facet Lu, Qincheng
Luan, Sitao
Chang, Xiao-Wen
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2404_14886
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle GCEPNet: Graph Convolution-Enhanced Expectation Propagation for Massive MIMO Detection
Lu, Qincheng
Luan, Sitao
Chang, Xiao-Wen
Machine Learning
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.
title GCEPNet: Graph Convolution-Enhanced Expectation Propagation for Massive MIMO Detection
topic Machine Learning
url https://arxiv.org/abs/2404.14886