ImageNet-trained CNNs are biased towards texture; increasing shape bias improves accuracy and robustness

Code

• github.com/rgeirhos/Stylized-ImageNet
  OfficialIn paperpytorch★ 0
• github.com/LiYingwei/ShapeTextureDebiasedTraining
  pytorch★ 111
• github.com/facebookresearch/augmentation-corruption
  pytorch★ 46
• github.com/rgeirhos/texture-vs-shape
  In paperpytorch★ 0
• github.com/mbuet2ner/local-global-features-cnn
  pytorch★ 0
• github.com/annstrange/breast-cancer-cnn
  tf★ 0
• github.com/frank-roesler/Image_Segmentation
  pytorch★ 0

Abstract

Convolutional Neural Networks (CNNs) are commonly thought to recognise objects by learning increasingly complex representations of object shapes. Some recent studies suggest a more important role of image textures. We here put these conflicting hypotheses to a quantitative test by evaluating CNNs and human observers on images with a texture-shape cue conflict. We show that ImageNet-trained CNNs are strongly biased towards recognising textures rather than shapes, which is in stark contrast to human behavioural evidence and reveals fundamentally different classification strategies. We then demonstrate that the same standard architecture (ResNet-50) that learns a texture-based representation on ImageNet is able to learn a shape-based representation instead when trained on "Stylized-ImageNet", a stylized version of ImageNet. This provides a much better fit for human behavioural performance in our well-controlled psychophysical lab setting (nine experiments totalling 48,560 psychophysical trials across 97 observers) and comes with a number of unexpected emergent benefits such as improved object detection performance and previously unseen robustness towards a wide range of image distortions, highlighting advantages of a shape-based representation.

Tasks

Benchmark Results

Reproductions