Discriminative Attribution Dataset A

This is a synthetic dataset built to benchmark the capabilities of explainable AI methods against difficult tasks.
The dataset has two classes, which depend on whether there is an odd or an even number of triangles in the image.

Also attached are the PyTorch model weights of a classifier trained on the task, both the raw weights and a JIT-compiled version. The raw weights can be loaded with the Funke Lab torch package.