Title: TRANSFERABLE RECOGNITION-AWARE IMAGE PROCESSING

Abstract: Recent progress in image recognition has stimulated the deployment of vision systems (e.g. image search engines) at an unprecedented scale. As a result, visual data are now often consumed not only by humans but also by machines. Meanwhile, existing image processing methods only optimize for better human perception, whereas the resulting images may not be accurately recognized by machines. This can be undesirable, e.g., the images can be improperly handled by search engines or recommendation systems. In this work, we propose simple approaches to improve machine interpretability of processed images: optimizing the recognition loss directly on the image processing neural network or through an intermediate transforming model, a process which we show can also be done in an unsupervised manner. Interestingly, the processing model's ability to enhance the recognition performance can transfer when evaluated on different recognition models, even if they are of different architectures, trained on different object categories or even different recognition tasks. This makes the solutions applicable even when we do not have the knowledge about future downstream recognition models, e.g., if we are to upload the processed images to the Internet. We conduct comprehensive experiments on three image processing tasks with two downstream recognition tasks, and confirm our method brings substantial accuracy improvement on both the same recognition model and when transferring to a different one, with minimal or no loss in the image processing quality.

1. INTRODUCTION:

Human: It's a bird ✔ Machine: It's a kite ✘ Image Processing Model ...

Input Image Output Image:

Figure 1 : Image processing has been used to generate images that look good for human, but not machines. In this work we study the problem of making processed images more recognizable by machines.

Unlike in image recognition where a neural network maps an image to a semantic label, a neural network used for image processing maps an input image to an output image with some desired properties. Examples include image super-resolution (Dong et al., 2014) , denoising (Xie et al., 2012) , deblurring (Eigen et al., 2013) , colorization (Zhang et al., 2016) and style transfer (Gatys et al., 2015) . The goal of such systems is to produce images of high perceptual quality to a human observer. For example, in image denoising, we aim to remove noise in the signal that is not useful to an observer and restore the image to its original "clean" form. Metrics like PSNR and SSIM (Wang et al., 2004) are often used (Dong et al., 2014; Tong et al., 2017) to approximate human-perceived similarity between the processed images with the original images, and direct human assessment on the fidelity of the output is often considered the "gold-standard" assessment (Ledig et al., 2017; Zhang et al., 2018b) . Therefore, many techniques (Johnson et al., 2016; Ledig et al., 2017; Isola et al., 2017) have been proposed for making the output images look perceptually pleasing to human.

However, image processing outputs may not be accurately recognized by image recognition systems. As shown in Fig. 1 , the output image of an denoising model could easily be recognized by a human as a bird, but a recognition model classifies it as a kite. One could specifically train a recognition model only on these output images produced by the denoising model to achieve better performance on such images, or could leverage some domain adaptation approaches to adapt the recognition model to this domain, but the performance on natural images can be harmed. This retraining/adaptation scheme might also be impractical considering the significant overhead induced by catering to various image processing tasks and models.

With the fast-growing size of image data, many images are often "viewed" and analyzed more by machines than by humans. Nowadays, any image uploaded to the Internet is likely to be analyzed by certain vision systems. For example, Facebook uses a system called Rosetta to extract texts from over 1 billion user-uploaded images every day (Maria, 2018) . It is of great importance that the processed images be recognizable by not only humans, but also by machines. In other words, recognition systems (e.g., image classifier or object detector), should be able to accurately explain the underlying semantic meaning of the image content. In this way, we make them potentially easier to search, recommended to more interested audience, and so on, as these procedures are mostly executed by machines based on their understanding of the images. Therefore, we argue that image processing systems should also aim at better machine recognizability. We call this problem "Recognition-Aware Image Processing".

It is also important that the enhanced recognizability is not specific to any concrete neural networkbased recognition model, i.e., the improvement on recognition performance is only achieved when the output images are evaluated on that particular model. Instead, the improvement should ideally be transferable when evaluated on different models, to support its usage without access to possible future recognition systems, since we may not decide what model will be used for recognizing the processed image, for example if we upload it to the Internet or share it on social media. We may not know what network architectures (e.g. ResNet or VGG) will be used for inference, what object categories the downstream model recognizes (e.g. animals or scenes), or even what task will be performed on the processed image (e.g. classification or detection). Without these specifications, it might be hard to enhance image's machine semantics.

In this work, we propose simple and highly effective approaches to make image processing outputs more accurately recognized by downstream recognition systems, transferable among different recognition architectures, categories and tasks. The approaches we investigate add a recognition loss optimized jointly with the image processing loss. The recognition loss is computed using a fixed recognition model that is pretrained on natural images, and can be done in an unsupervised manner, e.g., without semantic labels of the image. It can be optimized either directly by the original image processing network, or through an intermediate transforming network. We conduct extensive experiments, on multiple image enhancement/restoration (super-resolution, denoising, and JPEG-deblocking) and recognition (classification and detection) tasks, and demonstrate that our approaches can substantially boost the recognition accuracy on the downstream systems, with minimal or no loss in the image processing quality measured by conventional metrics. Also, the accuracy improvement transfers favorably among different recognition model architectures, object categories, and recognition tasks, which renders our simple solution effective even when we do not have access to the downstream recognition models. Our contributions can be summarized as follows:

• We propose to study the problem of enhancing the machine interpretability of image processing outputs, a desired property considering the amount of images analyzed by machines nowadays.

• We propose simple and effective methods towards this goal, suitable for different use cases, e.g., without ground truth semantic labels. Extensive experiments are conducted on multiple image processing and recognition tasks, demonstrating the wide applicability of the proposed methods.

• We show that using our simple approaches, the recognition accuracy improvement could transfer among recognition architectures, categories and tasks, a desirable behavior making the proposed methods applicable without access to the downstream recognition model.

2. RELATED WORK:

Image processing/enhancement problems such as super-resolution and denoising have a long history (Tsai, 1984; Park et al., 2003; Rudin et al., 1992; Candès et al., 2006) . Since the initial success of deep neural networks on these problems (Dong et al., 2014; Xie et al., 2012; Wang et al., 2016b) , a large body of works try to investigate better model architecture design and training techniques (Dong et al., 2016; Kim et al., 2016b; Shi et al., 2016; Kim et al., 2016a; Mao et al., 2016; Lai et al., 2017; Tai et al., 2017a; Tong et al., 2017; Tai et al., 2017b; Lim et al., 2017; Zhang et al., 2018d; Ahn et al., 2018; Lefkimmiatis, 2018; Chen et al., 2018; Haris et al., 2018b) , mostly on the image superresolution task. These works focus on generating high visual quality images under conventional metrics or human evaluation, without considering recognition performance on the output.

There are also a number of works that relate image recognition with processing. Some works (Zhang et al., 2016; Larsson et al., 2016; Zhang et al., 2018c; Sajjadi et al., 2017) 2019) also aims to enhance machine accuracy on poor-conditioned images but mostly focus on better image processing techniques without using recognition models. Wang et al. (2019) propose a method to make denoised images more accurately segmented, also presenting some interesting findings in transferability. Most existing works only consider one image processing task or image domain, and develop specific techniques, while our simpler approach is task-agnostic and potentially more widely applicable. Our work is also related but different from those which aims for robustness of the recognition model (Hendrycks & Dietterich, 2019; Li et al., 2019; Shankar et al., 2018) , since we focus on the training of the processing models and assume the recognition model is given.

3. METHOD:

In this section we first introduce the problem setting of "recognition-aware" image processing, and then we develop various approaches to address it, each suited for different use cases.

3.1. PROBLEM SETTING:

In a typical image processing problem, given a set of training input images {I k in } and corresponding target images

{I k target } (k = 1, • • • N ),

we aim to train a neural network that maps an input image to its corresponding target. For example, in image denoising, I k in is a noisy image and I k target is the corresponding clean image. Denoting this mapping network as P (for processing), parameterized by W P , during training our optimization objective is:

EQUATION

where P I k in is simply the output of the processing model I out , and l proc is the loss function for each sample. The pixel-wise mean-squared-error (MSE, or L 2 ) loss is one of the most popular choices. During evaluation, the performance is typically measured by average similarity (e.g., PSNR, SSIM) between I k target and I k out = P I k in , or through human assessment. In our problem setting of recognition-aware processing, we are interested in a recognition task, with a trained recognition model R (R for recognition), parameterized by W R . We assume each input/target image pair I k in /I k target is associated with a ground truth semantic label S k for the recognition task. Our goal is to train a image processing model P such that the recognition performance on the output images {I k out = P I k in } is high, when evaluated using R with the semantic labels {S k }. In practice, the recognition model R might not be available (e.g., on the cloud), in which case we could resort to other models if the performance improvement transfers among models. 

3.2. OPTIMIZING RECOGNITION LOSS:

Given our goal is to make the output images by P more recognizable by R, it is natural to add a recognition loss on top of the objective of the image processing task (Eqn. 1) during training:

min W P Lrecog = 1 N N k=1 lrecog R P I k in , S k (2)

l recog is the per-example recognition loss defined by the downstream recognition task. For example, for image classification, l recog could be the cross-entropy (CE) loss. Adding the image processing loss (Eqn. 1) and recognition loss (Eqn. 2) together, our total training objective becomes

min W P Lproc + λLrecog (3)

where λ is the coefficient controlling the weights of L recog relative to L proc . We denote this simple solution as "RA (Recognition-Aware) processing", which is visualized in Fig. 2 left. Note that once the training is finished, the recognition model used as loss is not needed anymore, and during inference, we only need the processing model P, thus no additional overhead is introduced when the model is actually put to deployment.

A potential shortcoming of directly optimizing L recog is that it might deviate P from optimizing the original loss L proc , and the trained P will generate images that are not as good as if we only optimize L proc . We will show that in experiments, however, with proper choice of λ, we could substantially boost the recognition performance with minimal or no sacrifice on image quality.

If using R as a fixed loss function can only boost the recognition accuracy on R itself, the use of the method could be restricted. Sometimes we do not have the knowledge about the downstream recognition model or even task, but we still would like to improve future recognition performance. Interestingly, we find that image processing models trained with the loss of one recognition model R 1 , can also boost the performance when evaluated using recognition model R 2 , even if model R 2 has a different architecture, recognizes a different set of categories or even is trained for a different task. This makes our method effective even when we cannot access the target downstream model, in which case we could use another trained model we do have access to as the loss function. This phenomenon also implies that the "recognizability" of a processed image can be a more general notion than just the extent it fits to a specific model. More details on how the improvement is transferable among different recognition models will be presented in the experiments.

3.3. UNSUPERVISED OPTIMIZATION OF RECOGNITION LOSS:

The solution mentioned above requires semantic labels available for training images, which however, may not be satisfied all the time. In this case, we could instead resort to regress the recognition model's output of the target image R(I k target ), given the target images {I k target } at hand, and that the recognition model R is pretrained and fixed. The recognition objective in Eqn. 2 changes to min

W P Lrecog = 1 N N k=1 l dis R P I k in , R I k target (4)

where l dis is a distance metric between R's output given input of processed image P I k in and ground truth target image I k target . For example, when R is a classification model and outputs a probability distribution over classes, l dis could be the KL divergence or simply a L 2 distance. During evaluation, the output of R is still compared to the ground truth semantic label S k . We call this approach "unsupervised RA". Note that it is only "unsupervised" for training model P , but not necessarily for the model R. The (pre)training of the model R is not our concern since in our problem setting (Sec. 3.1) R is a given trained model, and it can be trained in any manner, either with or without full supervision, and it can even be trained with another dataset as we later show in Sec. 4.4. This approach is to some extent related to the "knowledge distillation" paradigm (Hinton et al., 2014 ) used for network model compression, where the output of a large model is used to guide the output of a small model, given the same input images. Instead we use the same recognition model R but guide the upstream processing model to generate input to R which produces similar output with that of the target image.

3.4. USING AN INTERMEDIATE TRANSFORMER:

Sometimes we do want to guarantee that the added recognition loss L recog will not deviate the model P from optimizing its original loss. We can achieve this by introducing another intermediate transformation model T . After the input image going through the image processing model P , the output image is first fed to the model T , and T 's output image serves as the input for the recognition model R (Fig. 2 right ). In this case, T 's parameters W T are optimized for minimizing the recognition loss:

EQUATION

In this way, with the help of T on optimizing the recognition loss, the model P can now "focus on" its original image processing loss L proc . The optimization objective becomes:

EQUATION

In Eqn. 6, P is still solely optimizing L proc as in the original image processing problem (Eqn. 1). P is learned as if there is no recognition loss, and therefore the image processing quality of its output will not be affected. This could be achieved by "cutting" the gradient generated by L recog between the model T and P (Fig. 2 right). The responsibility for a better recognition performance falls on the model T . We term this solution as "RA with transformer".

The downside of using a transformer compared with directly optimizing recognition loss using the processing model, is that there are two instances for each image (the output of model P and T ), one is "for human" and the other is "for machines". Also, as we will show later, it can sometimes harm the transferability of the performance improvement, possibly because there is no image processing loss as a constraint on T 's output. Therefore, the transformer is best suited for the case where we want to guarantee the image processing quality not affected at all, at the expense of maintaining another image and losing some transferability.

4. EXPERIMENTS:

We evaluate our proposed methods on three image processing tasks, namely image super-resolution, denoising, and JPEG-deblocking. More specifically, these are image enhancement or restoration tasks, where usually the target image is an enhanced image or the original image. Other more broader image processing tasks such as pattern detection, segmentation, object extraction are not considered in this work. To obtain the input images, for super-resolution, we use a downsampling scale factor of 4×; for denoising, we add Gaussian noise on the images with a standard deviation of 0.1 to obtain the noisy images; for JPEG deblocking, a quality factor of 10 is used to compress the image to JPEG format. We pair these three tasks with two common visual recognition tasks, image classification and object detection. We adopt the SRResNet (Ledig et al., 2017) as the architecture of the image processing model P , due to its popularity and simplicity. For the transformer model T , we use the 6-block ResNet architecture in CycleGAN (Zhu et al., 2017) We first show our results when evaluating on the same recognition model, i.e., the R used for evaluation is the same as the R we use as the recognition loss in training. Table 1a shows our results on ImageNet classification. ImageNet-pretrained classification models ResNet-18/50/101, DenseNet-121 and VGG-16 are denoted as R18/50/101, D121, V16 in Table 1a . The "No Processing" row denotes the recognition performance on the input of the image processing model: for denoising/JPEGdeblocking, this corresponds to the noisy/JPEG-compressed images; for super-resolution, the lowresolution images are bicubic interpolated to the original resolution. "Plain Processing" denotes conventional image processing models trained without recognition loss as described in Eqn. 1. We observe that a plainly trained processing model can boost the accuracy over unprocessed images. These two are considered as baselines in our experiments.

From Table 1a , using RA processing can significantly boost the accuracy of output images over plainly processed ones, for all image processing tasks and recognition models. This is more prominent when the accuracy of plain processing is lower, e.g., in super-resolution and JPEG-deblocking, in which case we mostly obtain ∼10% accuracy improvement. Even without semantic labels, our unsupervised RA can still in most cases outperform baseline methods, despite achieves lower accuracy than its supervised counterpart. Also in super-resolution and JPEG-deblocking, using an intermediate transformer T can bring additional improvement over RA processing.

The results for object detection are shown in Table 1b . We observe similar trend as in classification: using recognition loss can consistently improve the mAP over plain image processing by a notable margin. On super-resolution, RA processing mostly performs on par with RA with transformer, but on the other two tasks using a transformer is slightly better. The model with transformer performs better more often possibly because with this extra network in the middle, the capacity of the whole system is increased: in RA Processing the processing model P optimizes both processing and recognition loss, but now P optimizes processing loss while T optimizes recognition loss

4.2. TRANSFER BETWEEN RECOGNITION ARCHITECTURES:

In reality, sometimes the recognition model R we want to eventually evaluate the output images on might not be available for us to use as a loss for training, e.g., it could be on the cloud, kept confidential or decided later. In this case, we could train an image processing model P using recognition model R A that is accessible to us, and after we obtain the trained model P , evaluate its output images' recognition accuracy using another unseen recognition model R B . We evaluate all model architecture pairs on ImageNet classification in Table 2 and Table 3 In Table 2 's each column, training with any model R A produces substantially higher accuracy than plainly processed images on R B . Thus, we conclude that the improvement on the recognition accuracy is transferable among different recognition architectures. A possible explanation for this is that these models are all trained on the same ImageNet dataset, such that their mapping functions from input to output are similar, and optimizing the loss of one would lead to the lower loss of another. This phenomenon enables us to use RA processing without the knowledge of the downstream recognition model architecture. However, among all rows, the R A that achieves the highest accuracy is still the same model as R B , indicated by the diagonal boldface numbers in Table 2 .

Task:

Super-resolution Denoising JPEG-deblocking Evaluation on R18 R50 R101 D121 V16 R18 R50 R101 D121 V16 R18 R50 R101 D121 V16 Plain Processing 52.6 58.8 61.9 57.7 50.2 61.9 68.0 69.1 66.4 60.9 48.2 53.8 56.0 52.9 42.4 RA w/T w/ R18 63.0 59.2 67.0 63.9 27.0 65.2 69.4 71.6 68.4 40.3 59.8 58.7 62.6 60.3 19.9 RA w/T w/ R50 60.5 68.2 68.9 65.8 40.4 63.1 70.9 71.5 68.6 48.7 55.0 65.1 63.9 61.9 31.5 RA w/T w/ R101 59.6 66.2 70.1 65.1 35.6 62.4 68.8 72.3 67.6 52.3 54.8 61.3 66.7 24.8 60.5 RA w/T w/ D121 58.5 64.2 66.9 66.5 27.3 58.0 66.8 67.3 69.6 46.7 46.6 57.2 59.0 63.9 9.0 RA w/T w/ V16 59.2 64.7 67.8 65.0 63.0 57.6 64.0 67.1 55.7 63.1 56.1 61.2 63.4 58.7 60.1

Table 3 : Transfer between architectures using RA with Transformer (T ), on ImageNet classification.

Meanwhile in Table 3 , in most cases improvement is still transferable when we use a transformer T , but there are a few exceptions. For example, when R A is ResNet or DenseNet and when R B is VGG-16, in most cases the accuracy fall behind plain processing by a large margin. This weaker transferability is possibly caused by the fact that there is no constraint imposed by the image processing loss on T 's output, thus it "overfits" more to the specific R it is trained with. For more results on object detection and unsupervised RA, please refer to the Appendix B.1. This is intuitive since the processing model optimizes the same recognition loss during training as that used in evaluation.

One of the reasons why our method attains transferability is possibly that these models learn many common features that could be useful for general computer vision, especially in shallower layers. More importantly, the reason could be similar to the reason why adversarial examples can transfer among models: different models' decision boundaries are similar. In Table 5 , note that rows indicate classification models used as loss and columns indicate detection models, so even if they are of the same name (e.g., "R18"), they are still different models, and are trained on different datasets for different tasks. We are transferring between architectures, categories as well as tasks in this experiment. There is even a domain shift since the model P is trained with ImageNet training set but fed with PASCAL VOC input images during evaluation. Here "Plain Processing" models are trained on the ImageNet instead of PASCAL VOC dataset, thus the results are different from those in Table 1b . We observe that except two cases on the "V16" column in denoising, using classification loss on model A (row) can boost the detection accuracy on model B notably upon plain processing. This improvement is even comparable with directly training using the detection loss, as in Table 1b . Such task transferability suggests the "machine semantics" of the image could even be a task-agnostic property, and makes our method even more broadly applicable.

4.5. IMAGE PROCESSING QUALITY COMPARISON:

We have analyzed the recognition accuracy of the output images, now we compare the output image quality using conventional metrics PSNR and SSIM. When using RA with transformer, the output quality of P is guaranteed unaffected, therefore here we evaluate RA processing. We use ResNet-18 on ImageNet as R, and report results with different λs (Eqn. 3) in Table 6 . λ = 0 corresponds to plain processing. When λ = 10 -4 , in super-resolution, the PSNR/SSIM metrics are even slightly higher, and in denoising and JPEG-deblocking they are only marginally worse. However, the accuracy obtained is significantly higher. This suggests that the added recognition loss is not harmful when λ is chosen properly. When λ is excessively large (10 -2 ), the image quality is hurt more, and interestingly even the recognition accuracy start to decrease. A proper balance between image processing loss and recognition loss is needed for both image quality and performance on downstream recognition tasks.

Target Image:

Input Image Plain Processing RA λ = 10 -4 RA λ = 10 -3 RA λ = 10 -2

Figure 3 : Examples where outputs of RA processing models can be correctly classified but those from plain processing models cannot. PSNR/SSIM/class prediction is shown below each output image. Slight differences between images from plain processing and RA processing models could be noticed when zoomed in.

In Fig. 3 , we visualize some examples where the output image is incorrectly classified with a plain image processing model, and correctly recognized with RA processing. With smaller λ (10 -2 and 10 -3 ), the image is nearly the same as the plainly processed images. When λ is too large (10 -2 ), we could see some extra textures when zooming in. For more results please refer to Appendix C.

5. ANALYSIS:

In this section we analyze some alternatives to our approaches. All experiments in this section are conducted using RA processing on super-resolution, with ResNet-18 trained on ImageNet as the recognition model, and λ= 10 -3 if used.

Training without the Image Processing Loss. It is possible to train the processing model on the recognition loss L recog , without even keeping the original image processing loss L proc (Eqn. 3). This may presumably lead to better recognition performance since the model P can now "focus on" optimizing the recognition loss. However, we found removing the original image processing loss hurts the recognition performance: the accuracy drops from 61.8% to 60.9%; even worse, the SSIM/PSNR metrics drop from 26.33/0.792 to 16.92/0.263, which is reasonable since the image processing loss is not optimized during training. This suggests the original image processing loss is helpful for the recognition accuracy, since it helps the corrupted image to restore to its original form.

Fine-tuning the Recognition Model. Instead of fixing the recognition model R, we could finetune it together with the training of image processing model P , to optimize the recognition loss.

Many prior works (Sharma et al., 2018; Bai et al., 2018; Zhang et al., 2018a) do train/fine-tune the recognition model jointly with the image processing model. We use SGD with momentum as R's optimizer, and the final accuracy reaches 63.0%. However, since we do not fix R, it becomes a model that specifically recognizes super-resolved images, and we found its performance on original target images drops from 69.8% to 60.5%. Moreover, when transferring the trained P on ResNet-56, the accuracy is 62.4 %, worse than 66.7% when we train with a fixed ResNet-18. We lose some transferability if we do not fix the recognition model R.

Training Recognition Models from Scratch. Other than fine-tuning a pretrained recognition model R, we could first train a super-resolution model, and then train R from scratch on the output images. We achieve 66.1% accuracy on the output images in the validation set, higher than 61.8% in RA processing. However, the accuracy on original clean images drops from 69.8% to 66.1%. Alternatively, we could even train R from scratch on the interpolated low-resolution images, in which case we achieve 66.0% on interpolated validation data but only 50.2% on the original validation data. In summary, training or fine-tuning R to cater the need of super-resolved or interpolated images can harm its performance on the original clean images, and causes additional overhead in storing models. In contrast, using our RA processing technique could boost the accuracy of output images with the performance on original images intact.

Figure 4 : Examples where output images from RA processing models can be correctly classified but those from plain processing models cannot. PSNR/SSIM/class prediction is shown below each output image. Slight differences between images from plain processing and RA processing models (especially with large λs) could be noticed when zoomed in.

:



Figures and Tables:

Figure 2: Left: RA (Recognition-Aware) processing. In addition to the image processing loss, we add a recognition loss using a fixed recognition model R, for the processing model P to optimize. Right: RA with transformer. "Recognition Loss" stands for the dashed box in the left figure. A Transformer T is introduced between the output of P and input of R, to optimize recognition loss. We cut the gradient from recognition loss flowing to P , such that P only optimizes the image processing loss and the image quality is guaranteed not affected.

Recognition-Aware (RA) processing techniques can substantially boost the recognition accuracy.

Table content: <table><tr><td colspan="3">4.1 EVALUATION ON THE SAME RECOGNITION MODEL</td><td/></tr><tr><td>Task</td><td>Super-resolution</td><td>Denoising</td><td>JPEG-deblocking</td></tr><tr><td colspan="4">Classification Model R18 R50 R101 D121 V16 R18 R50 R101 D121 V16 R18 R50 R101 D121 V16</td></tr><tr><td>No Processing</td><td colspan="3">46.3 50.4 55.5 51.6 42.1 46.8 55.8 61.3 59.7 46.7 43.1 47.7 55.2 49.2 43.9</td></tr><tr><td>Plain Processing</td><td colspan="3">52.6 58.8 61.9 57.7 50.2 61.9 68.0 69.1 66.4 60.9 48.2 53.8 56.0 52.9 42.4</td></tr><tr><td>RA Processing</td><td colspan="3">61.8 67.3 69.6 66.0 61.9 65.1 71.2 72.7 69.8 66.5 57.7 63.6 65.8 62.3 56.7</td></tr><tr><td>Unsupervised RA</td><td colspan="3">61.3 66.9 69.4 65.3 61.0 61.7 68.6 70.8 67.1 63.6 53.8 60.4 63.4 59.7 53.1</td></tr><tr><td colspan="4">RA w/ Transformer 63.0 68.2 70.1 66.5 63.0 65.2 70.9 72.3 69.6 65.9 59.8 65.1 66.7 63.9 58.7</td></tr><tr><td colspan="4">(a) Accuracy (%) on ImageNet classification. The five models achieve 69.8, 76.2, 77.4, 74.7, 73.4 on original</td></tr><tr><td>images.</td><td/><td/><td/></tr><tr><td>Task</td><td>Super-resolution</td><td>Denoising</td><td>JPEG-deblocking</td></tr><tr><td>Detection Model</td><td colspan="3">R18 R50 R101 V16 R18 R50 R101 V16 R18 R50 R101 V16</td></tr><tr><td>No Processing</td><td colspan="3">67.9 70.3 72.1 63.6 51.8 56.5 61.8 38.9 49.3 54.5 64.1 38.4</td></tr><tr><td>Plain Processing</td><td colspan="3">69.2 70.7 73.3 64.2 68.9 72.0 74.7 65.8 63.7 66.5 70.4 60.3</td></tr><tr><td>RA Processing</td><td colspan="3">71.2 74.4 75.6 68.1 70.9 73.7 75.6 67.6 67.4 70.4 72.9 63.9</td></tr><tr><td colspan="4">RA w/ Transformer 71.4 74.2 75.6 66.0 71.0 73.9 75.9 67.7 68.5 70.7 73.7 64.4</td></tr><tr><td colspan="4">(b) mAP on PASCAL VOC object detection. The four models achieve 74.2, 76.8, 77.9, 72.2 on original images.</td></tr></table>

, for RA Processing and RA with Transformer respectively, where row corresponds to the model used as recognition loss (R A ), and column corresponds to the evaluation model (R B ). For RA with Transformer, we use the processing model P and transformer T trained with R A together when evaluating on R B . Transfer between recognition architectures using RA processing, on ImageNet classification. A image processing model trained with model RA (row) as recognition loss can improve the recognition performance on model RB (column) over plain processing.

Table content: <table><tr><td>Task</td><td>Super-resolution</td><td>Denoising</td><td>JPEG-deblocking</td></tr><tr><td colspan="4">Evaluation on R18 R50 R101 D121 V16 R18 R50 R101 D121 V16 R18 R50 R101 D121 V16</td></tr><tr><td colspan="4">Plain Processing 52.6 58.8 61.9 57.7 50.2 61.9 68.0 69.1 66.4 60.9 48.2 53.8 56.0 52.9 42.4</td></tr><tr><td>RA w/ R18</td><td colspan="3">61.8 66.7 68.8 64.7 58.2 65.1 70.6 71.9 69.1 63.8 57.7 62.3 64.3 60.7 52.8</td></tr><tr><td>RA w/ R50</td><td colspan="3">59.3 67.3 68.8 64.3 59.1 64.2 71.2 72.2 69.2 64.7 55.8 63.6 64.7 61.0 53.5</td></tr><tr><td colspan="4">RA w/ R101 58.8 66.0 69.6 63.4 58.2 64.0 70.5 72.7 68.9 64.8 54.9 61.5 65.8 60.3 52.8</td></tr><tr><td colspan="4">RA w/ D121 59.0 65.6 67.8 66.0 57.4 64.2 70.6 72.0 69.8 64.3 54.8 61.8 64.4 62.3 52.9</td></tr><tr><td>RA w/ V16</td><td colspan="3">57.9 64.8 67.0 63.0 61.9 63.9 70.4 72.0 68.8 66.5 54.5 60.9 63.1 59.7 56.7</td></tr></table>

Liu et al. (2016) studies adversarial examples' transferability and shows decision boundaries of different models align well with each other;Tramèr et al. (2017) quantitatively analyzes similarity of different models' decision boundaries, and shows that the boundaries are close in arbitrary directions, whether adversarial or benign.4.3 TRANSFER BETWEEN OBJECT CATEGORIESWhat if the R A and R B recognize different categories of objects? Can RA processing still bring transferable improvement? To answer this question, we divide the 1000 classes from ImageNet into two splits (denoted as category A/B), each with 500 classes, and train two 500-way classification models (ResNet-18) on both splits, obtaining R A and R B . Next, we train two image processing models P A and P B with the R A and R B as recognition loss, using images from category A and B respectively. Note that neither the image processing model P nor the recognition model R has seen any images from the other split of categories during training, and R A and R B learn completely different mappings from input to output. The plain processing counterparts of P A and P B are also trained with category A and B respectively, but without the recognition loss. We evaluate obtained image processing models on both splits, and the results are shown in Table4. Transfer between different object categories (500-way accuracy %). RA processing on one set of categories can also improve the performance on another. "Cat" means category.We observe that using RA processing still benefits the recognition accuracy even when transferring across categories (e.g., in super-resolution, from 60.1% to 66.5% when transferring from category A to category B, on super resolution). The improvement is only marginally lower training with recognition model of the same category (e.g., from 60.2% to 67.8% when trained and evaluated both on category B). Such transferability between categories suggest the learned image processing models do not improve accuracy by adding category-specific signals to the output images, instead they generate more general signals that enable a wider set of classes to be better recognized.4.4 TRANSFER BETWEEN RECOGNITION TASKSWhat if we take a further step to the case when R A and R B not only recognize different categories, but also perform different tasks? We evaluate such task transferability for when task A is classification and task B is object detection in Table5. For results on the opposite direction and results for unsupervised RA, please refer to Appendix B.2.

Table content: <table><tr><td colspan="2">Task</td><td colspan="2">Super-resolution</td><td>Denoising</td><td>JPEG-deblocking</td></tr><tr><td colspan="6">Train/Eval Category Cat A Cat B Cat A CatB Cat A</td><td>Cat B</td></tr><tr><td colspan="2">Cat A Plain</td><td>59.6</td><td>60.1</td><td colspan="2">67.6 68.0 54.2</td><td>55.5</td></tr><tr><td>Cat A RA</td><td/><td>67.2</td><td>66.5</td><td colspan="2">69.7 69.4 63.0</td><td>62.3</td></tr><tr><td colspan="2">Cat B Plain</td><td>59.6</td><td>60.2</td><td colspan="2">67.0 67.5 54.7</td><td>56.0</td></tr><tr><td>Cat B RA</td><td/><td>66.4</td><td>67.8</td><td colspan="2">69.4 69.7 62.1</td><td>63.5</td></tr><tr><td>Task</td><td/><td>Super-resolution</td><td/><td>Denoising</td><td>JPEG-deblocking</td></tr><tr><td colspan="6">Evaluation on R18 R50 R101 V16 R18 R50 R101 V16 R18 R50 R101 V16</td></tr><tr><td colspan="6">Plain Processing 68.5 69.7 73.1 63.2 68.1 71.6 74.1 65.7 62.4 65.6 69.5 58.3</td></tr><tr><td>RA w/ R18</td><td colspan="5">71.3 73.5 75.6 67.8 70.6 73.1 75.5 64.1 67.7 70.3 73.2 62.4</td></tr><tr><td>RA w/ R50</td><td colspan="5">70.8 73.2 74.8 67.8 70.4 73.1 75.8 66.2 67.8 70.2 73.1 62.8</td></tr><tr><td colspan="6">RA w/ R101 70.7 73.2 75.3 67.0 70.5 73.5 75.7 66.9 68.1 70.2 72.8 63.2</td></tr><tr><td colspan="6">RA w/ D121 71.2 73.6 75.3 67.2 70.5 73.2 75.7 65.7 68.1 70.5 73.1 62.6</td></tr><tr><td>RA w/ V16</td><td colspan="5">70.4 72.4 74.6 67.5 70.6 73.0 75.7 67.7 67.8 70.3 73.2 63.7</td></tr></table>

Transfer from ImageNet classification to PASCAL VOC object detection (mAP). Processing model P trained with classification model A (row) can improve the performance on detection model B (column).

Table content: <table/>

.795/52.6 31.24/0.895/61.9 27.50/0.825/48.2 10 -4 26.33/0.803/59.2 31.18/0.894/64.4 27.50/0.823/56.0 10 -3 26.31/0.792/61.8 30.78/0.884/65.1 27.17/0.810/57.7 10 -2 25.47/0.760/61.3 29.71/0.855/64.3 26.32/0.776/56.6 PSNR/SSIM/Accuracy when using different λs, on ImageNet dataset.

Table content: <table><tr><td>λ</td><td>Super-resolution</td><td>Denoising</td><td>JPE-deblocking</td></tr><tr><td>0</td><td>26.29/0</td><td/><td/></tr></table>

Peer review (iclr2020):
Claims: The paper presents a concept of *recognition-aware (RA) image processing*: when one enhances image in a some way, not only human judjement should be taken into account, but also performance of various computer vision application using that image. As an example of processing tasks, authors take super-resolution, denoising and JPEG-artifacts removal. Downstream applications covered are image classification and object detection. Authors propose a several training schemas to solve this problem and discuss a limitations of each one: - *simple* preprocessing, when the only image enhancement loss is optimized - *RA* joint optimization of recognition and enhancement loss (supervised and unsupervised) - a variant when two images are created: one for human and one for machine. **** Recommendation: strong accept **** Comments: Experiments are vast and performed on a variety of CNN architectures: ResNets, DenseNet ant VGGNet. Because one cannot predict, which computer vision tasks will be needed in the future, the natural question arise: how the results got for one set of tasks, architectures and image enhancement types transfer to another. Paper carefully studies this aspect as well. Overall paper is well written and is pleasure to read. While reading, I made notes to ask in review - just to see the my questions answered in a next section. Authors also provide source code for training. I haven`t run them though, but glanced through them. Weaknesses: I cannot really find a significant one. As a minor points: - I would recommend to cite not the last papers for image enhancement porblems themselves like super-resolution and denoising: these are old problems with rich history, e.g. L. Rudin, S. Osher, and E. Fatemi, Nonlinear total variation based noise removal algorithms Physica D, 60 (1992), pp. 259–268. - *Transformer* is probably bad name for deep learning component, as it is already widely used for a specific seq2seq architecture **** After rebuttal: I am now even more convinced that paper should be accepted.