Skip to content

fuweijie/DRE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Differentiated Explanation
This is the pytorch implemention for paper “Differentiated Explanation of Deep Neural Networks with Skewed Distributions”.

We propose a simple but efficent approch for the differentiated explanations of black-box classifiers. To do this, we introduce a trainable relevance estimator that produces relevance scores in a skewed distribution. Specifically, we present the concept of distribution controllers and integrate it with a neural network to directly guide the distribution of relevance scores. By analyzing the effect of the skewness of distributions, we develop the controllers with right-skewed distributions for differentiated saliency maps. Then we introduce the classification loss to optimize the estimator. The benefit of this strategy is to better mimic the behavior of deep neural networks without non-trivial hyperparameter tuning, leading to higher faithfulness of explanation.

Running the Demo
Our code is implemented with:
  matplotlib==3.1.3 
  more-itertools==8.3.0 
  numpy==1.18.1 
  pillow==7.1.2 
  pytorch==1.1.0 
  scikit-image==0.16.2 
  scikit-learn==0.22.1 
  scipy==1.4.1 
  torchvision==0.3.0 
python saliency_demo.py
Cite this:
  @ARTICLE{9316988,  author={W. {Fu} and M. {Wang} and M. {Du} and N. {Liu} and S. {Hao} and X. {Hu}},  
  journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},   
  title={Differentiated Explanation of Deep Neural Networks with Skewed Distributions},   
  year={2021},  
  volume={},  
  number={},  
  pages={1-1},  
  doi={10.1109/TPAMI.2021.3049784}}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages