Skip to content

Implementation of layer-wise relevance propagation on a convolutional neural network trained on the MNIST handwritten digit dataset.

Notifications You must be signed in to change notification settings

seanseungbeomlee/PyTorch-CNN-LRP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Layer-wise Relevance Propagation for Convolutional Neural Networks

Implementation of layer-wise relevance propagation on a convolutional neural network trained on the MNIST handwritten digit dataset. Example of heat maps for select layers for a sample input image of 7:
alt text
The corresponing relevance scores for every output class for the smaple input image of 7:
alt text
Example of heat maps for select layers for a sample input image of 8:
alt text
The corresponing relevance scores for every output class for the smaple input image of 8:
alt text\

About

Implementation of layer-wise relevance propagation on a convolutional neural network trained on the MNIST handwritten digit dataset.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages