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

MBPO (paper: When to trust your model: Model-based policy optimization) in offline RL settings

transfer attack; adversarial examples; black-box attack; unrestricted Adversarial Attacks on ImageNet; CVPR2021 天池黑盒竞赛

Jittor 64*64 implementation of StyleGAN

The PyTorch re-implement of a 3D CNN Tracker to extract coronary artery centerlines with state-of-the-art (SOTA) performance. (paper: 'Coronary artery centerline extraction in cardiac CT angiography using a CNN-based orientation classiﬁer')

Pytorch implementation of few-shot semantic image synthesis

A simple configurable bot for sending arXiv article alert by mail

On the adaptation of recurrent neural networks for system identification

Prediction of MBA refinance Index (Mortgage prepayment)

A denoising autoencoder + adversarial losses and attention mechanisms for face swapping.

Boundary-aware Transformers for Skin Lesion Segmentation

Deep Learning ❤️ OneFlow

Code for the KDD 2021 paper 'Filtration Curves for Graph Representation'

U-Net implementation in PyTorch for FLAIR abnormality segmentation in brain MRI

Human Action Controller - A human action controller running on different platforms.

Code release for the ICML 2021 paper "PixelTransformer: Sample Conditioned Signal Generation".

Self-supervised Deep LiDAR Odometry for Robotic Applications

Code for "Unsupervised Source Separation via Bayesian inference in the latent domain"

[ICCV'2021] Image Inpainting via Conditional Texture and Structure Dual Generation

MIMO-UNet - Official Pytorch Implementation

Official PyTorch implementation for FastDPM, a fast sampling algorithm for diffusion probabilistic models