A Generic Layer Pruning Method for Signal Modulation Recognition Deep Learning Models
With the successful application of deep learning in communications systems, deep neural networks are becoming the preferred method for Automatic Modulation Recognition (AMR). Although these AMR models yield impressive results, they often come with high computational complexity and large model sizes, which hinders their practical deployment in communication systems. To address this challenge, we propose a novel layer pruning method, PSR. Specifically, we decompose the AMR model into several consecutive blocks, each containing consecutive layers with similar semantics. Then, we identify layers that need to be preserved within each block based on their contribution. Finally, we reassemble the pruned blocks and fine-tune the compact model. Extensive experiments on five datasets demonstrate the efficiency and effectiveness of PSR over a variety of state-of-the-art baselines, including layer pruning and channel pruning methods.
Authors
- Qi Xuan (ORCID: https://orcid.org/0000-0002-1042-470X)
- Dongwei Xu (ORCID: https://orcid.org/0000-0003-2693-922X)
- Xiaoniu Yang (ORCID: https://orcid.org/0000-0003-3117-2211)
- Yun Lin (ORCID: https://orcid.org/0000-0002-4002-1282)
- Yutao Zhu
- Yuqi Li (ORCID: https://orcid.org/0009-0002-7314-6863)
- Yao Lu (ORCID: https://orcid.org/0000-0003-0655-7814)
Institutions
- Harbin Engineering University (CN)
- Chinese Academy of Sciences (CN)
- Institute of Computing Technology (CN)
- National Engineering Research Center of Electromagnetic Radiation Control Materials (CN)
- China Information Technology Security Evaluation Center (CN)
- Artificial Intelligence in Medicine (Canada) (CA)
- Zhejiang University of Technology (CN)
Publication Details
- Journal
- IEEE Transactions on Cognitive Communications and Networking
- Published
- 2024-12-23
- DOI
- https://doi.org/10.1109/tccn.2024.3520958
- Citations
- 16
- Primary Topic
- Wireless Signal Modulation Classification
- Type
- article
- Field-Weighted Citation Impact
- 5.56
Funders
- National Natural Science Foundation of China