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

GeoRelight: Learning Joint Geometrical Relighting and Reconstruction with Flexible Multi-Modal Diffusion Transformers

About

Relighting a person from a single photo is an attractive but ill-posed task, as a 2D image ambiguously entangles 3D geometry, intrinsic appearance, and illumination. Current methods either use sequential pipelines that suffer from error accumulation, or they do not explicitly leverage 3D geometry during relighting, which limits physical consistency. Since relighting and estimation of 3D geometry are mutually beneficial tasks, we propose a unified Multi-Modal Diffusion Transformer (DiT) that jointly solves for both: GeoRelight. We make this possible through two key technical contributions: isotropic NDC-Orthographic Depth (iNOD), a distortion-free 3D representation compatible with latent diffusion models; and a strategic mixed-data training method that combines synthetic and auto-labeled real data. By solving geometry and relighting jointly, GeoRelight achieves better performance than both sequential models and previous systems that ignored geometry.

Yuxuan Xue, Ruofan Liang, Egor Zakharov, Timur Bagautdinov, Chen Cao, Giljoo Nam, Shunsuke Saito, Gerard Pons-Moll, Javier Romero• 2026

Related benchmarks

TaskDatasetResultRank
Albedo EstimationSynthetic
PSNR28.07
8
Normal estimationSynthetic
Angle Error8.64
5
RelightingSynthetic Data
PSNR27.22
5
RelightingLightStage Data
PSNR25.87
5
RelightingHumanOLAT
PSNR21.17
5
Geometric Shape EstimationSynthetic Data
Accuracy0.71
3
Showing 6 of 6 rows

Other info

Follow for update