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

Riemannian Geometry for Molecular Surface Approximation (RGMolSA)

A deep learning based semantic search platform that computes similarity scores between provided query and documents

[ICCV 2021] Released code for Causal Attention for Unbiased Visual Recognition

Geometric Vector Perceptrons --- a rotation-equivariant GNN for learning from biomolecular structure

This repository is an official implementation of the paper MOTR: End-to-End Multiple-Object Tracking with TRansformer.

Using this you can control your PC/Laptop volume by Hand Gestures (pinch-in, pinch-out) created with Python.

A simple API wrapper for Discord interactions.

This is the unofficial code of Deep Dual-resolution Networks for Real-time and Accurate Semantic Segmentation of Road Scenes. which achieve state-of-the-art trade-off between accuracy and speed on cityscapes and camvid, without using inference acceleration and extra data

Towards Understanding Quality Challenges of the Federated Learning: A First Look from the Lens of Robustness

Awesome Human Pose Estimation

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

This package proposes simplified exporting pytorch models to ONNX and TensorRT, and also gives some base interface for model inference.

An efficient framework for reinforcement learning.

Pytorch implementation of "MOSNet: Deep Learning based Objective Assessment for Voice Conversion"

OptNet: Differentiable Optimization as a Layer in Neural Networks

A Light CNN for Deep Face Representation with Noisy Labels

Understanding and Overcoming the Challenges of Efficient Transformer Quantization

minimizer-space de Bruijn graphs (mdBG) for whole genome assembly

The official implementation of the Interspeech 2021 paper WSRGlow: A Glow-based Waveform Generative Model for Audio Super-Resolution.

Codebase for "Revisiting spatio-temporal layouts for compositional action recognition" (Oral at BMVC 2021).