Deep unfolding of the weighted MMSE beamforming algorithm
arXiv:2006.08448
Abstract
Downlink beamforming is a key technology for cellular networks. However, computing the transmit beamformer that maximizes the weighted sum rate subject to a power constraint is an NP-hard problem. As a result, iterative algorithms that converge to a local optimum are used in practice. Among them, the weighted minimum mean square error (WMMSE) algorithm has gained popularity, but its computational complexity and consequent latency has motivated the need for lower-complexity approximations at the expense of performance. Motivated by the recent success of deep unfolding in the trade-off between complexity and performance, we propose the novel application of deep unfolding to the WMMSE algorithm for a MISO downlink channel. The main idea consists of mapping a fixed number of iterations of the WMMSE algorithm into trainable neural network layers, whose architecture reflects the structure of the original algorithm. With respect to traditional end-to-end learning, deep unfolding naturally incorporates expert knowledge, with the benefits of immediate and well-grounded architecture selection, fewer trainable parameters, and better explainability. However, the formulation of the WMMSE algorithm, as described in Shi et al., is not amenable to be unfolded due to a matrix inversion, an eigendecomposition, and a bisection search performed at each iteration. Therefore, we present an alternative formulation that circumvents these operations by resorting to projected gradient descent. By means of simulations, we show that, in most of the settings, the unfolded WMMSE outperforms or performs equally to the WMMSE for a fixed number of iterations, with the advantage of a lower computational load.
References in corpus (20)
- Adam: A Method for Stochastic Optimization
- Explainable Artificial Intelligence: Understanding, Visualizing and Interpreting Deep Learning Models
- Learning to Optimize: Training Deep Neural Networks for Wireless Resource Management
- Deep Learning-Based Communication Over the Air
- Learning to Detect
- AMP-Inspired Deep Networks for Sparse Linear Inverse Problems
- Neural Architecture Optimization
- Model-Driven Deep Learning for MIMO Detection
- A Deep Learning Framework for Optimization of MISO Downlink Beamforming
- Deep Unfolding: Model-Based Inspiration of Novel Deep Architectures
- Trainable ISTA for Sparse Signal Recovery
- Linear Precoding in Cooperative MIMO Cellular Networks with Limited Coordination Clusters
- Fast Converging Algorithm for Weighted Sum Rate Maximization in Multicell MISO Downlink
- Robust Monotonic Optimization Framework for Multicell MISO Systems
- Achieving Global Optimality for Weighted Sum-Rate Maximization in the K-User Gaussian Interference Channel with Multiple Antennas
- Deep Signal Recovery with One-Bit Quantization
- Deep Learning-Aided Trainable Projected Gradient Decoding for LDPC Codes
- Coordinated Linear Precoding in Downlink Multicell MU-MISO OFDMA Networks
- TurboNet: A Model-driven DNN Decoder Based on Max-Log-MAP Algorithm for Turbo Code
- ADMM-based Decoder for Binary Linear Codes Aided by Deep Learning