Neural network animations

During a machine learning project at NTNU, I made some educational animations that I'm quite proud of.

Card image cap

Each point in each frame describes how confident the neural network (NN) is that the point is red. Each frame is one iteration of learning. Eventually, the NN learns to guess that the points appear in a spiral pattern.

Card image cap

Each layer in the NN can be thought of as a linear transformation of all the points. This animation shows how the points are transformed from one layer to the next through the learning process. After the last transformation (Bottom right), a final linear transformation projects each point to a number between 0 and 1, describing the NN's confidence that the point is red.