This is the pytorch implementation of the paper - Axiomatic Attribution for Deep Networks.

Last update: Dec 23, 2022

Overview

Integrated Gradients

This is the pytorch implementation of "Axiomatic Attribution for Deep Networks". The original tensorflow version could be found here.

Acknowledgement

ankurtaly's tensorflow version

Requirements

python-3.5.2
pytorch-0.4.1
opencv-python

TODO List

add more functions as the original code.
finetune the results, make them close to the original paper.

Instructions

Highly recommend to use GPU to accelerate the computation. If you use CPU, I will recommend to select some small networks, such as resnet18. You also need to put your images under examples/.

Lists of networks that support (of course, you can add any networks by yourself)

inception
resnet18
resnet152
vgg19

Run the code

python main.py --cuda --model-type='inception' --img='01.jpg'

Results

Results are slightly different from the original paper, it may have some bugs or need to do some adjustments. I will keep updating it, any contributions are welcome!

This is the pytorch implementation of the paper - Axiomatic Attribution for Deep Networks.

Related tags

Overview

Integrated Gradients

Acknowledgement

Requirements

TODO List

Instructions

Lists of networks that support (of course, you can add any networks by yourself)

Run the code

Results

Inception-v3

ResNet-18

ResNet-152

VGG-19

Owner

Tianhong Dai

The official implementation of A Unified Game-Theoretic Interpretation of Adversarial Robustness.

Adaptive FNO transformer - official Pytorch implementation

[MedIA2021]MIDeepSeg: Minimally Interactive Segmentation of Unseen Objects from Medical Images Using Deep Learning

The codes for the work "Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentation"

Deep learning for spiking neural networks

[NeurIPS 2021]: Are Transformers More Robust Than CNNs? (Pytorch implementation & checkpoints)

Project page for End-to-end Recovery of Human Shape and Pose

FedScale: Benchmarking Model and System Performance of Federated Learning

Face Depixelizer based on "PULSE: Self-Supervised Photo Upsampling via Latent Space Exploration of Generative Models" repository.

Some tentative models that incorporate label propagation to graph neural networks for graph representation learning in nodes, links or graphs.

Taichi Course Homework Template

Gradient-free global optimization algorithm for multidimensional functions based on the low rank tensor train format

WPPNets: Unsupervised CNN Training with Wasserstein Patch Priors for Image Superresolution

K-Nearest Neighbor in Pytorch

Advances in Neural Information Processing Systems (NeurIPS), 2020.

Physical Anomalous Trajectory or Motion (PHANTOM) Dataset

LogAvgExp - Pytorch Implementation of LogAvgExp

Generative Flow Networks for Discrete Probabilistic Modeling

Official code for "End-to-End Optimization of Scene Layout" -- including VAE, Diff Render, SPADE for colorization (CVPR 2020 Oral)

Container : Context Aggregation Network