• example/AlexNet.py
  • Example usuage of how to plug in our idea into existing AlexNet
  • One only need to concern with the codes controlled by the boolean varialable adv_flag, including:
    • Line 182
    • Line 291
    • Line 317
• utility/
  • Other codes needed to run the AlexNet, not part of the contributions of this repository

• This repository is made for a clear demonstration of our method. For replication of the numbers in our paper, please visit: HaohanWang/PAR_experiments

• For the ImageNet-Sketch Data, please visit: HaohanWang/ImageNet-Sketch

If you find this method helpful, please consider cite:

  Haohan Wang, Songwei Ge, Eric P. Xing, and Zachary C. Lipton. "Learning Robust Global Representations by Penalizing Local Predictive Power" (NeurIPS 2019)

• You may be also interested in HaohanWang/HEX