RAW-Flow: Advancing RGB-to-RAW Image Reconstruction with Deterministic Latent Flow Matching
Zhen Liu, Diedong Feng, Hai Jiang, Liaoyuan Zeng, Hao Wang, Chaoyu Feng, Lei Lei, Bing Zeng, Shuaicheng Liu
Abstract
RGB-to-RAW reconstruction, or the reverse modeling of a camera Image Signal Processing (ISP) pipeline, aims to recover high-fidelity RAW data from RGB images. Despite notable progress, existing learning-based methods typically treat this task as a direct regression objective and still struggle with detail inconsistency and color deviation, due to the ill-posed nature of inverse ISP and the inherent information loss in quantized RGB images. To address these limitations, we pioneer a generative perspective by reformulating RGB-to-RAW reconstruction as a deterministic latent transport problem and introduce a novel framework named RAW-Flow, which leverages flow matching to learn a deterministic vector field in latent space, to effectively bridge the gap between RGB and RAW representations and enable accurate reconstruction of structural details and color information. To further enhance latent transport, we introduce a cross-scale context guidance module that injects hierarchical RGB features into the flow estimation process. Moreover, we design a Dual-domain Latent Autoencoder (DLAE) with a feature alignment constraint to support the proposed latent transport framework, which jointly encodes RGB and RAW inputs while promoting stable training and high-fidelity reconstruction. Extensive experiments demonstrate that RAW-Flow outperforms state-of-the-art approaches both quantitatively and visually.
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 edb15160-cf2c-4dff-8f58-97c5c188def1Cited by top-tier papers5
- Efficient Hybrid SE(3)-Equivariant Visuomotor Flow Policy via Spherical Harmonics for Robot ManipulationQinglun Zhang, Shen Cheng, Tian Dan, Haoqiang Fan et al.CVPR 2026 · 1 citation
- ExpoCM: Exposure-Aware One-Step Generative Single-Image HDR ReconstructionAoyu Liu, Zhen Liu, Ziyi Wang, Dian Chen et al.CVPR 2026 · 1 citation
- DMAligner: Enhancing Image Alignment via Diffusion Model Based View SynthesisXinglong Luo, Ao Luo, Zhengning Wang, Yueqi Yang et al.CVPR 2026 · 1 citation
- Bridging RGB and RAW: Single-step Deterministic Flow with Homogeneous Representation AlignmentDiedong Feng, Peiyi Zeng, Zhen Liu, Zhongyang Li et al.ICML 2026
- ZeroIDIR: Zero-Reference Illumination Degradation Image Restoration with Perturbed Consistency Diffusion ModelsHai Jiang, Zhen Liu, Yinjie Lei, Songchen Han et al.CVPR 2026
Builds on17
- 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
- Scaling Rectified Flow Transformers for High-Resolution Image SynthesisPatrick Esser, Sumith Kulal, Andreas Blattmann, Rahim Entezari et al.ICML 2024 · 3,620 citations
- Denoising Diffusion Restoration ModelsBahjat Kawar, Michael Elad, Stefano Ermon, Jiaming SongNeurIPS 2022 · 1,439 citations
- Rethinking Noise Synthesis and Modeling in Raw DenoisingYi Zhang, Hongwei Qin, Xiaogang Wang, Hongsheng LiICCV 2021 · 100 citations
Related papers
- ISPDiffuser: Learning RAW-to-sRGB Mappings with Texture-Aware Diffusion Models and Histogram-Guided Color ConsistencyYang Ren, Hai Jiang, Menglong Yang, Wei Li et al.AAAI 2025 · 7 citations
- Learning to Zoom Inside Camera Imaging PipelineChengzhou Tang, Yuqiang Yang, Bing Zeng, Ping Tan et al.CVPR 2022 · 4 citations
- Invertible Image Signal ProcessingYazhou Xing, Zian Qian, Qifeng ChenCVPR 2021
- Generalizing ISP Model by Unsupervised Raw-to-raw MappingDongyu Xie, Chaofan Qiao, Lanyue Liang, Zhiwen Wang et al.ACM MM 2024 · 5 citations
- SpiralDiff: Spiral Diffusion with LoRA for RGB-to-RAW Conversion Across CamerasHuanjing Yue, Shangbin Xie, Cong Cao, Qian Wu et al.CVPR 2026
