Skip to content

AlvinWen428/fighting-fire-with-fire

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

fighting-fire-with-fire

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.

concept

Image Classification on NICO

The code for image classification experiments is in the folder nico. Please refer to the doc for how to running the experiments.

Imitation Learning on CARLA

The code is coming soon!

Citation

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}
}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published