A GNN-Based Supervised Learning Framework for Resource Allocation in Wireless IoT Networks

Chen, Tianrui; Zhang, Xinruo; You, Minglei; Zheng, Gan; Lambotharan, Sangarapillai

Abstract

The Internet of Things (IoT) allows physical devices to be connected over the wireless networks. Although device-to-device (D2D) communication has emerged as a promising technology for IoT, the conventional solutions for D2D resource allocation are usually computationally complex and time consuming. The high complexity poses a significant challenge to the practical implementation of wireless IoT networks. A graph neural network (GNN)-based framework is proposed to address this challenge in a supervised manner. Specifically, the wireless network is modeled as a directed graph, where the desirable communication links are modeled as nodes and the harmful interference links are modeled as edges. The effectiveness of the proposed framework is verified via two case studies, namely the link scheduling in D2D networks and the joint channel and power allocation in D2D underlaid cellular networks. Simulation results demonstrate that the proposed framework outperforms the benchmark schemes in terms of the average sum rate and the sample efficiency. In addition, the proposed GNN approach shows potential generalizability to different system settings and robustness to the corrupted input features. It also accelerates the D2D resource optimization by reducing the execution time to only a few milliseconds.

• Journal Article Type: Article
• Acceptance Date: Jun 7, 2021
• Online Publication Date: Jun 22, 2021
• Publication Date: Feb 1, 2022
• Deposit Date: Oct 28, 2021
• Publicly Available Date: Nov 8, 2021
• Journal: IEEE Internet of Things Journal
• Electronic ISSN: 2327-4662
• Publisher: Institute of Electrical and Electronics Engineers (IEEE)
• Peer Reviewed: Peer Reviewed
• Volume: 9
• Issue: 3
• Pages: 1712-1724
• DOI: https://doi.org/10.1109/JIOT.2021.3091551
• Keywords: Computer Networks and Communications; Computer Science Applications; Hardware and Architecture; Information Systems; Signal Processing
• Public URL: https://nottingham-repository.worktribe.com/output/6537494
• Publisher URL: https://ieeexplore.ieee.org/document/9462385
• Additional Information: © 2022 IEEE. Personal use of this material is permitted. Permission from IEEE must be obtained for all other uses, in any current or future media, including reprinting/republishing this material for advertising or promotional purposes, creating new collective works, for resale or redistribution to servers or lists, or reuse of any copyrighted component of this work in other works.