Do Trendy ImageNet Classifiers Precisely Predict Perceptual Similarity?


The duty of figuring out the similarity between photographs is an open downside in laptop imaginative and prescient and is essential for evaluating the realism of machine-generated photographs. Although there are a variety of easy strategies of estimating picture similarity (e.g., low-level metrics that measure pixel variations, comparable to FSIM and SSIM), in lots of circumstances, the measured similarity variations don’t match the variations perceived by an individual. Nevertheless, more moderen work has demonstrated that intermediate representations of neural community classifiers, comparable to AlexNet, VGG and SqueezeNet skilled on ImageNet, exhibit perceptual similarity as an emergent property. That’s, Euclidean distances between encoded representations of photographs by ImageNet-trained fashions correlate a lot better with an individual’s judgment of variations between photographs than estimating perceptual similarity instantly from picture pixels.

Two units of pattern photographs from the BAPPS dataset. Educated networks agree extra with human judgements as in comparison with low-level metrics (PSNR, SSIM, FSIM). Picture supply: Zhang et al. (2018).

In “Do higher ImageNet classifiers assess perceptual similarity higher?” revealed in Transactions on Machine Studying Analysis, we contribute an intensive experimental examine on the connection between the accuracy of ImageNet classifiers and their emergent capacity to seize perceptual similarity. To guage this emergent capacity, we observe earlier work in measuring the perceptual scores (PS), which is roughly the correlation between human preferences to that of a mannequin for picture similarity on the BAPPS dataset. Whereas prior work studied the primary era of ImageNet classifiers, comparable to AlexNet, SqueezeNet and VGG, we considerably enhance the scope of the evaluation incorporating fashionable classifiers, comparable to ResNets and Imaginative and prescient Transformers (ViTs), throughout a variety of hyper-parameters.

Relationship Between Accuracy and Perceptual Similarity
It’s nicely established that options realized by way of coaching on ImageNet switch nicely to numerous downstream duties, making ImageNet pre-training a regular recipe. Additional, higher accuracy on ImageNet normally implies higher efficiency on a various set of downstream duties, comparable to robustness to widespread corruptions, out-of-distribution generalization and switch studying on smaller classification datasets. Opposite to prevailing proof that means fashions with excessive validation accuracies on ImageNet are more likely to switch higher to different duties, surprisingly, we discover that representations from underfit ImageNet fashions with modest validation accuracies obtain the most effective perceptual scores.

Plot of perceptual scores (PS) on the 64 × 64 BAPPS Dataset (y-axis) towards the ImageNet 64 × 64 validation accuracies (x-axis). Every blue dot represents an ImageNet classifier. Higher ImageNet classifiers obtain higher PS as much as a sure level (darkish blue), past which enhancing the accuracy lowers the PS. One of the best PS are attained by classifiers with average accuracy (20.0–40.0).

We examine the variation of perceptual scores as a perform of neural community hyperparameters: width, depth, variety of coaching steps, weight decay, label smoothing and dropout. For every hyperparameter, there exists an optimum accuracy as much as which enhancing accuracy improves PS. This optimum is pretty low and is attained fairly early within the hyperparameter sweep. Past this level, improved classifier accuracy corresponds to worse PS.

As illustration, we current the variation of PS with respect to 2 hyperparameters: coaching steps in ResNets and width in ViTs. The PS of ResNet-50 and ResNet-200 peak very early on the first few epochs of coaching. After the height, PS of higher classifiers lower extra drastically. ResNets are skilled with a studying charge schedule that causes a stepwise enhance in accuracy as a perform of coaching steps. Apparently, after the height, in addition they exhibit a step-wise lower in PS that matches this step-wise accuracy enhance.

Early-stopped ResNets attain the most effective PS throughout completely different depths of 6, 50 and 200.

ViTs encompass a stack of transformer blocks utilized to the enter picture. The width of a ViT mannequin is the variety of output neurons of a single transformer block. Growing its width is an efficient approach to enhance its accuracy. Right here, we differ the width of two ViT variants, B/8 and L/4 (i.e., Base and Giant ViT fashions with patch sizes 4 and eight respectively), and consider each the accuracy and PS. Just like our observations with early-stopped ResNets, narrower ViTs with decrease accuracies carry out higher than the default widths. Surprisingly, the optimum width of ViT-B/8 and ViT-L/4 are 6 and 12% of their default widths. For a extra complete record of experiments involving different hyperparameters comparable to width, depth, variety of coaching steps, weight decay, label smoothing and dropout throughout each ResNets and ViTs, take a look at our paper.

Slim ViTs attain the most effective PS.

Scaling Down Fashions Improves Perceptual Scores
Our outcomes prescribe a easy technique to enhance an structure’s PS: scale down the mannequin to scale back its accuracy till it attains the optimum perceptual rating. The desk under summarizes the enhancements in PS obtained by cutting down every mannequin throughout each hyperparameter. Apart from ViT-L/4, early stopping yields the very best enchancment in PS, no matter structure. As well as, early stopping is essentially the most environment friendly technique as there is no such thing as a want for an costly grid search.

Mannequin Default Width Depth Weight
ResNet-6 69.1 +0.4 +0.3 0.0 +0.5 69.6
ResNet-50 68.2 +0.4 +0.7 +0.7 +1.5 69.7
ResNet-200 67.6 +0.2 +1.3 +1.2 +1.9 69.5
ViT B/8 67.6 +1.1 +1.0 +1.3 +0.9 +1.1 68.9
ViT L/4 67.9 +0.4 +0.4 -0.1 -1.1 +0.5 68.4
Perceptual Rating improves by cutting down ImageNet fashions. Every worth denotes the development obtained by cutting down a mannequin throughout a given hyperparameter over the mannequin with default hyperparameters.

International Perceptual Features
In prior work, the perceptual similarity perform was computed utilizing Euclidean distances throughout the spatial dimensions of the picture. This assumes a direct correspondence between pixels, which can not maintain for warped, translated or rotated photographs. As a substitute, we undertake two perceptual features that depend on international representations of photographs, particularly the style-loss perform from the Neural Model Switch work that captures stylistic similarity between two photographs, and a normalized imply pool distance perform. The style-loss perform compares the inter-channel cross-correlation matrix between two photographs whereas the imply pool perform compares the spatially averaged international representations.

International perceptual features persistently enhance PS throughout each networks skilled with default hyperparameters (high) and ResNet-200 as a perform of practice epochs (backside).

We probe numerous hypotheses to elucidate the connection between accuracy and PS and are available away with a couple of further insights. For instance, the accuracy of fashions with out generally used skip-connections additionally inversely correlate with PS, and layers near the enter on common have decrease PS as in comparison with layers near the output. For additional exploration involving distortion sensitivity, ImageNet class granularity, and spatial frequency sensitivity, take a look at our paper.

On this paper, we discover the query of whether or not enhancing classification accuracy yields higher perceptual metrics. We examine the connection between accuracy and PS on ResNets and ViTs throughout many various hyperparameters and observe that PS displays an inverse-U relationship with accuracy, the place accuracy correlates with PS as much as a sure level, after which displays an inverse-correlation. Lastly, in our paper, we talk about intimately numerous explanations for the noticed relationship between accuracy and PS, involving skip connections, international similarity features, distortion sensitivity, layerwise perceptual scores, spatial frequency sensitivity and ImageNet class granularity. Whereas the precise rationalization for the noticed tradeoff between ImageNet accuracy and perceptual similarity is a thriller, we’re excited that our paper opens the door for additional analysis on this space.

That is joint work with Neil Houlsby and Nal Kalchbrenner. We might moreover prefer to thank Basil Mustafa, Kevin Swersky, Simon Kornblith, Johannes Balle, Mike Mozer, Mohammad Norouzi and Jascha Sohl-Dickstein for helpful discussions.


Leave a Reply

Your email address will not be published. Required fields are marked *