chainer-saliency

Saliency calculation examples.

Article

• Library release: visualize saliency map of deep neural network
• NNの予測根拠可視化をライブラリ化する

From left: 1. Classification saliency map visualization of VGG16, CNN model. 2. iris dataset feature importance calculation of MLP model. 3. Water solubility contribution visualization of Graph convolutional network model.

setup

• chainer>=5.0.0 : To use LinkHook
• chainer-chemistry>=0.5.0 : saliency module is introduced from 0.5.0

# please update your chainer (version>=5.0.0 is necessary)
pip install -U chainer

# install chainer-chemistry
pip install -U chainer-chemistry

saliency module usage

# model is chainer.Chain, x is dataset
calculator = GradientCalculator(model)
saliency_samples = calculator.compute(x)
saliency = calculator.aggregate(saliency_samples)
 
visualizer = ImageVisualizer()
visualizer.visualize(saliency)

Basically that's all to show saliency plot like top figure!