Bridging RGB and RAW: Single-step Deterministic Flow with Homogeneous Aligned Guidance
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 Aligned Guidance 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.
BibTeX
@inproceedings{
feng2026bridging,
title={Bridging {RGB} and {RAW}: Single-step Deterministic Flow with Homogeneous Representation Alignment},
author={Diedong Feng and Peiyi Zeng and Zhen Liu and Zhongyang Li and Bing Zeng and Shuaicheng Liu},
booktitle={Forty-third International Conference on Machine Learning},
year={2026},
url={https://openreview.net/forum?id=FpL30b32RT}
}