Enabling ISP-less Low-Power Computer Vision

TL;DR

This paper introduces methods to enable training and deploying computer vision models directly on raw sensor data without relying on traditional image signal processing, improving accuracy and efficiency on low-power devices.

Contribution

It proposes an invertible ISP pipeline to convert RGB datasets to raw images for training, and introduces an energy-efficient analog in-pixel demosaicing method for ISP-less vision systems.

Findings

Training on raw images increases accuracy by 7.1% on VWW dataset.

Analog in-pixel demosaicing improves mAP by 8.1% on PASCALRAW.

Few-shot learning further boosts mAP by 20.5% on PASCALRAW.

Abstract

In order to deploy current computer vision (CV) models on resource-constrained low-power devices, recent works have proposed in-sensor and in-pixel computing approaches that try to partly/fully bypass the image signal processor (ISP) and yield significant bandwidth reduction between the image sensor and the CV processing unit by downsampling the activation maps in the initial convolutional neural network (CNN) layers. However, direct inference on the raw images degrades the test accuracy due to the difference in covariance of the raw images captured by the image sensors compared to the ISP-processed images used for training. Moreover, it is difficult to train deep CV models on raw images, because most (if not all) large-scale open-source datasets consist of RGB images. To mitigate this concern, we propose to invert the ISP pipeline, which can convert the RGB images of any dataset to its raw counterparts, and enable model training on raw images. We release the raw version of the COCO dataset, a large-scale benchmark for generic high-level vision tasks. For ISP-less CV systems, training on these raw images result in a 7.1% increase in test accuracy on the visual wake works (VWW) dataset compared to relying on training with traditional ISP-processed RGB datasets. To further improve the accuracy of ISP-less CV models and to increase the energy and bandwidth benefits obtained by in-sensor/in-pixel computing, we propose an energy-efficient form of analog in-pixel demosaicing that may be coupled with in-pixel CNN computations. When evaluated on raw images captured by real sensors from the PASCALRAW dataset, our approach results in a 8.1% increase in mAP. Lastly, we demonstrate a further 20.5% increase in mAP by using a novel application of few-shot learning with thirty shots each for the novel PASCALRAW dataset, constituting 3 classes.