Efficient Automatic Modulation Classification in Non-Terrestrial Networks with SNN-Based Transformer
Dingyi Zeng,Yichen Xiao,Wanlong Liu,Huilin Du,Enqi Zhang,Dehao Zhang,Yuchen Wang,Malu Zhang,Wenyu Chen
DOI: https://doi.org/10.1109/jiot.2024.3467284
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract:With the development of informatization of IOT devices, non-terrestrial networks (NTN) are becoming more and more important. NTN, including air and space networks, face challenges such as high computational complexity, bandwidth requirements, and memory constraints. An intelligent automatic modulation classification (AMC) mechanism based on neural networks plays a pivotal role in enhancing spectrum efficiency, throughput, and link reliability. Past work in AMC has evolved from likelihood-based and feature-based methods to traditional machine learning techniques and, more recently, to deep neural networks (DNNs). However, existing DNN architectures pose challenges for NTN due to high computational complexity, bandwidth requirements, and memory consumption. Addressing this problems, we proposes a spiking transformer-based model for automatic modulation classification, exploiting temporal dynamics for enhanced performance. Biologically inspired spiking neural networks enable us to exploit the sparse and binarized activation properties of spiking neurons, allowing us to build AMC models with high energy efficiency and high availability that can be used in NTN systems. Furthermore, we introduce a weight binarization method to reduce the model size, which also further reduces the bandwidth and memory requirements of AMC in NTN edge deployment. Experimental results demonstrate the superiority of our approach over state-of-the-art methods, with the binarized model achieving comparable accuracy at a fraction of the size.