A Lightweight Denoising Network with TCN–Mamba Fusion for Modulation Classification
Yubo Kong, Yang Ge, Zhengbing Guo · Electronics · 2025
Automatic modulation classification (AMC) under low signal-to-noise ratio (SNR) and complex channel conditions remains a significant challenge due to the trade-off between robustness and efficiency. This study proposes a lightweight temporal convolutional network (TCN) and Mamba fusion architecture designed to enhance modulation recognition performance. In the modulation signal denoising stage, a non-local adaptive thresholding denoising module (NATM) is introduced to explicitly improve the effective signal-to-noise ratio. In the parallel feature extraction stage, TCN captures local symbol-level dependencies, while Mamba models long-range temporal relationships. In the output stage, their outputs are integrated through additive layer-wise fusion, which prevents parameter explosion. Experiments were conducted on the RadioML 2016.10A, 2016.10B, and 2018.01A datasets with leakage-controlled partitioning strategies including GroupKFold and Leave-One-SNR-Out cross-validation. The proposed method achieves up to a 3.8 dB gain in the required signal-to-noise ratio at 90 percent accuracy compared with state-of-the-art baselines, while maintaining a substantially lower parameter count and reduced inference latency. The denoising module provides clear robustness improvements under low signal-to-noise ratio conditions, particularly below −8 dB. The results show that the proposed network strikes a balance between accuracy and efficiency, highlighting its application potential in real-time wireless receivers under resource constraints.