Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Reversing Flow for Image Restoration

About

Image restoration aims to recover high-quality (HQ) images from degraded low-quality (LQ) ones by reversing the effects of degradation. Existing generative models for image restoration, including diffusion and score-based models, often treat the degradation process as a stochastic transformation, which introduces inefficiency and complexity. In this work, we propose ResFlow, a novel image restoration framework that models the degradation process as a deterministic path using continuous normalizing flows. ResFlow augments the degradation process with an auxiliary process that disambiguates the uncertainty in HQ prediction to enable reversible modeling of the degradation process. ResFlow adopts entropy-preserving flow paths and learns the augmented degradation flow by matching the velocity field. ResFlow significantly improves the performance and speed of image restoration, completing the task in fewer than four sampling steps. Extensive experiments demonstrate that ResFlow achieves state-of-the-art results across various image restoration benchmarks, offering a practical and efficient solution for real-world applications.

Haina Qin, Wenyang Luo, Libin Wang, Dandan Zheng, Jingdong Chen, Ming Yang, Bing Li, Weiming Hu• 2025

Related benchmarks

TaskDatasetResultRank
Shadow RemovalISTD (test)
RMSE (Shadow)11.1
76
Metal Artifact ReductionDeepLesion Metal Size Group 3 (test)
PSNR34.92
10
Metal Artifact ReductionDeepLesion Metal Size Group 4 (test)
PSNR37.01
10
Metal Artifact ReductionDeepLesion Small Metal (test)
PSNR37.83
10
Metal Artifact ReductionDeepLesion Average (test)
PSNR33.86
10
Metal Artifact ReductionDeepLesion Metal Size Group 2 (test)
PSNR31.09
10
Metal Artifact ReductionDeepLesion Large Metal (test)
PSNR28.47
10
MRI Bias Field CorrectionHCP clinical
SSIM95
8
Showing 8 of 8 rows

Other info

Follow for update