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

神经退化表示学习用于全能图像恢复

Neural Degradation Representation Learning for All-In-One Image Restoration

University of Science and Technology of China(中国科学技术大学)

Mingde Yao, Ruikang Xu, Yuanshen Guan, Jie Huang, Zhiwei Xiong

AI总结:

本文提出一种基于神经退化表示(NDR)的全能图像恢复网络,通过退化查询与注入模块及双向优化策略,实现对噪声、雾霾、雨和降采样等多种退化的有效恢复,实验验证了其有效性和泛化能力。

摘要:

现有方法在单一退化类型上已展现出有效性能。然而,在实际应用中,退化往往是未知的,模型与退化之间的不匹配会导致严重的性能下降。在本文中,我们提出了一种处理多种退化的全能图像恢复网络。由于不同类型退化的异质性,在单一网络中处理多种退化是困难的。为此,我们提出学习一种神经退化表示(NDR),以捕获各种退化的潜在特征。学习到的NDR自适应地分解不同类型的退化,类似于表示基本退化成分的神经字典。随后,我们开发了一个退化查询模块和一个退化注入模块,以基于NDR有效识别和利用特定退化,从而实现多种退化的全能恢复能力。此外,我们提出了一种双向优化策略,通过交替优化退化和恢复过程,有效驱动NDR学习退化表示。在代表性退化类型(包括噪声、雾霾、雨和降采样)上的综合实验证明了我们方法的有效性和泛化能力。

英文摘要:

Existing methods have demonstrated effective performance on a single degradation type. In practical applications, however, the degradation is often unknown, and the mismatch between the model and the degradation will result in a severe performance drop. In this paper, we propose an all-in-one image restoration network that tackles multiple degradations. Due to the heterogeneous nature of different types of degradations, it is difficult to process multiple degradations in a single network. To this end, we propose to learn a neural degradation representation (NDR) that captures the underlying characteristics of various degradations. The learned NDR decomposes different types of degradations adaptively, similar to a neural dictionary that represents basic degradation components. Subsequently, we develop a degradation query module and a degradation injection module to effectively recognize and utilize the specific degradation based on NDR, enabling the all-in-one restoration ability for multiple degradations. Moreover, we propose a bidirectional optimization strategy to effectively drive NDR to learn the degradation representation by optimizing the degradation and restoration processes alternately. Comprehensive experiments on representative types of degradations (including noise, haze, rain, and downsampling) demonstrate the effectiveness and generalization capability of our method.

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

↑