MobileRaT: A Lightweight Radio Transformer Method for Automatic Modulation Classification in Drone Communication Systems

Nowadays, automatic modulation classification (AMC) has become a key component of next-generation drone communication systems, which are crucial for improving communication efficiency in non-cooperative environments. The contradiction between the accuracy and efficiency of current methods hinders th...

Full description

Bibliographic Details
Main Authors: Qinghe Zheng, Xinyu Tian, Zhiguo Yu, Yao Ding, Abdussalam Elhanashi, Sergio Saponara, Kidiyo Kpalma
Format: Article
Language:English
Published: MDPI AG 2023-09-01
Series:Drones
Subjects:
Online Access:https://www.mdpi.com/2504-446X/7/10/596
_version_ 1797574155858608128
author Qinghe Zheng
Xinyu Tian
Zhiguo Yu
Yao Ding
Abdussalam Elhanashi
Sergio Saponara
Kidiyo Kpalma
author_facet Qinghe Zheng
Xinyu Tian
Zhiguo Yu
Yao Ding
Abdussalam Elhanashi
Sergio Saponara
Kidiyo Kpalma
author_sort Qinghe Zheng
collection DOAJ
description Nowadays, automatic modulation classification (AMC) has become a key component of next-generation drone communication systems, which are crucial for improving communication efficiency in non-cooperative environments. The contradiction between the accuracy and efficiency of current methods hinders the practical application of AMC in drone communication systems. In this paper, we propose a real-time AMC method based on the lightweight mobile radio transformer (MobileRaT). The constructed radio transformer is trained iteratively, accompanied by pruning redundant weights based on information entropy, so it can learn robust modulation knowledge from multimodal signal representations for the AMC task. To the best of our knowledge, this is the first attempt in which the pruning technique and a lightweight transformer model are integrated and applied to processing temporal signals, ensuring AMC accuracy while also improving its inference efficiency. Finally, the experimental results—by comparing MobileRaT with a series of state-of-the-art methods based on two public datasets—have verified its superiority. Two models, MobileRaT-A and MobileRaT-B, were used to process RadioML 2018.01A and RadioML 2016.10A to achieve average AMC accuracies of 65.9% and 62.3% and the highest AMC accuracies of 98.4% and 99.2% at +18 dB and +14 dB, respectively. Ablation studies were conducted to demonstrate the robustness of MobileRaT to hyper-parameters and signal representations. All the experimental results indicate the adaptability of MobileRaT to communication conditions and that MobileRaT can be deployed on the receivers of drones to achieve air-to-air and air-to-ground cognitive communication in less demanding communication scenarios.
first_indexed 2024-03-10T21:18:51Z
format Article
id doaj.art-aa3e9c865528419ab76b940cf8b54bc3
institution Directory Open Access Journal
issn 2504-446X
language English
last_indexed 2024-03-10T21:18:51Z
publishDate 2023-09-01
publisher MDPI AG
record_format Article
series Drones
spelling doaj.art-aa3e9c865528419ab76b940cf8b54bc32023-11-19T16:15:09ZengMDPI AGDrones2504-446X2023-09-0171059610.3390/drones7100596MobileRaT: A Lightweight Radio Transformer Method for Automatic Modulation Classification in Drone Communication SystemsQinghe Zheng0Xinyu Tian1Zhiguo Yu2Yao Ding3Abdussalam Elhanashi4Sergio Saponara5Kidiyo Kpalma6School of Intelligent Engineering, Shandong Management University, Jinan 250357, ChinaSchool of Intelligent Engineering, Shandong Management University, Jinan 250357, ChinaSchool of Intelligent Engineering, Shandong Management University, Jinan 250357, ChinaKey Laboratory of Optical Engineering, Xi’an Research Institute of High Technology, Xi’an 710025, ChinaDepartment of Information Engineering, University of Pisa, 56122 Pisa, ItalyDepartment of Information Engineering, University of Pisa, 56122 Pisa, ItalyDepartment of Electronics and Industrial Informatics, National Institute for Applied Sciences of Rennes, F-35000 Rennes, FranceNowadays, automatic modulation classification (AMC) has become a key component of next-generation drone communication systems, which are crucial for improving communication efficiency in non-cooperative environments. The contradiction between the accuracy and efficiency of current methods hinders the practical application of AMC in drone communication systems. In this paper, we propose a real-time AMC method based on the lightweight mobile radio transformer (MobileRaT). The constructed radio transformer is trained iteratively, accompanied by pruning redundant weights based on information entropy, so it can learn robust modulation knowledge from multimodal signal representations for the AMC task. To the best of our knowledge, this is the first attempt in which the pruning technique and a lightweight transformer model are integrated and applied to processing temporal signals, ensuring AMC accuracy while also improving its inference efficiency. Finally, the experimental results—by comparing MobileRaT with a series of state-of-the-art methods based on two public datasets—have verified its superiority. Two models, MobileRaT-A and MobileRaT-B, were used to process RadioML 2018.01A and RadioML 2016.10A to achieve average AMC accuracies of 65.9% and 62.3% and the highest AMC accuracies of 98.4% and 99.2% at +18 dB and +14 dB, respectively. Ablation studies were conducted to demonstrate the robustness of MobileRaT to hyper-parameters and signal representations. All the experimental results indicate the adaptability of MobileRaT to communication conditions and that MobileRaT can be deployed on the receivers of drones to achieve air-to-air and air-to-ground cognitive communication in less demanding communication scenarios.https://www.mdpi.com/2504-446X/7/10/596drone communicationsnon-cooperative communicationscognitive radioautomatic modulation classification (AMC)deep learninglightweight transformer
spellingShingle Qinghe Zheng
Xinyu Tian
Zhiguo Yu
Yao Ding
Abdussalam Elhanashi
Sergio Saponara
Kidiyo Kpalma
MobileRaT: A Lightweight Radio Transformer Method for Automatic Modulation Classification in Drone Communication Systems
Drones
drone communications
non-cooperative communications
cognitive radio
automatic modulation classification (AMC)
deep learning
lightweight transformer
title MobileRaT: A Lightweight Radio Transformer Method for Automatic Modulation Classification in Drone Communication Systems
title_full MobileRaT: A Lightweight Radio Transformer Method for Automatic Modulation Classification in Drone Communication Systems
title_fullStr MobileRaT: A Lightweight Radio Transformer Method for Automatic Modulation Classification in Drone Communication Systems
title_full_unstemmed MobileRaT: A Lightweight Radio Transformer Method for Automatic Modulation Classification in Drone Communication Systems
title_short MobileRaT: A Lightweight Radio Transformer Method for Automatic Modulation Classification in Drone Communication Systems
title_sort mobilerat a lightweight radio transformer method for automatic modulation classification in drone communication systems
topic drone communications
non-cooperative communications
cognitive radio
automatic modulation classification (AMC)
deep learning
lightweight transformer
url https://www.mdpi.com/2504-446X/7/10/596
work_keys_str_mv AT qinghezheng mobileratalightweightradiotransformermethodforautomaticmodulationclassificationindronecommunicationsystems
AT xinyutian mobileratalightweightradiotransformermethodforautomaticmodulationclassificationindronecommunicationsystems
AT zhiguoyu mobileratalightweightradiotransformermethodforautomaticmodulationclassificationindronecommunicationsystems
AT yaoding mobileratalightweightradiotransformermethodforautomaticmodulationclassificationindronecommunicationsystems
AT abdussalamelhanashi mobileratalightweightradiotransformermethodforautomaticmodulationclassificationindronecommunicationsystems
AT sergiosaponara mobileratalightweightradiotransformermethodforautomaticmodulationclassificationindronecommunicationsystems
AT kidiyokpalma mobileratalightweightradiotransformermethodforautomaticmodulationclassificationindronecommunicationsystems