Boosting All-in-One Image Restoration via Self-Improved Privilege Learning

TL;DR

This paper introduces Self-Improved Privilege Learning (SIPL), a novel approach that enhances all-in-one image restoration models by enabling iterative self-refinement using learned privileged information during inference.

Contribution

SIPL extends privileged information utility from training to inference, incorporating a learnable dictionary for self-correction, improving performance with minimal overhead.

Findings

Achieves +4.58 dB PSNR on composite tasks

Improves diverse five-task benchmarks by +1.28 dB PSNR

Seamlessly integrates with various architectures

Abstract

Unified image restoration models for diverse and mixed degradations often suffer from unstable optimization dynamics and inter-task conflicts. This paper introduces Self-Improved Privilege Learning (SIPL), a novel paradigm that overcomes these limitations by innovatively extending the utility of privileged information (PI) beyond training into the inference stage. Unlike conventional Privilege Learning, where ground-truth-derived guidance is typically discarded after training, SIPL empowers the model to leverage its own preliminary outputs as pseudo-privileged signals for iterative self-refinement at test time. Central to SIPL is Proxy Fusion, a lightweight module incorporating a learnable Privileged Dictionary. During training, this dictionary distills essential high-frequency and structural priors from privileged feature representations. Critically, at inference, the same learned dictionary then interacts with features derived from the model's initial restoration, facilitating a self-correction loop. SIPL can be seamlessly integrated into various backbone architectures, offering substantial performance improvements with minimal computational overhead. Extensive experiments demonstrate that SIPL significantly advances the state-of-the-art on diverse all-in-one image restoration benchmarks. For instance, when integrated with the PromptIR model, SIPL achieves remarkable PSNR improvements of +4.58 dB on composite degradation tasks and +1.28 dB on diverse five-task benchmarks, underscoring its effectiveness and broad applicability. Codes are available at our project page https://github.com/Aitical/SIPL.