Learning to Branch: Accelerating Resource Allocation in Wireless Networks

Abstract

Resource allocation in wireless networks, such as device-to-device (D2D) communications, is usually formulated as mixed integer nonlinear programming (MINLP) problems, which are generally NP-hard and difficult to get the optimal solutions. Traditional methods to solve these MINLP problems are all based on mathematical optimization techniques, such as the branch-and-bound (B&B) algorithm that converges slowly and has forbidding complexity for real-time implementation. Therefore, machine leaning (ML) has been used recently to address the MINLP problems in wireless communications. In this paper, we use imitation learning method to accelerate the B&B algorithm. With invariant problem-independent features and appropriate problem-dependent feature selection for D2D communications, a good auxiliary prune policy can be learned in a supervised manner to speed up the most time-consuming branch process of the B&B algorithm. Moreover, we develop a mixed training strategy to further reinforce the generalization ability and a deep neural network (DNN) with a novel loss function to achieve better dynamic control over optimality and computational complexity. Extensive simulation demonstrates that the proposed method can achieve good optimality and reduce computational complexity simultaneously.

to appear in IEEE Transactions on Vehicular Technology

References in corpus (4)

• A Survey on Device-to-Device Communication in Cellular Networks
• Learning to Optimize: Training Deep Neural Networks for Wireless Resource Management
• Multi-Agent Deep Reinforcement Learning for Dynamic Power Allocation in Wireless Networks
• Spatial Deep Learning for Wireless Scheduling

Cited by in corpus (5)

• Compacting Deep Neural Networks for Internet of Things: Methods and Applications
• Optimal Solutions for Joint Beamforming and Antenna Selection: From Branch and Bound to Graph Neural Imitation Learning
• Diffusion Models as Network Optimizers: Explorations and Analysis
• Programming Wireless Security through Learning-Aided Spatiotemporal Digital Coding Metamaterial Antenna
• Computational Offloading in Semantic-Aware Cloud-Edge-End Collaborative Networks