Joint Multidimensional Pattern for Spectrum Prediction Using GNN

Abstract

In general, judging the use/idle state of the wireless spectrum is the foundation for cognitive radio users (secondary users, SUs) to access limited spectrum resources efficiently. Rich information can be mined by the inherent correlation of electromagnetic spectrum data from SUs in time, frequency, space, and other dimensions. Therefore, how to efficiently use the spectrum status of each SU implementation of reception multidimensional combination forecasting is the core of this paper. In this paper, we propose a deep-learning hybrid model called TensorGCN-LSTM based on the tensor data structure. The model treats SUs deployed at different spatial locations under the same frequency, and the spectrum status of SUs themselves under different frequencies in the task area as nodes and constructs two types of graph structures. Graph convolutional operations are used to sequentially extract corresponding spatial-domain and frequency-domain features from the two types of graph structures. Then, the long short-term memory (LSTM) model is used to fuse the spatial, frequency, and temporal features of the cognitive radio environment data. Finally, the prediction task of the spectrum distribution situation is accomplished through fully connected layers. Specifically, the model constructs a tensor graph based on the spatial similarity of SUs’ locations and the frequency correlation between different frequency signals received by SUs, which describes the electromagnetic wave’s dependency relationship in spatial and frequency domains. LSTM is used to capture the electromagnetic wave’s dependency relationship in the temporal domain. To evaluate the effectiveness of the model, we conducted ablation experiments on LSTM, GCN, GC-LSTM, and TensorGCN-LSTM models using simulated data. The experimental results showed that our model achieves better prediction performance in RMSE, and the correlation coefficient R2 of 0.8753 also confirms the feasibility of the model.