A Learning Based Branch-and-Bound Algorithm for Single-Group Multicast Beamforming

Zhe Zhang,Meixia Tao
DOI: https://doi.org/10.1109/globecom46510.2021.9686002
2021-01-01
Abstract:Consider the single-group multicast beamforming problem in wireless systems, where a multi-antenna transmitter has a common message intended to a group of users. The problem is known be non-convex and NP-hard except in very few special cases. Traditional optimization-based algorithms either obtain sub-optimal solutions or take exponential complexity to reach the optimum. In this paper, we propose a learning based branch-and-bound (LBB) algorithm to find a near-optimal solution of the quality-of-service (QoS)-constrained multicast beamforming problem with affordable computational complexity. LBB regards the search procedure of the argument-cut based branch-and-bound (AC-BB) algorithm as a sequential decision problem in a binary tree and learns the optimal pruning policy via supervised learning. We propose to apply ensemble learning to train multiple classifiers and then combine them to achieve better performance. To tackle the imbalanced issue, we propose an undersampling-supervised learning method to sample several balanced subsets and train a classifier on each of them. We also show that the computational complexity of LBB is determined by the depth of the binary tree and give its expression in the worst case. Numerical results show that LBB runs significantly faster than AC-BB while achieving nearly the same optimal performance.
What problem does this paper attempt to address?