ReasonX: MLLM-Guided Intrinsic Image Decomposition
About
Intrinsic image decomposition aims to separate images into physical components such as albedo, depth, normals, and illumination. While recent diffusion- and transformer-based models benefit from paired supervision from synthetic datasets, their generalization to diverse, real-world scenarios remains challenging. We propose ReasonX, a novel framework that leverages a multimodal large language model (MLLM) as a perceptual judge providing relative intrinsic comparisons, and uses these comparisons as GRPO rewards for fine-tuning intrinsic decomposition models on unlabeled, in-the-wild images. Unlike RL methods for generative models, our framework aligns conditional intrinsic predictors by rewarding agreement between the judge's relational assessments and analytically derived relations from the model's outputs. ReasonX is model-agnostic and can be applied to different intrinsic predictors. Across multiple base architectures and modalities, ReasonX yields significant improvements, including 9-25% WHDR reduction on IIW albedo and up to 46% depth accuracy gains on ETH3D, highlighting the promise of MLLM-guided comparative supervision to bridge low- and high-level vision reasoning.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Surface Normal Estimation | NYU V2 | -- | 23 | |
| Albedo Estimation | ARAP | LMSE0.02 | 19 | |
| Depth Estimation | ETH3D (test) | AbsRel0.077 | 17 | |
| Albedo Estimation | IIW v1.1 (test) | WHDR 10%12.9 | 11 | |
| Albedo Estimation | Interiorverse (test) | PSNR20.7 | 10 | |
| Intrinsic Image Decomposition (Albedo) | Hypersim (test) | PSNR19.9 | 10 | |
| Albedo Estimation | MAW | Intensity (×100)0.43 | 10 | |
| Relative Depth Estimation | NYU v2 (test) | AbsRel0.053 | 9 | |
| Intrinsic Image Decomposition (Irradiance) | Hypersim (test) | PSNR18.9 | 8 | |
| Surface Normal Estimation | DIODE | Mean Angle Error14.5 | 8 |