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

掩码预训练实现通用零样本去噪器

Masked Pre-training Enables Universal Zero-shot Denoiser

University of Science and Technology of China(中国科学技术大学);Shanghai AI Laboratory(上海人工智能实验室)

Xiaoxiao Ma, Zhixiang Wei, Yi Jin, Pengyang Ling, Tianle Liu, Ben Wang, Junkang Dai, Huaian Chen

AI总结:

本文提出 MPI 零样本去噪范式,先通过掩码预训练学习自然图像分布,再利用预训练权重对单张含噪图像进行迭代填充重建,在多种噪声场景下以更少推理时间取得优于已有方法的去噪效果。

摘要:

在这项工作中,我们观察到,通过掩码策略在海量通用图像上训练的模型,已经自然地嵌入了这些图像的分布知识,从而自发地获得了强大图像去噪的潜在能力。基于这一观察,我们提出了一种新颖的零样本去噪范式,即掩码预训练后迭代填充(Masked Pre-train then Iterative fill,MPI)。MPI 首先通过掩码方式训练模型,然后利用预训练权重在单张含噪图像上进行高质量的零样本图像去噪。具体而言,MPI 包含两个关键过程:1)掩码预训练(Masked Pre-training)涉及训练模型以随机掩码方式重建大量自然图像,从而获得可泛化的表示,为在具有不同噪声退化、甚至不同图像类型的图像上进行有效的零样本去噪积累潜力。2)迭代填充(Iterative filling)利用预训练知识实现有效的零样本去噪。它通过利用预训练权重迭代优化图像,专注于交替重建图像的不同部分,并在有限的迭代次数内逐步组装出完全去噪的图像。在各种噪声场景下的综合实验凸显了 MPI 相对于先前方法的显著进步,同时推理时间明显减少。代码可在 https://github.com/krennic999/MPI 获取。

英文摘要:

In this work, we observe that model trained on vast general images via masking strategy, has been naturally embedded with their distribution knowledge, thus spontaneously attains the underlying potential for strong image denoising. Based on this observation, we propose a novel zero-shot denoising paradigm, i.e., Masked Pre-train then Iterative fill (MPI). MPI first trains model via masking and then employs pre-trained weight for high-quality zero-shot image denoising on a single noisy image. Concretely, MPI comprises two key procedures: 1) Masked Pre-training involves training model to reconstruct massive natural images with random masking for generalizable representations, gathering the potential for valid zero-shot denoising on images with varying noise degradation and even in distinct image types. 2) Iterative filling exploits pre-trained knowledge for effective zero-shot denoising. It iteratively optimizes the image by leveraging pre-trained weights, focusing on alternate reconstruction of different image parts, and gradually assembles fully denoised image within limited number of iterations. Comprehensive experiments across various noisy scenarios underscore the notable advances of MPI over previous approaches with a marked reduction in inference time. Code available at https://github.com/krennic999/MPI.

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

↑