用一步解码提升无线图像传输速度,显著降低延迟。
DriftDecode: One-Step Wireless Image Decoding via Drifting-Inspired Detail Recovery

- 基于信噪比条件的一步U-Net解码器,结合漂移启发的纹理损失。
- 30~ms解码延迟,相比10步流匹配解码快4.8倍,性能提升1.13dB PSNR。
- 适合低延迟无线图像传输场景,尤其在瑞利衰落信道下表现优异。
无线图像传输中的生成式接收机可提升重建质量,但基于扩散和流模型的解码依赖迭代推理,导致显著延迟。然而,接收信号已保留源图像的粗略结构,因此无线解码更应视为恢复任务而非从零生成图像,主要挑战在于恢复信道受损的细节。受此恢复导向视角启发,本文提出DriftDecode,一种信噪比(SNR)条件化的一步解码器。该方法将一步U-Net解码器与漂移启发的实例级纹理损失相结合,损失函数在感知特征空间中重构生成漂移模型中的漂移场机制,引导每个局部特征向其空间对齐的真实对应物靠拢,同时抑制错误纹理。在添加白高斯噪声(AWGN)和瑞利衰落信道下的DIV2K和MNIST数据集上实验表明,DriftDecode实现了优越的质量-延迟权衡:解码延迟仅为30~ms,相较10步流匹配解码器提速4.8倍,且持续优于仅使用MSE训练的模型,在瑞利衰落条件下于MNIST上最高获得1.13~dB PSNR增益。结果支持恢复导向的一步解码是低延迟无线图像传输中迭代生成解码的有效替代方案。
原文摘要 · Abstract (English)
Generative receivers for wireless image transmission can improve reconstruction quality, but diffusion-based and flow-based decoding relies on iterative inference and therefore incurs substantial latency. In wireless image transmission, however, the received signal already preserves the coarse structure of the source image. Wireless decoding is therefore better viewed as a recovery task than as image generation from scratch, and the main challenge lies in restoring channel-impaired details. Motivated by this recovery-oriented perspective, this paper proposes DriftDecode, a signal-to-noise ratio (SNR)-conditioned one-step decoder for wireless image reconstruction. DriftDecode couples a one-step U-Net decoder with a drift-inspired instance-level texture loss. The loss reformulates the drifting-field mechanism from generative drifting models in perceptual feature space, guiding each reconstructed local feature toward its spatially aligned ground-truth counterpart while suppressing mismatched textures. Experiments on DIV2K and MNIST under additive white Gaussian noise (AWGN) and Rayleigh fading channels show a favorable quality-latency tradeoff. DriftDecode achieves 30~ms decoding latency, providing a 4.8$\times$ speedup over a 10-step flow-matching decoder, while consistently outperforming MSE-only training and yielding up to 1.13~dB PSNR gain on MNIST under Rayleigh fading. These results support recovery-oriented one-step decoding as an effective alternative to iterative generative decoding for low-latency wireless image transmission.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。