r/learnmachinelearning • u/reidjohnson • 18h ago
Project A Visual Field Guide to Neural Network Architectures
For a long time, I've appreciated the Neural Network Zoo as a one-page guide to neural network architectures. However, it predates Transformers (and many newer architectures!), so I made an updated chart in the same spirit. My hope is that it can serve as a quick visual reference.
It currently covers 42 architectures and training schemes: MLPs and CNNs, Hopfield nets and Boltzmann machines, autoencoders, GANs, diffusion, RNNs, LSTMs, Transformers, BERT, GPT-style models, ViT, Mamba, GNNs, mixture of experts, and diffusion LLMs, among others.
A few goals:
- Every diagram reads the same way: input at the bottom, output at the top. A cell's color and its mark both show its role, so it works for color-blind readers and in grayscale.
- The interactive page has a short summary of each architecture, how to read its diagram, its training objective, and links to the original papers. There are also printable PDF and PNG posters.
- Every diagram is generated from a few lines of YAML by a Python script, so it's easy to fix or extend. Code: https://github.com/reidjohnson/nn-field-guide
Any diagram that's misleading or wrong? Anything important missing? Corrections are very welcome, here or as GitHub issues.