Paper Reading AI Learner

ODCR: Orthogonal Decoupling Contrastive Regularization for Unpaired Image Dehazing

2024-04-27 08:13:13
Zhongze Wang, Haitao Zhao, Jingchao Peng, Lujian Yao, Kaijie Zhao

Abstract

Unpaired image dehazing (UID) holds significant research importance due to the challenges in acquiring haze/clear image pairs with identical backgrounds. This paper proposes a novel method for UID named Orthogonal Decoupling Contrastive Regularization (ODCR). Our method is grounded in the assumption that an image consists of both haze-related features, which influence the degree of haze, and haze-unrelated features, such as texture and semantic information. ODCR aims to ensure that the haze-related features of the dehazing result closely resemble those of the clear image, while the haze-unrelated features align with the input hazy image. To accomplish the motivation, Orthogonal MLPs optimized geometrically on the Stiefel manifold are proposed, which can project image features into an orthogonal space, thereby reducing the relevance between different features. Furthermore, a task-driven Depth-wise Feature Classifier (DWFC) is proposed, which assigns weights to the orthogonal features based on the contribution of each channel's feature in predicting whether the feature source is hazy or clear in a self-supervised fashion. Finally, a Weighted PatchNCE (WPNCE) loss is introduced to achieve the pulling of haze-related features in the output image toward those of clear images, while bringing haze-unrelated features close to those of the hazy input. Extensive experiments demonstrate the superior performance of our ODCR method on UID.

Abstract (translated)

未配对图像去雾(UID)具有重要研究价值,因为获取具有相同背景的雾/清晰图像对具有挑战性。本文提出了一种名为Orthogonal Decoupling Contrastive Regularization(ODCR)的新方法来解决UID。我们的方法基于一个假设,即图像由雾相关特征和雾无关特征(如纹理和语义信息)组成。ODCR旨在确保去雾结果的雾相关特征与清晰图像的雾相关特征相似,而雾无关特征与输入雾图像对齐。为了实现这一目标,我们提出了Orthogonal MLPs优化几何地在Stiefel维度的方法,这些方法可以投影图像特征到正交空间,从而降低不同特征之间的相关性。此外,我们还提出了一个基于任务的有条件深度卷积特征分类器(DWFC),其中基于每个通道特征对预测功能源是否为雾或清晰进行自监督的贡献为权重分配给正交特征。最后,我们引入了加权PatchNCE(WPNCE)损失,以实现将输出图像中雾相关特征向清晰图像的雾相关特征的拉动,同时将雾无关特征带到雾输入的附近。大量实验证明,我们的ODCR方法在UID上具有卓越的性能。

URL

https://arxiv.org/abs/2404.17825

PDF

https://arxiv.org/pdf/2404.17825.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot