用扩散模型降噪,信号检测性能超越传统方法。
Erasing Noise in Signal Detection with Diffusion Model: From Theory to Application
- 基于随机微分方程构建新理论,实现信号去噪
- 在BPSK/QAM下误差率显著降低,计算复杂度为O(n²)
- 无需微调即可适应多种信噪比,适合实际部署
本文提出一种基于去噪扩散模型(DM)的信号检测方法,其性能优于长期被视为最优的极大似然(ML)估计。理论上,建立了基于随机微分方程(SDEs)的智能信号检测新框架,证明了DM在抑制加性白高斯噪声中的有效性。同时,揭示了信噪比(SNR)与扩散模型步数之间的数学关系,表明对任意给定SNR,均可确定最优步数。为应对扩散模型在分布外输入下的潜在问题,引入数学缩放技术,使训练好的模型无需微调即可处理广泛SNR范围的信号检测。基于此理论,采用扩散变压器(DiT)作为主干网络,该方法计算复杂度为$/mathcal{O}(n^2)$。仿真结果表明,在BPSK和QAM调制下,该方法相比ML估计显著降低符号错误率(SER),且保持更低的计算开销。
原文摘要 · Abstract (English)
In this paper, a signal detection method based on the denoise diffusion model (DM) is proposed, which outperforms the maximum likelihood (ML) estimation method that has long been regarded as the optimal signal detection technique. Theoretically, a novel mathematical theory for intelligent signal detection based on stochastic differential equations (SDEs) is established in this paper, demonstrating the effectiveness of DM in reducing the additive white Gaussian noise in received signals. Moreover, a mathematical relationship between the signal-to-noise ratio (SNR) and the timestep in DM is established, revealing that for any given SNR, a corresponding optimal timestep can be identified. Furthermore, to address potential issues with out-of-distribution inputs in the DM, we employ a mathematical scaling technique that allows the trained DM to handle signal detection across a wide range of SNRs without any fine-tuning. Building on the above theoretical foundation, we propose a DM-based signal detection method, with the diffusion transformer (DiT) serving as the backbone neural network, whose computational complexity of this method is $\mathcal{O}(n^2)$. Simulation results demonstrate that, for BPSK and QAM modulation schemes, the DM-based method achieves a significantly lower symbol error rate (SER) compared to ML estimation, while maintaining a much lower computational complexity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。