Skip to yearly menu bar Skip to main content


Poster

InstaRevive: One-Step Image Enhancement via Dynamic Score Matching

Yixuan Zhu · Haolin Wang · Ao Li · Wenliang Zhao · Yansong Tang · Jingxuan Niu · Lei Chen · Jie Zhou · Jiwen Lu

Hall 3 + Hall 2B #556
[ ]
Thu 24 Apr midnight PDT — 2:30 a.m. PDT

Abstract:

Image enhancement finds wide-ranging applications in real-world scenarios due to complex environments and the inherent limitations of imaging devices. Recent diffusion-based methods yield promising outcomes but necessitate prolonged and computationally intensive iterative sampling. In response, we propose InstaRevive, a straightforward yet powerful image enhancement framework that employs score-based diffusion distillation to harness potent generative capability and minimize the sampling steps. To fully exploit the potential of the pre-trained diffusion model, we devise a practical and effective diffusion distillation pipeline using dynamic noise control to address inaccuracies in updating direction during score matching. Our noise control strategy enables a dynamic diffusing scope, facilitating precise learning of denoising trajectories within the diffusion model and ensuring accurate distribution matching gradients during training. Additionally, to enrich guidance for the generative power, we incorporate textual prompts via image captioning as auxiliary conditions, fostering further exploration of the diffusion model. Extensive experiments substantiate the efficacy of our framework across a diverse array of challenging tasks and datasets, unveiling the compelling efficacy and efficiency of InstaRevive in delivering high-quality and visually appealing results.

Live content is unavailable. Log in and register to view live content