WVDNet: Time-Frequency Analysis Via Semi-Supervised Learning

Naihao Liu,Jingyu Wang,Yang,Zhen Li,Jinghuai Gao
DOI: https://doi.org/10.1109/lsp.2023.3235646
2023-01-01
IEEE Signal Processing Letters
Abstract:The bilinear based method is one of the commonly used tools in time-frequency analysis (TFA) fields. However, it suffers from the trade-off of high resolution and cross-term interference. We propose WVDNet, a semi-supervised learning model for time-frequency analysis based on the Wigner-Ville distribution (WVD), to reduce the cross-term existing in WVD and relax the requirements of the training data set. The proposed WVDNet is based on the Mean-Teacher model to enable the task model to exploit the unlabeled training data. We first build a synthetic data set for model training, that contains different kinds of amplitude-modulated and frequency-modulated (AM-FM) signals. Next, a task model of WVDNet is designed and the consistency regularization based method is utilized to promote model training. Finally, experiments are conducted on both synthetic and real-world data, showing the effectiveness of suppressing cross-term and strong generalization ability.
What problem does this paper attempt to address?