RE-MIMO: Recurrent and Permutation Equivariant Neural MIMO Detection

Abstract

In this paper, we present a novel neural network for MIMO symbol detection. It is motivated by several important considerations in wireless communication systems; permutation equivariance and a variable number of users. The neural detector learns an iterative decoding algorithm that is implemented as a stack of iterative units. Each iterative unit is a neural computation module comprising of 3 sub-modules: the likelihood module, the encoder module, and the predictor module. The likelihood module injects information about the generative (forward) process into the neural network. The encoder-predictor modules together update the state vector and symbol estimates. The encoder module updates the state vector and employs a transformer based attention network to handle the interactions among the users in a permutation equivariant manner. The predictor module refines the symbol estimates. The modular and permutation equivariant architecture allows for dealing with a varying number of users. The resulting neural detector architecture is unique and exhibits several desirable properties unseen in any of the previously proposed neural detectors. We compare its performance against existing methods and the results show the ability of our network to efficiently handle a variable number of transmitters with high accuracy.

arXiv commentcopyright 2020 IEEE TSP. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works

References in corpus (14)

• Adam: A Method for Stochastic Optimization
• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Semi-Supervised Classification with Graph Convolutional Networks
• Massive MIMO for Next Generation Wireless Systems
• Spectral Networks and Locally Connected Networks on Graphs
• FFDNet: Toward a Fast and Flexible Solution for CNN based Image Denoising
• Fifty Years of MIMO Detection: The Road to Large-Scale MIMOs
• Model-Driven Deep Learning for MIMO Detection
• Deep-Learning-based Millimeter-Wave Massive MIMO for Hybrid Precoding
• Deep Learning Based MIMO Communications
• Maximal Sparsity with Deep Networks?
• Recurrent Inference Machines for Solving Inverse Problems
• Invert to Learn to Invert
• Neural Enhanced Belief Propagation on Factor Graphs

Cited by in corpus (5)

• Deep Learning-Aided 6G Wireless Networks: A Comprehensive Survey of Revolutionary PHY Architectures
• Multidimensional Graph Neural Networks for Wireless Communications
• Machine Learning for MU-MIMO Receive Processing in OFDM Systems
• Towards Unified AI Models for MU-MIMO Communications: A Tensor Equivariance Framework
• Graph Neural Networks for Scalable Radio Resource Management: Architecture Design and Theoretical Analysis