Modular Neural Image Signal Processing
Mahmoud Afifi, Zhongling Wang, Ran Zhang, Michael S. Brown
TL;DR
This work introduces a fine-grained modular neural ISP that replaces monolithic end-to-end mappings with interpretable, independently trainable stages spanning raw enhancement, color correction, photofinishing (gain, GTM, LTM, chroma, gamma), guided upsampling, and detail enhancement. The framework enables camera-agnostic rendering, multiple picture styles, and an interactive photo-editing tool, including the ability to embed raw data into final JPEGs for unlimited post-editable re-rendering. Denoising uses pseudo ground-truth for training, while color correction and photofinishing rely on lightweight networks to predict stage-specific parameters, with a comprehensive loss balancing fidelity and perceptual quality. Experiments on the S24 dataset demonstrate state-of-the-art results with moderate parameter counts, and cross-camera generalization is supported by generic denoisers and cross-camera AWB models, complemented by a user study showing strong perceptual preferences. The approach also discusses practical considerations such as artifact mitigation, data misalignment challenges, and the ability to process sRGB inputs via linearization, underscoring the method’s practical impact for scalable, editable mobile imaging pipelines.
Abstract
This paper presents a modular neural image signal processing (ISP) framework that processes raw inputs and renders high-quality display-referred images. Unlike prior neural ISP designs, our method introduces a high degree of modularity, providing full control over multiple intermediate stages of the rendering process.~This modular design not only achieves high rendering accuracy but also improves scalability, debuggability, generalization to unseen cameras, and flexibility to match different user-preference styles. To demonstrate the advantages of this design, we built a user-interactive photo-editing tool that leverages our neural ISP to support diverse editing operations and picture styles. The tool is carefully engineered to take advantage of the high-quality rendering of our neural ISP and to enable unlimited post-editable re-rendering. Our method is a fully learning-based framework with variants of different capacities, all of moderate size (ranging from ~0.5 M to ~3.9 M parameters for the entire pipeline), and consistently delivers competitive qualitative and quantitative results across multiple test sets. Watch the supplemental video at: https://youtu.be/ByhQjQSjxVM
