Automatic Modulation Classification Scheme Based on LSTM With Random Erasing and Attention Mechanism
Automatic modulation classification (AMC) is a key technology of cognitive radio used in non-cooperative communication. Recently, deep learning has been applied to AMC tasks. In this paper, an AMC scheme based on deep learning is proposed, which combines random erasing and attention mechanism to ach...
| Main Authors: | , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
IEEE
2020-01-01
|
| Series: | IEEE Access |
| Subjects: | |
| Online Access: | https://ieeexplore.ieee.org/document/9170506/ |
| _version_ | 1819132741601460224 |
|---|---|
| author | Yufan Chen Wei Shao Jin Liu Lu Yu Zuping Qian |
| author_facet | Yufan Chen Wei Shao Jin Liu Lu Yu Zuping Qian |
| author_sort | Yufan Chen |
| collection | DOAJ |
| description | Automatic modulation classification (AMC) is a key technology of cognitive radio used in non-cooperative communication. Recently, deep learning has been applied to AMC tasks. In this paper, an AMC scheme based on deep learning is proposed, which combines random erasing and attention mechanism to achieve high classification accuracy. Firstly, we propose two data augmentation methods, random erasing at sample level and random erasing at amplitude/phase (AP) channel level. The former replaces training samples with noise information, while the latter replaces AP channel information of training samples with noise information. Erased data segments are randomly stitched to enable training data expansion. Training data of different qualities enables deep learning model to have stronger generalization capability and higher robustness. Then, we propose a single-layer Long Short-Term Memory (LSTM) model based on attention mechanism. In the first part of this model, we propose the signal embedding, which enables the input to contain modulation information more comprehensively and accurately. Then hidden state output by LSTM is input into the attention module, and weighting is applied to the hidden state to help the LSTM model capture the temporal features of modulated signals. Compared to a model without attention mechanism, this model has faster convergence speed and better classification performance. Lastly, we propose a random erasing-based test time augmentation (RE-TTA) method. Test data is randomly erased for multiple times and classification results are comprehensively evaluated, in order to further improve classification accuracy. Experimental results on dataset RML2016.10a show that classification accuracy of the proposed scheme is competitive compared with the state-of-the-art methods. |
| first_indexed | 2024-12-22T09:36:13Z |
| format | Article |
| id | doaj.art-f1ab5f0b7b6645f4ab4c0850b9e24bf9 |
| institution | Directory Open Access Journal |
| issn | 2169-3536 |
| language | English |
| last_indexed | 2024-12-22T09:36:13Z |
| publishDate | 2020-01-01 |
| publisher | IEEE |
| record_format | Article |
| series | IEEE Access |
| spelling | doaj.art-f1ab5f0b7b6645f4ab4c0850b9e24bf92022-12-21T18:30:51ZengIEEEIEEE Access2169-35362020-01-01815429015430010.1109/ACCESS.2020.30176419170506Automatic Modulation Classification Scheme Based on LSTM With Random Erasing and Attention MechanismYufan Chen0https://orcid.org/0000-0003-2064-7541Wei Shao1https://orcid.org/0000-0002-8248-869XJin Liu2https://orcid.org/0000-0001-7082-2163Lu Yu3Zuping Qian4https://orcid.org/0000-0001-8713-157XCollege of Communications Engineering, Army Engineering University of PLA, Nanjing, ChinaCollege of Communications Engineering, Army Engineering University of PLA, Nanjing, ChinaSchool of Remote Sensing and Information Engineering, Wuhan University, Wuhan, ChinaCollege of Communications Engineering, Army Engineering University of PLA, Nanjing, ChinaCollege of Communications Engineering, Army Engineering University of PLA, Nanjing, ChinaAutomatic modulation classification (AMC) is a key technology of cognitive radio used in non-cooperative communication. Recently, deep learning has been applied to AMC tasks. In this paper, an AMC scheme based on deep learning is proposed, which combines random erasing and attention mechanism to achieve high classification accuracy. Firstly, we propose two data augmentation methods, random erasing at sample level and random erasing at amplitude/phase (AP) channel level. The former replaces training samples with noise information, while the latter replaces AP channel information of training samples with noise information. Erased data segments are randomly stitched to enable training data expansion. Training data of different qualities enables deep learning model to have stronger generalization capability and higher robustness. Then, we propose a single-layer Long Short-Term Memory (LSTM) model based on attention mechanism. In the first part of this model, we propose the signal embedding, which enables the input to contain modulation information more comprehensively and accurately. Then hidden state output by LSTM is input into the attention module, and weighting is applied to the hidden state to help the LSTM model capture the temporal features of modulated signals. Compared to a model without attention mechanism, this model has faster convergence speed and better classification performance. Lastly, we propose a random erasing-based test time augmentation (RE-TTA) method. Test data is randomly erased for multiple times and classification results are comprehensively evaluated, in order to further improve classification accuracy. Experimental results on dataset RML2016.10a show that classification accuracy of the proposed scheme is competitive compared with the state-of-the-art methods.https://ieeexplore.ieee.org/document/9170506/Automatic modulation classificationrandom erasinglong short-term memoryattention mechanism |
| spellingShingle | Yufan Chen Wei Shao Jin Liu Lu Yu Zuping Qian Automatic Modulation Classification Scheme Based on LSTM With Random Erasing and Attention Mechanism IEEE Access Automatic modulation classification random erasing long short-term memory attention mechanism |
| title | Automatic Modulation Classification Scheme Based on LSTM With Random Erasing and Attention Mechanism |
| title_full | Automatic Modulation Classification Scheme Based on LSTM With Random Erasing and Attention Mechanism |
| title_fullStr | Automatic Modulation Classification Scheme Based on LSTM With Random Erasing and Attention Mechanism |
| title_full_unstemmed | Automatic Modulation Classification Scheme Based on LSTM With Random Erasing and Attention Mechanism |
| title_short | Automatic Modulation Classification Scheme Based on LSTM With Random Erasing and Attention Mechanism |
| title_sort | automatic modulation classification scheme based on lstm with random erasing and attention mechanism |
| topic | Automatic modulation classification random erasing long short-term memory attention mechanism |
| url | https://ieeexplore.ieee.org/document/9170506/ |
| work_keys_str_mv | AT yufanchen automaticmodulationclassificationschemebasedonlstmwithrandomerasingandattentionmechanism AT weishao automaticmodulationclassificationschemebasedonlstmwithrandomerasingandattentionmechanism AT jinliu automaticmodulationclassificationschemebasedonlstmwithrandomerasingandattentionmechanism AT luyu automaticmodulationclassificationschemebasedonlstmwithrandomerasingandattentionmechanism AT zupingqian automaticmodulationclassificationschemebasedonlstmwithrandomerasingandattentionmechanism |