Cross-Validated Cross-Channel Self-Attention and Denoising for Automatic Modulation Classification

TL;DR

This paper introduces a novel deep learning AMC model with cross-channel self-attention and denoising blocks, significantly improving modulation classification accuracy under noisy conditions.

Contribution

It proposes a new architecture combining cross-channel self-attention and residual shrinkage denoising, enhancing robustness in noisy environments for AMC tasks.

Findings

Achieved up to 14% accuracy improvement over benchmarks at low SNRs.

Denoising depth critically affects robustness at low and moderate SNRs.

Cross-validation confirms the model's robustness across diverse modulation types.

Abstract

This study addresses a key limitation in deep learning Automatic Modulation Classification (AMC) models, which perform well at high signal-to-noise ratios (SNRs) but degrade under noisy conditions due to conventional feature extraction suppressing both discriminative structure and interference. The goal was to develop a feature-preserving denoising method that mitigates the loss of modulation class separation. A deep learning AMC model was proposed, incorporating a cross-channel self-attention block to capture dependencies between in-phase and quadrature components, along with dual-path deep residual shrinkage denoising blocks to suppress noise. Experiments using the RML2018.01a dataset employed stratified sampling across 24 modulation types and 26 SNR levels. Results showed that denoising depth strongly influences robustness at low and moderate SNRs. Compared to benchmark models PET-CGDNN, MCLDNN, and DAE, the proposed model achieved notable accuracy improvements across -8 dB to +2 dB SNR, with increases of 3%, 2.3%, and 14%, respectively. Cross-validation confirmed the model's robustness, yielding a mean accuracy of 62.6%, macro precision of 65.8%, macro-recall of 62.6%, and macro-F1 score of 62.9%. The architecture advances interference-aware AMC by formalizing baseband modeling as orthogonal subproblems and introducing cross-channel attention as a generalized complex interaction operator, with ablations confirming the critical role of feature-preserving denoising for robustness at low-to-medium SNR.