Towards Low-Cost Learning-based Camera ISP via Unrolled Optimization

James J. Clark,V. Nia,Marzieh S. Tahaei,A. Mosleh
DOI: https://doi.org/10.1109/CRV60082.2023.00010
2023-06-01
Abstract:Recently, learning-based image signal processor (ISP) pipelines modeled using convolutional neural networks (CNNs) have been able to provide higher quality images over traditional model-based ISPs at the expense of significant memory, energy, and computation overhead. We propose an unrolled optimization network that models the ISP pipeline with considerably lower number of parameters and computation overhead. The unrolled optimization solves the image reconstruction problem of the ISP by leveraging both model-based and learning-based methods. In the proposed ISP model, the image formation operators namely, blur kernels and sensor sampling functions are formulated with learnable parameters such that the physical constraints are respected during conventional training. A CNN that is shared across the iterations of the unrolled model plays the role of the prior and performs denoising. An efficient tone mapper network is also utilized to further improve the quality of the resulting images. The entire pipeline is then trained in an end-to-end fashion using perceptual loss. The proposed ISP has over $\mathbf{34}\times$ fewer parameters in comparison to the state-of-the art deep ISPs.
Computer Science,Engineering
What problem does this paper attempt to address?