When and how convolutional neural networks generalize to out-of-distribution category–viewpoint combinations

TitleWhen and how convolutional neural networks generalize to out-of-distribution category–viewpoint combinations
Publication TypeJournal Article
Year of Publication2022
AuthorsMadan, S, Henry, T, Dozier, J, Ho, H, Bhandari, N, Sasaki, T, Durand, F, Pfister, H, Boix, X
JournalNature Machine Intelligence
Volume4
Issue2
Pagination146 - 153
Date Published02/2022
Abstract

Object recognition and viewpoint estimation lie at the heart of visual understanding. Recent studies have suggested that convolutional neural networks (CNNs) fail to generalize to out-of-distribution (OOD) category–viewpoint combinations, that is, combinations not seen during training. Here we investigate when and how such OOD generalization may be possible by evaluating CNNs trained to classify both object category and three-dimensional viewpoint on OOD combinations, and identifying the neural mechanisms that facilitate such OOD generalization. We show that increasing the number of in-distribution combinations (data diversity) substantially improves generalization to OOD combinations, even with the same amount of training data. We compare learning category and viewpoint in separate and shared network architectures, and observe starkly different trends on in-distribution and OOD combinations, that is, while shared networks are helpful in distribution, separate networks significantly outperform shared ones at OOD combinations. Finally, we demonstrate that such OOD generalization is facilitated by the neural mechanism of specialization, that is, the emergence of two types of neuron—neurons selective to category and invariant to viewpoint, and vice versa.

URLhttps://www.nature.com/articles/s42256-021-00437-5
DOI10.1038/s42256-021-00437-5
Short TitleNat Mach Intell

Associated Module: 

CBMM Relationship: 

  • CBMM Funded