arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2502.06432cs.CVcs.AI

Prompt-SID: 通过潜在扩散学习结构表示提示进行单图像去噪

Prompt-SID: Learning Structural Representation Prompt via Latent Diffusion for Single-Image Denoising

Huaqiu Li, Wang Zhang, Xiaowan Hu, Tao Jiang, Zikang Chen, Haoqian Wang

AI总结:

Prompt-SID通过潜在扩散学习结构表示提示,提升单图像去噪效果,有效保留结构细节。

摘要:

许多研究集中在利用配对数据集构建监督模型进行图像去噪,这证明是昂贵且耗时的。当前的自监督和无监督方法通常依赖于盲点网络或子图像对采样,导致像素信息丢失和细节结构信息的破坏,从而显著限制了这些方法的效果。在本文中,我们介绍了Prompt-SID,一种基于提示学习的单图像去噪框架,强调保留结构细节。该方法通过下采样图像对以自监督的方式进行训练。它通过结构编码捕获原始尺度的图像信息,并将此提示整合到去噪器中。为此,我们提出了一种基于潜在扩散过程的结构表示生成模型,并在基于变换器的去噪器架构中设计了一个结构注意力模块以解码提示。此外,我们引入了一种尺度重放训练机制,有效缓解了不同分辨率图像之间的尺度差距。我们在合成、真实世界和荧光成像数据集上进行了全面实验,展示了Prompt-SID的显著有效性。我们的代码将在https://github.com/huaqlili/Prompt-SID上发布。

英文摘要:

Many studies have concentrated on constructing supervised models utilizing paired datasets for image denoising, which proves to be expensive and time-consuming. Current self-supervised and unsupervised approaches typically rely on blind-spot networks or sub-image pairs sampling, resulting in pixel information loss and destruction of detailed structural information, thereby significantly constraining the efficacy of such methods. In this paper, we introduce Prompt-SID, a prompt-learning-based single image denoising framework that emphasizes preserving of structural details. This approach is trained in a self-supervised manner using downsampled image pairs. It captures original-scale image information through structural encoding and integrates this prompt into the denoiser. To achieve this, we propose a structural representation generation model based on the latent diffusion process and design a structural attention module within the transformer-based denoiser architecture to decode the prompt. Additionally, we introduce a scale replay training mechanism, which effectively mitigates the scale gap from images of different resolutions. We conduct comprehensive experiments on synthetic, real-world, and fluorescence imaging datasets, showcasing the remarkable effectiveness of Prompt-SID. Our code will be released at https://github.com/huaqlili/Prompt-SID.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑