Transformer Based Approach for Wireless Resource Allocation Problems Involving Mixed Discrete and Continuous Variables

Bingqing Song, Zhicheng Zhou, Chenliang Li, Dongning Guo, Xiao Fu, Mingyi Hong

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Many challenging wireless communication problems involve jointly optimizing a set of discrete variables (e.g., subset of antennas) and continuous variables (e.g., power allocation), where the subproblem involving discrete variables are intrinsically combinatorial. Although many heuristic methods have been developed to deal with these problems (e.g., greedy based, exhaustive search based methods), they still incur high computational costs. In this work, we propose a machine learning-based algorithm to learn an approximate high-quality solution for this class of problems. Differently than the existing learning-based methods which mostly only focusing on continuous problems, we propose a two-stage approach, where in the first stage a Transformer is used to find the set of discrete variables, followed by a second stage where the continuous variables are optimized (while fixing the dicrete variables). We demonstrate the effectiveness of our approach using a joint user scheduling and beamforming problem in MIMO systems. We show that the proposed method can generate high-quality active user sets, even with low-quality channel state information, while only using a fraction of computational time compared with a heuristic greedy algorithm.

Original languageEnglish (US)
Title of host publication2023 IEEE 24th International Workshop on Signal Processing Advances in Wireless Communications, SPAWC 2023 - Proceedings
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages636-640
Number of pages5
ISBN (Electronic)9781665496261
DOIs
StatePublished - 2023
Externally publishedYes
Event24th IEEE International Workshop on Signal Processing Advances in Wireless Communications, SPAWC 2023 - Shanghai, China
Duration: Sep 25 2023Sep 28 2023

Publication series

NameIEEE Workshop on Signal Processing Advances in Wireless Communications, SPAWC

Conference

Conference24th IEEE International Workshop on Signal Processing Advances in Wireless Communications, SPAWC 2023
Country/TerritoryChina
CityShanghai
Period9/25/239/28/23

Bibliographical note

Publisher Copyright:
© 2023 IEEE.

Keywords

  • Machine learning
  • Resource allocation
  • User Selection

Fingerprint

Dive into the research topics of 'Transformer Based Approach for Wireless Resource Allocation Problems Involving Mixed Discrete and Continuous Variables'. Together they form a unique fingerprint.

Cite this