Algorithm Unrolling for Massive Access via Deep Neural Network with Theoretical Guarantee

Abstract

Massive access is a critical design challenge of Internet of Things (IoT) networks. In this paper, we consider the grant-free uplink transmission of an IoT network with a multiple-antenna base station (BS) and a large number of single-antenna IoT devices. Taking into account the sporadic nature of IoT devices, we formulate the joint activity detection and channel estimation (JADCE) problem as a group-sparse matrix estimation problem. This problem can be solved by applying the existing compressed sensing techniques, which however either suffer from high computational complexities or lack of algorithm robustness. To this end, we propose a novel algorithm unrolling framework based on the deep neural network to simultaneously achieve low computational complexity and high robustness for solving the JADCE problem. Specifically, we map the original iterative shrinkage thresholding algorithm (ISTA) into an unrolled recurrent neural network (RNN), thereby improving the convergence rate and computational efficiency through end-to-end training. Moreover, the proposed algorithm unrolling approach inherits the structure and domain knowledge of the ISTA, thereby maintaining the algorithm robustness, which can handle non-Gaussian preamble sequence matrix in massive access. With rigorous theoretical analysis, we further simplify the unrolled network structure by reducing the redundant training parameters. Furthermore, we prove that the simplified unrolled deep neural network structures enjoy a linear convergence rate. Extensive simulations based on various preamble signatures show that the proposed unrolled networks outperform the existing methods in terms of the convergence rate, robustness and estimation accuracy.

arXiv comment15 pages, 15 figures, this paper has been submitted to IEEE Transactions on Wireless Communications

References in corpus (14)

• Learning to Optimize: Training Deep Neural Networks for Wireless Resource Management
• Massive Connectivity with Massive MIMO-Part I: Device Activity Detection and Channel Estimation
• Sparse Signal Processing for Grant-Free Massive Connectivity: A Future Paradigm for Random Access Protocols in the Internet of Things
• AMP-Inspired Deep Networks for Sparse Linear Inverse Problems
• Towards Massive Machine Type Communications in Ultra-Dense Cellular IoT Networks: Current Issues and Machine Learning-Assisted Solutions
• Compressive Sensing Based Adaptive Active User Detection and Channel Estimation: Massive Access Meets Massive MIMO
• Sparse Activity Detection for Massive Connectivity
• Learning Optimal Resource Allocations in Wireless Systems
• Sparse Representation for Wireless Communications: A Compressive Sensing Approach
• Trainable ISTA for Sparse Signal Recovery
• Efficient High-Dimensional Inference in the Multiple Measurement Vector Problem
• Theoretical Linear Convergence of Unfolded ISTA and its Practical Weights and Thresholds
• Dynamic Screening: Accelerating First-Order Algorithms for the Lasso and Group-Lasso
• Learning to Continuously Optimize Wireless Resource In Episodically Dynamic Environment