arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2508.03373cs.CV

一次性扩散:面向高效全能图像恢复的退化感知LoRA

Diffusion Once and Done: Degradation-Aware LoRA for Efficient All-in-One Image Restoration

Ni Tang, Xiaotong Luo, Zihan Cheng, Liangtai Zhou, Dongxiao Zhang, Yanyun Qu

AI总结:

提出DOD方法,利用退化感知LoRA和一步采样稳定扩散模型,实现高效且高质量的全能图像恢复。

摘要:

扩散模型在全能图像恢复(AiOIR)中展现出强大的潜力,擅长生成丰富的纹理细节。现有的AiOIR方法要么重新训练扩散模型,要么使用额外的条件引导对预训练扩散模型进行微调。然而,这些方法常常面临推理成本高和对多种退化类型适应性有限的问题。在本文中,我们提出了一种高效的AiOIR方法,即一次性扩散(DOD),旨在仅通过稳定扩散(SD)模型的一步采样即可实现卓越的恢复性能。具体来说,首先引入多退化特征调制,利用预训练扩散模型捕获不同的退化提示。然后,参数高效的条件下低秩适应整合这些提示,以实现对SD模型的微调,从而适应不同的退化类型。此外,在SD的解码器中集成高保真细节增强模块,以改善结构和纹理细节。实验表明,我们的方法在视觉质量和推理效率方面均优于现有的基于扩散的恢复方法。

英文摘要:

Diffusion models have revealed powerful potential in all-in-one image restoration (AiOIR), which is talented in generating abundant texture details. The existing AiOIR methods either retrain a diffusion model or fine-tune the pretrained diffusion model with extra conditional guidance. However, they often suffer from high inference costs and limited adaptability to diverse degradation types. In this paper, we propose an efficient AiOIR method, Diffusion Once and Done (DOD), which aims to achieve superior restoration performance with only one-step sampling of Stable Diffusion (SD) models. Specifically, multi-degradation feature modulation is first introduced to capture different degradation prompts with a pretrained diffusion model. Then, parameter-efficient conditional low-rank adaptation integrates the prompts to enable the fine-tuning of the SD model for adapting to different degradation types. Besides, a high-fidelity detail enhancement module is integrated into the decoder of SD to improve structural and textural details. Experiments demonstrate that our method outperforms existing diffusion-based restoration approaches in both visual quality and inference efficiency.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑