DPMambaIR: All-in-One Image Restoration via Degradation-Aware Prompt State Space Model
This provides a more practical and versatile solution for image restoration tasks, addressing limitations in fine-grained degradation modeling and multi-task balancing, though it appears incremental in improving existing approaches.
The paper tackles the problem of all-in-one image restoration for multiple degradation types using a single model, achieving state-of-the-art results with 27.69dB PSNR and 0.893 SSIM on a dataset with seven degradation types.
All-in-One image restoration aims to address multiple image degradation problems using a single model, offering a more practical and versatile solution compared to designing dedicated models for each degradation type. Existing approaches typically rely on Degradation-specific models or coarse-grained degradation prompts to guide image restoration. However, they lack fine-grained modeling of degradation information and face limitations in balancing multi-task conflicts. To overcome these limitations, we propose DPMambaIR, a novel All-in-One image restoration framework that introduces a fine-grained degradation extractor and a Degradation-Aware Prompt State Space Model (DP-SSM). The DP-SSM leverages the fine-grained degradation features captured by the extractor as dynamic prompts, which are then incorporated into the state space modeling process. This enhances the model's adaptability to diverse degradation types, while a complementary High-Frequency Enhancement Block (HEB) recovers local high-frequency details. Extensive experiments on a mixed dataset containing seven degradation types show that DPMambaIR achieves the best performance, with 27.69dB and 0.893 in PSNR and SSIM, respectively. These results highlight the potential and superiority of DPMambaIR as a unified solution for All-in-One image restoration.