Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

A Whac-A-Mole Dilemma: Shortcuts Come in Multiples Where Mitigating One Amplifies Others

About

Machine learning models have been found to learn shortcuts -- unintended decision rules that are unable to generalize -- undermining models' reliability. Previous works address this problem under the tenuous assumption that only a single shortcut exists in the training data. Real-world images are rife with multiple visual cues from background to texture. Key to advancing the reliability of vision systems is understanding whether existing methods can overcome multiple shortcuts or struggle in a Whac-A-Mole game, i.e., where mitigating one shortcut amplifies reliance on others. To address this shortcoming, we propose two benchmarks: 1) UrbanCars, a dataset with precisely controlled spurious cues, and 2) ImageNet-W, an evaluation set based on ImageNet for watermark, a shortcut we discovered affects nearly every modern vision model. Along with texture and background, ImageNet-W allows us to study multiple shortcuts emerging from training on natural images. We find computer vision models, including large foundation models -- regardless of training set, architecture, and supervision -- struggle when multiple shortcuts are present. Even methods explicitly designed to combat shortcuts struggle in a Whac-A-Mole dilemma. To tackle this challenge, we propose Last Layer Ensemble, a simple-yet-effective method to mitigate multiple shortcuts without Whac-A-Mole behavior. Our results surface multi-shortcut mitigation as an overlooked challenge critical to advancing the reliability of vision systems. The datasets and code are released: https://github.com/facebookresearch/Whac-A-Mole.

Zhiheng Li, Ivan Evtimov, Albert Gordo, Caner Hazirbas, Tal Hassner, Cristian Canton Ferrer, Chenliang Xu, Mark Ibrahim• 2022

Related benchmarks

TaskDatasetResultRank
Image ClassificationImageNet-1k (val)--
1453
Image ClassificationImageNet-1K
Top-1 Acc86.84
524
Image ClassificationImageNet-R
Top-1 Acc68.7
474
Image ClassificationImageNet-Sketch
Top-1 Accuracy53.39
360
Image ClassificationImageNet-1k (val)
Top-1 Acc85.84
287
Image ClassificationWaterbirds (test)
Worst-Group Accuracy92.1
92
ClassificationCelebA (test)
Average Accuracy91.1
92
Image ClassificationStylized-ImageNet
Top-1 Accuracy31.15
89
Image ClassificationImageNet-W
IN-W Gap-1.11
74
Image ClassificationImageNet-1K--
51
Showing 10 of 19 rows

Other info

Code

Follow for update