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

ReLo-IRR: Reflection-Guided LoRA Framework for Image Reflection Removal

About

Single-image reflection removal (SIRR) aims to recover the clean transmission layer from a reflection-contaminated image. Although recent methods achieve promising results with large diffusion models, they rely on image-agnostic adaptation strategies, e.g., fine-tuning or ControlNet, that enforce uniform suppression regardless of reflection severity. As a result, heavy reflections often leave residuals, while weak ones suffer from detail loss. To this end, we propose ReLo-IRR, a reflection-guided LoRA framework built upon the rectified flow model. First, a lightweight estimator is designed to predict the reflection strength descriptor, providing an explicit prior of reflection dominance for each image and enabling image-dependent LoRA modulation. Second, we introduce a time-conditioned mechanism that fuses this reflection descriptor with timestep embeddings, enabling LoRA modulation to evolve consistently with the coarse-to-fine denoising process. By jointly modeling reflection strength and denoising dynamics, our ReLo-IRR achieves robust suppression of diverse reflection conditions. Extensive experiments on challenging benchmarks validate the effectiveness of ReLo-IRR, demonstrating superior dereflection performance and robust generalization. The code is released at https://github.com/KONGBAI-8080/ReLo-IRR.

Chaoqun Wang, Yuehuan Wei, Haoxiang Cao, Shaobo Min• 2026

Related benchmarks

TaskDatasetResultRank
Single Image Reflection RemovalOpenRR-1K (test)
PSNR28.18
13
Image Reflection RemovalNature (test)
PSNR27.3
6
Image Reflection Removalreal (test)
PSNR25.58
6
Showing 3 of 3 rows

Other info

Follow for update