FlashTorch
A Python visualization toolkit, built with PyTorch, for neural networks in PyTorch.
Neural networks are often described as "black box". The lack of understanding on how neural networks make predictions enables unpredictable/biased models, causing real harm to society and a loss of trust in AI-assisted systems.
Feature visualization is an area of research, which aims to understand how neural networks perceive images. However, implementing such techniques is often complicated.
FlashTorch was created to solve this problem!
You can apply feature visualization techniques (such as saliency maps and activation maximization) on your model, with as little as a few lines of code.
It is compatible with pre-trained models that come with torchvision, and seamlessly integrates with other custom models built in PyTorch.
Interested?
Take a look at the quick 3min intro/demo to FlashTorch below!
Want to try?
Head over to example notebooks on Colab!
Overview
Installation
If you are installing FlashTorch for the first time:
$ pip install flashtorch
Or if you are upgrading it:
$ pip install flashtorch -U
API guide
These are currently available modules.
flashtorch.utils
: some useful utility functions for data handling & transformationflashtorch.utils.imagenet
:ImageNetIndex
class for easy-ish retrieval of class indexflashtorch.saliency.backprop
:Backprop
class for calculating gradientsflashtorch.activmax.gradient_ascent
:GradientAscent
class for activation maximization
You can inspect each module with Python built-in function help
. The output of that is available on Quick API Guide for your convenience.
Quickstart
Use FlashTorch
Below, you can find simple demos to get you started, as well as links to some handy notebooks showing additional examples of using FlashTorch.
flashtorch.utils
)
Image handling (- Image handling notebook
flashtorch.saliency
)
Saliency maps (- Saliency map with backpropagation notebook
- Google Colab version - best for trying it out
Saliency in human visual perception is a subjective quality that makes certain things within the field of view stand out from the rest and grabs our attention.
Saliency maps in computer vision provide indications of the most salient regions within images. By creating a saliency map for neural networks, we can gain some intuition on "where the network is paying the most attention to" in an input image.
Using flashtorch.saliency
module, let's visualize image-specific class saliency maps of AlexNet pre-trained on ImageNet classification tasks.
The network is focusing on the sunken eyes and the round head for this owl.
flashtorch.activmax
)
Activation maximization (- Activation maximization notebook
- Google Colab version - best for trying it out
Activation maximization is one form of feature visualization that allows us to visualize what CNN filters are "looking for", by applying each filter to an input image and updating the input image so as to maximize the activation of the filter of interest (i.e. treating it as a gradient ascent task with filter activation values as the loss).
Using flashtorch.activmax
module, let's visualize images optimized with filters from VGG16 pre-trained on ImageNet classification tasks.
Concepts such as 'eyes' (filter 45) and 'entrances (?)' (filter 271) seem to appear in the conv5_1 layer of VGG16.
Visit the notebook above to see what earlier layers do!
Develop FlashTorch
Here is how to setup a dev environment for FlashTorch.
From the project root:
-
Create a conda environment.
$ conda env create -f environment.yml
-
Activate the environment.
$ conda activate flashtorch
-
Install FlashTorch in a development mode.
$ pip install -e .
-
Run the linter & test suit.
$ flake8 flashtorch tests && pytest
-
Add a kernel to Jupyter notebook.
$ python -m ipykernel install --user --name flashtorch \ --display-name <e.g. flashtorch-dev>
-
Launch Jupyter notebook
$ jupyter notebook
-
Open a notebook in the
./examples
directory. -
From the top menu,
Kernel
->Change kernel
->flashtorch-dev
-
From the top menu,
Cell
->Run All
If the test suit runs and all the cells in the notebook execute - congratulations, you're good to go!
If you encounter any problem setting up the dev environment, please open an issue.
How to contribute
Thanks for your interest in contributing!
Please first head over to the Code of Conduct, which helps set the ground rules for participation in communities and helps build a culture of respect.
Next, please make sure that you have a dev environment set up (see the Develop FlashTorch section above).
Still here? Great! There are many ways to contribute to this project. Get started here.
Resources
Talks & blog posts
-
EBU AI IN PRODUCTION AND DISTRIBUTION WORKSHOP, November 2019 - slide deck
Reading
-
Introduction and overview of feature visualization: Feature Visualization
-
The latest development in feature visualization: Exploring Neural Networks with Activation Atlases
-
Using backpropagation for gradient visualization: Deep Inside Convolutional Networks: Visualising Image Classification Models and Saliency Maps
-
Guided backprobagation: Striving for Simplicity: The All Convolutional Net
-
Activation maximization: Visualizing Higher-Layer Features of a Deep Network
Inspiration
-
pytorch-cnn-visualizations by utkuozbulak
-
keras-vis by raghakot
Citation
Misa Ogura, & Ravi Jain. (2020, January 2).
MisaOgura/flashtorch: 0.1.2 (Version v0.1.2).
Zenodo. http://doi.org/10.5281/zenodo.3596650