arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.05075cs.CVcs.AI

Uni-Removal:一种同时处理真实世界图像中多种退化的半监督框架

Uni-Removal: A Semi-Supervised Framework for Simultaneously Addressing Multiple Degradations in Real-World Images

Beijing University of Posts and Telecommunications(北京邮电大学)

Yongheng Zhang, Danfeng Yan, Yuanqiang Cai

AI总结:

Uni-Removal提出两阶段半监督框架,通过多教师知识迁移与对抗域适应,用统一模型同时处理真实世界图像去雾、去雨和去模糊。

摘要:

从真实世界图像中去除多种退化(如雾霾、雨水和模糊)是一个具有挑战性且不适定的问题。近年来,能够处理不同退化的统一模型被提出并取得了有前景的结果。然而,这些方法主要关注合成图像,在应用于真实世界图像时性能显著下降。本文提出了Uni-Removal,一个两阶段半监督框架,使用统一模型和参数来解决真实世界图像中多种退化的去除问题。在知识迁移阶段,Uni-Removal利用监督式多教师和学生架构,促进从专门处理不同退化类型的预训练教师网络中学习。引入多粒度对比损失以增强从特征空间和图像空间的学习。在领域适应阶段,通过在真实世界图像上引入对抗判别器进行无监督微调。扩展的多粒度对比损失与生成对抗损失的结合使学生网络能够从合成域适应到真实世界域。在真实世界退化数据集上的大量实验证明了所提方法的有效性。我们将Uni-Removal框架与最先进的监督和无监督方法进行比较,展示了其在真实世界图像去雾、去雨和去模糊方面同时取得的良好结果。

英文摘要:

Removing multiple degradations, such as haze, rain, and blur, from real-world images poses a challenging and illposed problem. Recently, unified models that can handle different degradations have been proposed and yield promising results. However, these approaches focus on synthetic images and experience a significant performance drop when applied to realworld images. In this paper, we introduce Uni-Removal, a twostage semi-supervised framework for addressing the removal of multiple degradations in real-world images using a unified model and parameters. In the knowledge transfer stage, Uni-Removal leverages a supervised multi-teacher and student architecture in the knowledge transfer stage to facilitate learning from pretrained teacher networks specialized in different degradation types. A multi-grained contrastive loss is introduced to enhance learning from feature and image spaces. In the domain adaptation stage, unsupervised fine-tuning is performed by incorporating an adversarial discriminator on real-world images. The integration of an extended multi-grained contrastive loss and generative adversarial loss enables the adaptation of the student network from synthetic to real-world domains. Extensive experiments on real-world degraded datasets demonstrate the effectiveness of our proposed method. We compare our Uni-Removal framework with state-of-the-art supervised and unsupervised methods, showcasing its promising results in real-world image dehazing, deraining, and deblurring simultaneously.

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

↑