Neural Network from Scratch in the Browser

Try it yourself: https://nn-viz.ameo.design

About

The goal of this project was to understand neural networks better by building them from the ground up. I wanted to be able to see dynamically how choosing different network architectures + training parameters affects network performance and how well networks can various functions.

It allows users to define a neural network infrastructure by adding hidden layers, picking neuron counts + activation function types, and setting training parameters like learning rate. The network then learns one of a variety of selectable target functions from which examples are randomly sampled to train it.

The "response" of the network over the entire range of possible inputs is then plotted as a 3D surface along with the target function to show how well the network has learned.

Technical Details

The neural network implementation itself is built in Rust and compiled to WebAssembly using Wasm SIMD to accelerate training.

The training takes place on a dedicated thread by using a web worker and the excellent Comlink library to communicate between the main/render thread and the training thread.

All the charts + visualizations are created using the excellent echarts library.

The UI is created using react-control-panel which is a React port of the excellent control-panel library for easy GUI creation.

Building + Developing

You'll need Rust nightly with WebAssembly support. You can install Rust via easily via rustup: https://rustup.rs/

Then, add the latest nightly toolchain + switch to it:

rustup default nightly

Add WebAssembly support:

rustup target add wasm32-unknown-unknown

This project uses the just command runner to simplify many tasks. Install it with: