Files:

Data preparation:

- improved_diff - contains our modifications to improved diffusion code base to enable training and generating PAGs using Score-Based Gradients.

- data_notebooks - for generating heuristic PAG sources.

Training:

- AT_CIFAR and AT_STL for training the Adversarial Training baseline for CIFAR and STL.

- main_CIFAR and main_STL to apply our approach 
