CDDM: Channel Denoising Diffusion Models for Wireless Communications

Tong Wu,Zhiyong Chen,Dazhi He,Liang Qian,Yin Xu,Meixia Tao,Wenjun Zhang

2023-05-16

Abstract:Diffusion models (DM) can gradually learn to remove noise, which have been widely used in artificial intelligence generated content (AIGC) in recent years. The property of DM for removing noise leads us to wonder whether DM can be applied to wireless communications to help the receiver eliminate the channel noise. To address this, we propose channel denoising diffusion models (CDDM) for wireless communications in this paper. CDDM can be applied as a new physical layer module after the channel equalization to learn the distribution of the channel input signal, and then utilizes this learned knowledge to remove the channel noise. We design corresponding training and sampling algorithms for the forward diffusion process and the reverse sampling process of CDDM. Moreover, we apply CDDM to a semantic communications system based on joint source-channel coding (JSCC). Experimental results demonstrate that CDDM can further reduce the mean square error (MSE) after minimum mean square error (MMSE) equalizer, and the joint CDDM and JSCC system achieves better performance than the JSCC system and the traditional JPEG2000 with low-density parity-check (LDPC) code approach.

Information Theory,Signal Processing

What problem does this paper attempt to address?

The problem that this paper attempts to solve is: in wireless communication, how to use Diffusion Models (DM) to eliminate channel noise and improve the quality of the received signal. Specifically, the paper proposes Channel Denoising Diffusion Models (CDDM), which are designed to be physical - layer modules, learn the distribution of the input signal after channel equalization, and use this learned knowledge to remove channel noise. In this way, CDDM can further reduce the Mean Square Error (MSE) after the Minimum Mean Square Error (MMSE) equalizer and achieve better performance in the Joint Source - Channel Coding (JSCC) system, especially in Rayleigh fading channels and Additive White Gaussian Noise (AWGN) channels.

The main contributions of the paper are:
1. **Proposing the CDDM model**: Designing the training and sampling algorithms for the forward diffusion process and the reverse sampling process, enabling CDDM to effectively remove channel noise.
2. **Applying CDDM to the JSCC system**: Applying CDDM to the semantic communication system based on JSCC for wireless image transmission. The experimental results show that CDDM can significantly reduce MSE and increase Peak Signal - to - Noise Ratio (PSNR).
3. **Experimental verification**: Through experiments under different Signal - to - Noise Ratio (SNR) conditions, it is proved that CDDM can effectively improve the performance of the system in both AWGN and Rayleigh fading channels, especially more prominent under low SNR conditions.

In conclusion, by introducing the CDDM model, this paper provides a new solution for channel denoising in wireless communication and significantly improves the performance of the system.