Image restoration aims to recover high quality images from inputs degraded by various factors, such as adverse weather, blur, or low light. While recent studies have shown remarkable progress across individual or unified restoration tasks, they still suffer from limited generalization and inefficiency when handling unknown or composite degradations. To address these limitations, we propose RAR, a Restore, Assess and Repeat process, that integrates Image Quality Assessment (IQA) and Image Restoration (IR) into a unified framework to iteratively and efficiently achieve high quality image restoration. Specifically, we introduce a restoration process that operates entirely in the latent domain to jointly perform degradation identification, image restoration, and quality verification. The resulting model is fully trainable end to end and allows for an all-in-one assess and restore approach that dynamically adapts the restoration process. Also, the tight integration of IQA and IR into a unified model minimizes the latency and information loss that typically arises from keeping the two modules disjoint, (e.g. during image and/or text decoding). Extensive experiments show that our approach consistent improvements under single, unknown and composite degradations, thereby establishing a new state-of-the-art.
RAR consists of two tightly integrated modules: Latent Quality Assessment (LQA) and Unified Image Restoration (UIR). By operating entirely in the shared latent space, RAR enables iterative degradation identification, restoration, and verification in a unified end-to-end framework.
Latent Quality Assessment (LQA) projects degraded image latents into the IQA space via an image adapter, and aligns IQA outputs directly to restoration conditioning embeddings through a text adapter. This removes lossy text decoding and enables end-to-end differentiable feedback between assessment and restoration.
Unified Image Restoration (UIR) is a flow-matching–based generative restoration backbone that directly maps degraded latents to high-quality latents. Its noise-free formulation allows intermediate latent reassessment, enabling stable multi-round restoration.
Together, LQA and UIR form a closed-loop latent feedback system that enables adaptive Restore–Assess–Repeat iterations with a quality-aware stopping criterion.
RAR performs restoration through iterative assess–restore steps. For each example, we visualize the progressive improvement across multiple rounds, demonstrating how degradations are identified and removed sequentially.