Title | When and how convolutional neural networks generalize to out-of-distribution category–viewpoint combinations |
Publication Type | Journal Article |
Year of Publication | 2022 |
Authors | Madan, S, Henry, T, Dozier, J, Ho, H, Bhandari, N, Sasaki, T, Durand, F, Pfister, H, Boix, X |
Journal | Nature Machine Intelligence |
Volume | 4 |
Issue | 2 |
Pagination | 146 - 153 |
Date Published | 02/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. |
URL | https://www.nature.com/articles/s42256-021-00437-5 |
DOI | 10.1038/s42256-021-00437-5 |
Short Title | Nat Mach Intell |
Associated Module:
CBMM Relationship:
- CBMM Funded