IntelLabs / MART

Modular Adversarial Robustness Toolkit
BSD 3-Clause "New" or "Revised" License
16 stars 0 forks source link

Universal Perturbations #132

Open dxoigmn opened 1 year ago

dxoigmn commented 1 year ago

What does this PR do?

A big missing feature of MART is the ability to generate universal perturbations. This PR tries to change that.

The way universal perturbations should work is that we "train" a perturbation on a dataset and frozen model. Right now we can only generate perturbations in "test" mode, however. This is because it is unclear how to "freeze" a model in MART.

python -m mart experiment=COCO_TorchvisionFasterRCNN_ShapeShifter trainer=gpu

My sense is that Perturber should really just be a LitModular. That will greatly unify everything and make it easier to implement universal perturbations.

This PR depends upon the following PRs:

Type of change

Please check all relevant options.

Testing

Please describe the tests that you ran to verify your changes. Consider listing any relevant details of your test configuration.

Before submitting

Did you have fun?

Make sure you had fun coding 🙃