Introduction to Convolutional Neural Network Explainer

Introduction to Convolutional Neural Network Explainer
Photo by Vincent Yuan @USA / Unsplash

1 Introduction

In the article Introduction to TensorFlow Playground, the author introduced a product that uses visualization to understand neural networks (CNNs). This article will share a similar website - CNN Explainer, which focuses on explaining Convolutional Neural Networks in a visual way. The product shows convolutional neural networks and related knowledge points in a very intuitive way. The CNN Explainer website address is: https://poloclub.github.io/cnn-explainer/.

2 Product Features

The header of the CNN Explainer website is a convolutional neural network used for image classification. The input is the three channels of the image (red, green, and blue), and the style of the image after each layer of calculation in the network is also displayed. Hover over the mouse to see the input of each convolution kernel.

CNN Interpreter

Clicking on a convolutional layer displays the convolution operation, and the convolution kernel scans the input image step by step to form the output:

Convolution operation

Clicking on the exclamation mark will take you to the corresponding documentation and examples. For example, for the convolution operation, you can experience its working principle by adjusting the parameters of the input and convolution kernel. The animation will change according to the input, and it also comes with detailed document instructions and reference links. The developers of the product are really attentive!

Interactive understanding of convolution operation

The operation diagram and formula of the ReLU activation function commonly used in convolutional layers can also be displayed by clicking on the corresponding layer:

ReLU activation function diagram

Similarly, the diagram and formula of the MaxPooling layer are also displayed in an imageable way, and the maximum value in the input is taken out as the output:

Max Pooling diagram

Finally, the output layer uses a different degree of highlight to display the predicted category of the model.

3 Summary

The CNN Explainer shows the important technical principles such as convolution operation, ReLU function, Softmax function, pooling, and flattening layer in a very vivid way. You can also upload your own images for the model to make predictions, so please try it out for yourself! I hope this share is helpful to you and welcome to leave a message in the comment section for discussion!