Learning to Optimize: Training Deep Neural Networks for Interference Management

Haoran Sun, Xiangyi Chen, Qingjiang Shi, Mingyi Hong, Xiao Fu, Nicholas D. Sidiropoulos

Research output: Contribution to journalArticle

112 Scopus citations

Abstract

Numerical optimization has played a central role in addressing key signal processing (SP) problems. Highly effective methods have been developed for a large variety of SP applications such as communications, radar, filter design, and speech and image analytics, just to name a few. However, optimization algorithms often entail considerable complexity, which creates a serious gap between theoretical design/analysis and real-time processing. In this paper, we aim at providing a new learning-based perspective to address this challenging issue. The key idea is to treat the input and output of an SP algorithm as an unknown nonlinear mapping and use a deep neural network (DNN) to approximate it. If the nonlinear mapping can be learned accurately by a DNN of moderate size, then SP tasks can be performed effectively-since passing the input through a DNN only requires a small number of simple operations. In our paper, we first identify a class of optimization algorithms that can be accurately approximated by a fully connected DNN. Second, to demonstrate the effectiveness of the proposed approach, we apply it to approximate a popular interference management algorithm, namely, the WMMSE algorithm. Extensive experiments using both synthetically generated wireless channel data and real DSL channel data have been conducted. It is shown that, in practice, only a small network is sufficient to obtain high approximation accuracy, and DNNs can achieve orders of magnitude speedup in computational time compared to the state-of-the-art interference management algorithm.

Original languageEnglish (US)
Article number8444648
Pages (from-to)5438-5453
Number of pages16
JournalIEEE Transactions on Signal Processing
Volume66
Issue number20
DOIs
StatePublished - Oct 15 2018

Keywords

  • Optimization algorithms approximation
  • WMMSE algorithm
  • deep neural networks
  • interference management

Fingerprint Dive into the research topics of 'Learning to Optimize: Training Deep Neural Networks for Interference Management'. Together they form a unique fingerprint.

  • Cite this