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

WSDM2022 Challenge - Large scale temporal graph link prediction

Serving PyTorch 1.0 Models as a Web Server in C++

Official code for the ICLR 2021 paper Neural ODE Processes

Deep Residual Learning for Image Recognition

[ICCV21] Self-Calibrating Neural Radiance Fields

I created My own Virtual Artificial Intelligence named genesis, He can assist with my Tasks and also perform some analysis,,

Code for "Optimizing risk-based breast cancer screening policies with reinforcement learning"

Multi-Objective Reinforced Active Learning

Pytorch implementation of AngularGrad: A New Optimization Technique for Angular Convergence of Convolutional Neural Networks

Parallel and High-Fidelity Text-to-Lip Generation; AAAI 2022 ; Official code

Mesh TensorFlow: Model Parallelism Made Easier

This is a work in progress reimplementation of Instant Neural Graphics Primitives

A Distributional Approach To Controlled Text Generation

An example project demonstrating how the Autonomous Learning Library can be used to build new reinforcement learning agents.

Face Library is an open source package for accurate and real-time face detection and recognition

Align and Prompt: Video-and-Language Pre-training with Entity Prompts

[CVPR'20] TTSR: Learning Texture Transformer Network for Image Super-Resolution

Code for Mining the Benefits of Two-stage and One-stage HOI Detection

An elaborate and exhaustive paper list for Named Entity Recognition (NER)

PyTorch implementation of PNASNet-5 on ImageNet