Visualisation of a fully connected neural network classifying MNIST digits

submited by
Style Pass
2021-05-23 17:00:05

The visualisation shows a fully connected artificial neural network classifying MNIST handwritten digits. The net consists of an input layer, an output layer, and no hidden layers. The digits shown are from the test set.

Each column in the visualisation corresponds to an output node, and each output node corresponds to a class (from 0 to 9). The largest value output node (i.e. the tallest bar on the bottom row) tells us which class is predicted. Green highlights the true class, and orange highlights an incorrect prediction.

The weights of the 0 column tend to make 0 digits larger (relative to the other columns), when multiplied and summed. I.e. 0s tend to look more white in the third row of the 0 column of the visualisation, and to have a larger bar in the fourth row.

Also, the weights of the 0 column tend to make a non-zero digit smaller (relative to the column of the non-zero digit), when multiplied and summed. I.e. non-zero digits tend to look more black in the third row of the 0 column of the visualisation, and to have a smaller bar in the fourth row.

Leave a Comment