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

Permeability Prediction Via Multi Scale 3D CNN

Pytorch implementation of Depth-conditioned Dynamic Message Propagation forMonocular 3D Object Detection

Codes accompanying the paper "Believe What You See: Implicit Constraint Approach for Offline Multi-Agent Reinforcement Learning" (NeurIPS 2021 Spotlight

State-Relabeling Adversarial Active Learning

Elevation Mapping on GPU.

An Abstract Cyber Security Simulation and Markov Game for OpenAI Gym

A highly modular PyTorch framework with a focus on Neural Architecture Search (NAS).

Repository of best practices for deep learning in Julia, inspired by fastai

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch

Pytorch library for fast transformer implementations

GAN example for Keras. Cuz MNIST is too small and there should be something more realistic.

tsai is an open-source deep learning package built on top of Pytorch & fastai focused on state-of-the-art techniques for time series classification, regression and forecasting.

A modular PyTorch library for optical flow estimation using neural networks

A collection of differentiable SVD methods and also the official implementation of the ICCV21 paper "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance Pooling?"

This implementation contains the application of GPlearn's symbolic transformer on a commodity futures sector of the financial market.

ICCV2021 Expert-Goal Trajectory Prediction

Official repository for HOTR: End-to-End Human-Object Interaction Detection with Transformers (CVPR'21, Oral Presentation)

Multi-Scale Geometric Consistency Guided Multi-View Stereo

Seeing if I can put together an interactive version of 3b1b's Manim in Streamlit

PyTorch implementation of UNet++ (Nested U-Net).