This is the official code of the paper Fighting Fire with Fire: Avoiding DNN Shortcuts through Priming, published at ICML 2022. We propose a method PrimeNet to prime the Deep Neural Networks away from the shortcut solutions during optimization.
In our paper, there are two experiments on image classification and imitation learning respectively, illustrating the effectiveness of PrimeNet to improve the generalization performance. These two experiments are introduced as following.
The code for image classification experiments is in the folder nico. Please refer to the doc for how to running the experiments.
The code is coming soon!
If you find this work is useful for your research, please cite our paper:
@inproceedings{wen2022fighting,
title={Fighting Fire with Fire: Avoiding DNN Shortcuts through Priming},
author={Wen, Chuan and Qian, Jianing and Lin, Jierui and Teng, Jiaye and Jayaraman, Dinesh and Gao, Yang},
booktitle={International Conference on Machine Learning},
pages={23723--23750},
year={2022},
organization={PMLR}
}