Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

LoRA-IR: Taming Low-Rank Experts for Efficient All-in-One Image Restoration

About

Prompt-based all-in-one image restoration (IR) frameworks have achieved remarkable performance by incorporating degradation-specific information into prompt modules. Nevertheless, handling the complex and diverse degradations encountered in real-world scenarios remains a significant challenge. To tackle this, we propose LoRA-IR, a flexible framework that dynamically leverages compact low-rank experts to facilitate efficient all-in-one image restoration. Specifically, LoRA-IR consists of two training stages: degradation-guided pre-training and parameter-efficient fine-tuning. In the pre-training stage, we enhance the pre-trained CLIP model by introducing a simple mechanism that scales it to higher resolutions, allowing us to extract robust degradation representations that adaptively guide the IR network. In the fine-tuning stage, we refine the pre-trained IR network through low-rank adaptation (LoRA). Built upon a Mixture-of-Experts (MoE) architecture, LoRA-IR dynamically integrates multiple low-rank restoration experts through a degradation-guided router. This dynamic integration mechanism significantly enhances our model's adaptability to diverse and unknown degradations in complex real-world scenarios. Extensive experiments demonstrate that LoRA-IR achieves SOTA performance across 14 IR tasks and 29 benchmarks, while maintaining computational efficiency. Code and pre-trained models will be available at: https://github.com/shallowdream204/LoRA-IR.

Yuang Ai, Huaibo Huang, Ran He• 2024

Related benchmarks

TaskDatasetResultRank
Video RestorationTUD-Common (test)
Runtime (s)5.45
10
Video RestorationDAVIS Noise & Blur (test)
PSNR25.79
10
Video RestorationDAVIS Noise & Compression (test)
PSNR26.85
10
Video RestorationDAVIS Blur & Compression (test)
PSNR29.27
10
Video RestorationSet8 Noise & Blur (test)
PSNR23.92
10
Video RestorationSet8 Noise & Compression (test)
PSNR23.68
10
Video RestorationSet8 Blur & Compression (test)
PSNR27.08
10
Video RestorationDAVIS t=6 (test)
PSNR27.12
10
Video RestorationDAVIS t=12 (test)
PSNR26.75
10
Video RestorationDAVIS t=24 (test)
PSNR26.76
10
Showing 10 of 13 rows

Other info

Follow for update