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

Instance Segmentation by Jointly Optimizing Spatial Embeddings and Clustering Bandwidth

Lightweight mmm - Lightweight (Bayesian) Media Mix Model

pytorch implementation of "Contrastive Multiview Coding", "Momentum Contrast for Unsupervised Visual Representation Learning", and "Unsupervised Feature Learning via Non-Parametric Instance-level Discrimination"

Official PyTorch code for WACV 2022 paper "CFLOW-AD: Real-Time Unsupervised Anomaly Detection with Localization via Conditional Normalizing Flows"

exponential adaptive pooling for PyTorch

《K-Adapter: Infusing Knowledge into Pre-Trained Models with Adapters》(2020)

An efficient PyTorch implementation of the evaluation metrics in recommender systems.

This repository contains the implementation of the HealthGen model, a generative model to synthesize realistic EHR time series data with missingness

Instantaneous Motion Generation for Robots and Machines.

classify fashion-mnist dataset with pytorch

the official implementation of the paper "Isometric Multi-Shape Matching" (CVPR 2021)

Object Detection and Multi-Object Tracking

Official PyTorch implementation of PS-KD

Attack on Confidence Estimation algorithm from the paper "Disrupting Deep Uncertainty Estimation Without Harming Accuracy"

Source codes for Improved Few-Shot Visual Classification (CVPR 2020), Enhancing Few-Shot Image Classification with Unlabelled Examples

CurriculumNet: Weakly Supervised Learning from Large-Scale Web Images

Space Ship Simulator using python

Natural Posterior Network: Deep Bayesian Predictive Uncertainty for Exponential Family Distributions

In this tutorial, you will perform inference across 10 well-known pre-trained object detectors and fine-tune on a custom dataset. Design and train your own object detector.

Tensor-Based Quantum Machine Learning