CVMay 15, 2021

Are Convolutional Neural Networks or Transformers more like human vision?

arXiv:2105.07197v2232 citations
Originality Incremental advance
AI Analysis

This work addresses the problem of building more human-like vision models for AI researchers and cognitive scientists, though it is incremental as it builds on prior behavioral analyses.

The study compared Convolutional Neural Networks (CNNs) and Vision Transformers (ViTs) to assess which better mimics human vision, finding that ViTs produce errors more consistent with human errors using new granular metrics.

Modern machine learning models for computer vision exceed humans in accuracy on specific visual recognition tasks, notably on datasets like ImageNet. However, high accuracy can be achieved in many ways. The particular decision function found by a machine learning system is determined not only by the data to which the system is exposed, but also the inductive biases of the model, which are typically harder to characterize. In this work, we follow a recent trend of in-depth behavioral analyses of neural network models that go beyond accuracy as an evaluation metric by looking at patterns of errors. Our focus is on comparing a suite of standard Convolutional Neural Networks (CNNs) and a recently-proposed attention-based network, the Vision Transformer (ViT), which relaxes the translation-invariance constraint of CNNs and therefore represents a model with a weaker set of inductive biases. Attention-based networks have previously been shown to achieve higher accuracy than CNNs on vision tasks, and we demonstrate, using new metrics for examining error consistency with more granularity, that their errors are also more consistent with those of humans. These results have implications both for building more human-like vision models, as well as for understanding visual object recognition in humans.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes