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

Geodesic Flow Matching on a Riemannian Degradation Manifold for Blind Image Restoration

About

Blind image restoration requires recovering clean images from observations corrupted by unknown and potentially mixed degradations. While recent deterministic flow-based methods model restoration as transport processes that map degraded images to clean ones, they typically rely on Euclidean interpolation, implicitly assuming linear degradation geometry. In this paper, we explicitly model degradations as points on a low-dimensional Riemannian manifold and formulate restoration as geodesic transport on the joint image-manifold space. Using a geodesic flow matching objective, we learn intrinsic transport dynamics that respect the curvature of degradation space. This framework generalizes linear flow matching, provides a principled treatment of mixed degradations as geodesic compositions, and yields a clean theoretical interpretation for generalization beyond observed degradations.

Akshay Janardan Bankar, Ankita Chatterjee, Sayan Banerjee, Shreyas Pandith, Kalakonda Sai Shashank, Amit Satish Unde• 2026

Related benchmarks

TaskDatasetResultRank
Image DeblurringRealBlur-J (test)--
259
DeblurringRealBlur-R
PSNR36.27
117
DeblurringRealBlur-J
PSNR29.97
114
Image DehazingDense-Haze (test)
SSIM51.4
56
Single Image Defocus DeblurringDPD (Combined)
PSNR26.56
37
Image DehazingNH-HAZE (test)
PSNR21.2
27
DerainingOutdoor-Rain (test)
PSNR32.09
8
DesnowingSnow100K (test)
PSNR31.01
8
DerainingRealRain (test)
FID48.09
7
DesnowingDense-Snow (test)
FID33.02
7
Showing 10 of 16 rows

Other info

Follow for update