Variational disentanglement for task-agnostic image restoration
摘要
Image restoration tasks, such as shadow removal, deraining, and denoising, traditionally necessitate specialized architectures. This paper introduces a task-agnostic image restoration framework based on variational feature disentanglement, utilizing dual variational autoencoder (VAE) encoders to separate degradation-specific features from content-preserving representations. Our approach, integrating a dual-scale attention mechanism, degrade-aware decoder, and a degrade area localizer, demonstrates competitive performance across shadow removal (AISTD, SRD), deraining (Rain100L), and denoising (BSD68) tasks, achieving computational efficiency (58.39 GFLOPs, 79.64 FPS). Here, we show that our architecture, trained independently per task, effectively disentangles features without task-specific architectural components at inference, offering a scalable solution for diverse restoration scenarios. The source code and pre-trained models are available at https://github.com/Midrees330/VDTA_IRNet.