MTL-SRN: Multi-task Learning-based Signal Recognition Network

Wideband signal recognition is a crucial task in the cognitive wireless communication, involving accurate classification of different signal types, modulation types, center frequencies, etc. However, most conventional approaches treat the recognition of different parameters as multiple independent tasks, and often face performance bottlenecks due to the complexity and diversity of wideband signals. To overcome these challenges, we propose a multi-task learning (MTL) network that integrates multiple tasks of signal recognition into an end-to-end model to accomplish spectrum sensing, modulation recognition, and signal classification simultaneously. By employing a shared feature extraction network and a multi-task classification header, the proposed framework effectively captures the correlations and shared information among different tasks, thereby enhancing overall recognition performance. To validate the effectiveness of the proposed scheme, we compare its performance with other state-of-the-art recognition and classification networks. Experimental results demonstrate the significant performance of the proposed MTL network in spectrum sensing, modulation recognition, and signal classification tasks.