Bridging RGB and RAW: Single-step Deterministic Flow with Homogeneous Representation Alignment
Diedong Feng, Peiyi Zeng, Zhen Liu, Zhongyang Li, Bing Zeng, Shuaicheng Liu
Abstract
Reconstructing high-fidelity RAW sensor data from processed RGB images is a fundamental yet ill-posed problem, plagued by irreversible information loss and complex non-linear ISP transformations. While generative models offer high-quality reconstruction, they suffer from prohibitive computational costs. Conversely, dominant regression-based methods are fast but susceptible to incoherent observational deviations, often yielding over-smoothed predictions that drift from the authentic signal manifold. To reconcile this trade-off, we propose SHADE, a Single-step Homogeneous Aligned DEterministic flow framework. We validate that, unlike point-to-point regression, the single-step deterministic flow captures global transport trends and enables intrinsic robustness against input perturbations. Furthermore, we introduce Homogeneous Representation Alignment to maximize fidelity. By leveraging a homogeneously initialized student-teacher DINO pair, this mechanism enforces alignment within a shared feature space, significantly amplifying the representational capacity. Extensive experiments demonstrate that SHADE achieves state-of-the-art performance on multiple benchmarks, establishing a new paradigm for accurate and efficient sensor data reconstruction.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c343bf5e-833c-4316-bd57-cc545f3415aaBuilds on23
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Denoising Diffusion Implicit ModelsJiaming Song, Chenlin Meng, Stefano ErmonICLR 2021 · 11,743 citations
- Scalable Diffusion Models with TransformersWilliam Peebles, Saining XieICCV 2023 · 5,568 citations
- Scaling Rectified Flow Transformers for High-Resolution Image SynthesisPatrick Esser, Sumith Kulal, Andreas Blattmann, Rahim Entezari et al.ICML 2024 · 3,620 citations
- Palette: Image-to-Image Diffusion ModelsChitwan Saharia, William Chan, Huiwen Chang, Chris A. Lee et al.SIGGRAPH 2022 · 1,638 citations
Related papers
- RAW-Flow: Advancing RGB-to-RAW Image Reconstruction with Deterministic Latent Flow MatchingZhen Liu, Diedong Feng, Hai Jiang, Liaoyuan Zeng et al.AAAI 2026 · 3 citations
- HDRFlow: Real-Time HDR Video Reconstruction with Large MotionsGangwei Xu, Yujin Wang, Jinwei Gu, Tianfan Xue et al.CVPR 2024 · 12 citations
- FMPose3D: monocular 3D pose estimation via flow matchingTi Wang, Xiaohang Yu, Mackenzie Weygandt MathisCVPR 2026 · 6 citations
- ExpoCM: Exposure-Aware One-Step Generative Single-Image HDR ReconstructionAoyu Liu, Zhen Liu, Ziyi Wang, Dian Chen et al.CVPR 2026 · 1 citation
- Monocular Normal Estimation via Shading Sequence EstimationZongrui Li, Xinhua Ma, Minghui Hu, Yunqing Zhao et al.ICLR 2026 · 3 citations
