MirrorVerse: Pushing Diffusion Models to Realistically Reflect the World
Ankit Dhiman, Manan Shah, R Venkatesh Babu
TL;DR
MirrorFusion 2.0 tackles the challenge of photorealistic mirror reflections by coupling a richly augmented synthetic dataset, SynMirrorV2, with a depth-conditioned, dual-branch diffusion inpainting model trained through a three-stage curriculum. The approach progressively exposes the model to single and multiple-object scenes before bridging to real-world data, achieving improved geometric fidelity, shading, and occlusion handling over prior diffusion methods. Quantitative metrics (PSNR, SSIM, LPIPS, CLIP similarity) and a user study show substantial gains on MirrorBenchV2, GSO, and MSD datasets, indicating better generalization to real-world imagery. While effective, the method still encounters artifacts in very cluttered scenes, motivating future work on further data augmentation and curriculum refinements to close remaining gaps in realism and consistency.
Abstract
Diffusion models have become central to various image editing tasks, yet they often fail to fully adhere to physical laws, particularly with effects like shadows, reflections, and occlusions. In this work, we address the challenge of generating photorealistic mirror reflections using diffusion-based generative models. Despite extensive training data, existing diffusion models frequently overlook the nuanced details crucial to authentic mirror reflections. Recent approaches have attempted to resolve this by creating synhetic datasets and framing reflection generation as an inpainting task; however, they struggle to generalize across different object orientations and positions relative to the mirror. Our method overcomes these limitations by introducing key augmentations into the synthetic data pipeline: (1) random object positioning, (2) randomized rotations, and (3) grounding of objects, significantly enhancing generalization across poses and placements. To further address spatial relationships and occlusions in scenes with multiple objects, we implement a strategy to pair objects during dataset generation, resulting in a dataset robust enough to handle these complex scenarios. Achieving generalization to real-world scenes remains a challenge, so we introduce a three-stage training curriculum to develop the MirrorFusion 2.0 model to improve real-world performance. We provide extensive qualitative and quantitative evaluations to support our approach. The project page is available at: https://mirror-verse.github.io/.
