Harry24k / adversarial-attacks-pytorch

PyTorch implementation of adversarial attacks [torchattacks].
https://adversarial-attacks-pytorch.readthedocs.io/en/latest/index.html
MIT License
1.79k stars 337 forks source link

[BUG] Incorrect random start of PGD L2 #161

Open Framartin opened 11 months ago

Framartin commented 11 months ago

✨ Short description of the bug [tl;dr]

The current implementation of the random start of PGDL2 does not sample uniformly in the L2 ball (as done by the original paper).

Currently, PGDL2 first samples a random directions (normalized vector), and then samples a radius uniformly between 0 and 1. This sampling scheme does not sample uniformly in the L2 ball. Under the current sampling, the probability of sampling in the orange ring, illustrated by the figure below, is the same as the probability of the blue ring. Therefore, the probability is not proportional to the area.

diagram_rings

This bug can be an issue, since for high dimensional balls, the probability concentrates on the outer sphere, i.e., the expected value of the L2 norm of a random perturbation tends to epsilon, when the number of dimensions grows to infinity.

I should be able to work on resolving this issue in the following days.

💬 Detailed code and results

The code from torchattacks/attacks/pgdl2.py from line 58:

        if self.random_start:
            # Starting at a uniformly random point
            delta = torch.empty_like(adv_images).normal_()
            d_flat = delta.view(adv_images.size(0), -1)
            n = d_flat.norm(p=2, dim=1).view(adv_images.size(0), 1, 1, 1)
            r = torch.zeros_like(n).uniform_(0, 1)
            delta *= r/n*self.eps
            adv_images = torch.clamp(adv_images + delta, min=0, max=1).detach()
rikonaka commented 11 months ago

Hi @Framartin , long time no see , 🤪 hh, I have same question here https://github.com/Harry24k/adversarial-attacks-pytorch/issues/123, and the Harry's explanation:

First, the random start of PGDL2 is followed by some previous work (but I cannot remember the source code. Sorry). To my knowledge, there is no standard method to initialize the random noise at the beginning. I think there are several works to investigate the importance of random noise.

Oh, and could you also please fix the low PGDL2 attack success rate https://github.com/Harry24k/adversarial-attacks-pytorch/issues/142 ? I cannot find out what the problem is. 🥲

Framartin commented 11 months ago

Thanks for your prompt reply! I was busy preparing and defending my PhD thesis during the last few months 😄