An Unsupervised Deep Unrolling Framework for Constrained Optimization Problems in Wireless Networks

In wireless network, the optimization problems generally have complex constraints, and are usually solved via utilizing the traditional optimization methods that have high computational complexity and need to be executed repeatedly with the change of network environments. In this paper, to overcome these shortcomings, an unsupervised deep unrolling framework based on projection gradient descent, i.e., unrolled PGD network (UPGDNet), is designed to solve a family of constrained optimization problems. The set of constraints is divided into two categories according to the coupling relations among optimization variables and the convexity of constraints. One category of constraints includes convex constraints with decoupling among optimization variables, and the other category of constraints includes non-convex or convex constraints with coupling among optimization variables. Then, the first category of constraints is directly projected onto the feasible region, while the second category of constraints is projected onto the feasible region using neural network. Finally, an unrolled sum rate maximization network (USRMNet) is designed based on UPGDNet to solve the weighted SR maximization problem for the multiuser ultra-reliable low latency communication system. Numerical results show that USRMNet has a comparable performance with low computational complexity and an acceptable generalization ability in terms of the user distribution. S. He, S. Xiong, and W. Zhang are with the School of Computer Science and Engineering, Central South University, Changsha 410083, China. S. He is also with the National Mobile Communications Research Laboratory, Southeast University, and the Purple Mountain Laboratories, Nanjing 210096, China. (email: {shiwen.he.hn, shaowen.xiong, csuzwzbn}@csu.edu.cn). Z. An is with the Purple Mountain Laboratories, Nanjing 210096, China. (email: anzhenyu@pmlabs.com.cn). Y. Huang is with the National Mobile Communications Research Laboratory, School of Information Science and Engineering, Southeast University, Nanjing 210096, China. He is also with the Purple Mountain Laboratories, Nanjing 210096, China. (email: huangym@seu.edu.cn). Y. Zhang is with the Department of Computer Science and Technology, Tsinghua University, Beijing 100084, China. (email: zhangyx@tsinghua.edu.cn)