From d57a3ae382a10382f8ad47a51f0b98f6fb9e6e93 Mon Sep 17 00:00:00 2001 From: spencerwooo Date: Mon, 26 Feb 2024 15:25:00 +0800 Subject: [PATCH] docs: register FIA --- README.md | 1 + src/torchattack/__init__.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/README.md b/README.md index 42409a7..7210462 100644 --- a/README.md +++ b/README.md @@ -58,6 +58,7 @@ Gradient-based attacks: | VMI-FGSM | $\ell_\infty$ | [Enhancing the Transferability of Adversarial Attacks through Variance Tuning](https://arxiv.org/abs/2103.15571) | `torchattack.VMIFGSM` | | VNI-FGSM | $\ell_\infty$ | [Enhancing the Transferability of Adversarial Attacks through Variance Tuning](https://arxiv.org/abs/2103.15571) | `torchattack.VNIFGSM` | | Admix | $\ell_\infty$ | [Admix: Enhancing the Transferability of Adversarial Attacks](https://arxiv.org/abs/2102.00436) | `torchattack.Admix` | +| FIA | $\ell_\infty$ | [Feature Importance-aware Transferable Adversarial Attacks](https://arxiv.org/abs/2107.14185) | `torchattack.FIA` | Others: diff --git a/src/torchattack/__init__.py b/src/torchattack/__init__.py index 7c71a31..92d3573 100644 --- a/src/torchattack/__init__.py +++ b/src/torchattack/__init__.py @@ -2,6 +2,7 @@ from torchattack.deepfool import DeepFool from torchattack.difgsm import DIFGSM from torchattack.fgsm import FGSM +from torchattack.fia import FIA from torchattack.geoda import GeoDA from torchattack.mifgsm import MIFGSM from torchattack.nifgsm import NIFGSM @@ -18,6 +19,7 @@ 'Admix', 'DeepFool', 'DIFGSM', + 'FIA', 'FGSM', 'GeoDA', 'MIFGSM',