arXiv:2603.03839cs.CV2026-03中稿 · IEEE TIP 2026被引 4

提出新模型统一修复多种图像退化,提升真实场景适用性。

All-in-One Image Restoration via Causal-Deconfounding Wavelet-Disentangled Prompt Network

  • 用小波注意力解耦退化与语义特征,消除虚假关联
  • 在两个全场景设置下超越现有最佳方法,性能显著提升
  • 适合需要通用图像修复的工业应用或复杂退化场景

图像修复是应对图像内容失真的有效方法。传统方法依赖已知退化类型和程度,存储开销高,在动态实际场景中难以满足。相比之下,全功能图像修复(AiOIR)通过统一模型处理多种退化,克服上述问题。然而,我们通过因果分析发现,当前AiOIR模型仍存在两大缺陷:1)非退化语义特征与退化模式间的虚假相关;2)退化模式估计偏差。为揭示退化图像与恢复图像间的真正因果关系,本文提出因果去混淆小波解缠提示网络(CWP-Net)。该模型引入编码器和解码器的小波注意力模块,显式解耦退化与语义特征,解决虚假相关问题。同时,采用小波提示块生成替代变量,实现因果去混淆,缓解估计偏差。在两个全场景设置下的大量实验表明,CWP-Net在效果和泛化能力上均优于当前最优的AiOIR方法。

原文摘要 · Abstract (English)

Image restoration represents a promising approach for addressing the inherent defects of image content distortion. Standard image restoration approaches suffer from high storage cost and the requirement towards the known degradation pattern, including type and degree, which can barely be satisfied in dynamic practical scenarios. In contrast, all-in-one image restoration (AiOIR) eliminates multiple degradations within a unified model to circumvent the aforementioned issues. However, according to our causal analysis, we disclose that two significant defects still exacerbate the effectiveness and generalization of AiOIR models: 1) the spurious correlation between non-degradation semantic features and degradation patterns; 2) the biased estimation of degradation patterns. To obtain the true causation between degraded images and restored images, we propose Causal-deconfounding Wavelet-disentangled Prompt Network (CWP-Net) to perform effective AiOIR. CWP-Net introduces two modules for decoupling, i.e., wavelet attention module of encoder and wavelet attention module of decoder. These modules explicitly disentangle the degradation and semantic features to tackle the issue of spurious correlation. To address the issue stemming from the biased estimation of degradation patterns, CWP-Net leverages a wavelet prompt block to generate the alternative variable for causal deconfounding. Extensive experiments on two all-in-one settings prove the effectiveness and superior performance of our proposed CWP-Net over the state-of-the-art AiOIR methods.

图像修复因果学习小波网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。