All-in-One Image Restoration (AiOIR) has emerged as a promising yet challenging research direction. To address the core challenges of diverse degradation modeling and detail preservation, we propose UniLDiff, a unified framework enhanced with degradation- and detail-aware mechanisms, unlocking the power of diffusion priors for robust image restoration. Specifically, we introduce a Degradation-Aware Feature Fusion (DAFF) to dynamically inject low-quality features into each denoising step via decoupled fusion and adaptive modulation, enabling implicit modeling of diverse and compound degradations. Furthermore, we design a Detail-Aware Expert Module (DAEM) in the decoder to enhance texture and fine-structure recovery through expert routing. Extensive experiments across multi-task and mixed degradation settings demonstrate that our method consistently achieves state-of-the-art performance, highlighting the practical potential of diffusion priors for unified image restoration. Our code will be released.
@article{arxiv.2507.23685,
title = {UniLDiff: Unlocking the Power of Diffusion Priors for All-in-One Image Restoration},
author = {Zihan Cheng and Liangtai Zhou and Dian Chen and Ni Tang and Xiaotong Luo and Yanyun Qu},
journal= {arXiv preprint arXiv:2507.23685},
year = {2025}
}