UnReflectAnything: RGB-Only Highlight Removal by Rendering Synthetic Specular Supervision
About
Specular highlights distort appearance, obscure texture, and hinder geometric reasoning in both natural and surgical imagery. We present UnReflectAnything, an RGB-only framework that removes highlights from a single image by predicting a highlight map together with a reflection-free diffuse reconstruction. The model uses a frozen vision transformer encoder to extract multi-scale features, a lightweight head to localize specular regions, and a token-level inpainting module that restores corrupted feature patches before producing the final diffuse image. To overcome the lack of paired supervision, we introduce a Virtual Highlight Synthesis pipeline that renders physically plausible specularities using monocular geometry, Fresnel-aware shading, and randomized lighting which enables training on arbitrary RGB images with correct geometric structure. UnReflectAnything generalizes across natural and surgical domains where non-Lambertian surfaces and non-uniform lighting create severe highlights and it achieves competitive performance with state-of-the-art results on several benchmarks. Project Page: https://alberto-rota.github.io/UnReflectAnything/
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Specular Highlight Removal | PSD | PSNR17.23 | 15 | |
| Specular Highlight Removal | SHIQ | PSNR17.61 | 15 | |
| Highlight Removal | CroMo | LSR0.012 | 8 | |
| Highlight Removal | SCRREAM | LSR0.002 | 8 | |
| Highlight Removal | SCARED | LSR0.011 | 8 | |
| Pixel-matching | StereoMIS (train) | Eep0.167 | 8 | |
| Pixel-matching | SCARED | Eep0.877 | 8 | |
| Pixel-matching | HouseCat6D | Eep75.9 | 8 | |
| Specular Highlight Removal | SSHR | MSEm0.002 | 8 | |
| Pixel-matching | SCRREAM | Eep42.3 | 8 |