Multi-Scale Feature Fusion and Distribution Similarity Network for Few-Shot Automatic Modulation Classification
Haoyue Tan,Zhenxi Zhang,Yu Li,Xiaoran Shi,Feng Zhou
DOI: https://doi.org/10.1109/lsp.2024.3470762
2024-01-01
IEEE Signal Processing Letters
Abstract:Automatic modulation classification (AMC), as a key technology of cognitive radio, has become a focal point of research. However, most deep learning-based AMC methods require an extensive number of labeled signals to acquire a comprehensive understanding of modulation types, placing substantial pressure on signal acquisition and labeling. To solve this issue, we propose a few-shot AMC (FSAMC) method to facilitate rapid generalization and recognition with limited data, namely multi-scale feature fusion and distribution similarity network (MS2F-DS). Firstly, we design a multi-scale feature fusion (MS2F) model, which aims to extract features with varying fields of view and boost feature fusion, enabling the derivation of contextual information from the signal. Furthermore, we introduce a distribution similarity (DS) classifier to address the insufficient measurement of current similarity measurement functions by considering both micro and macro perspectives of vectors, further increasing intra-class compactness and inter-class separability. Finally, extensive experiments were conducted on 3-way 1, 3, and 5-shot FSAMC tasks using public datasets RML2016.10a and RML2016.10b, and the results demonstrated the effectiveness of our method.