SciFig
Tutorials··5 min read

How to Draw Neural Network Diagrams

How to draw neural network architecture diagrams that reviewers can read, covering layer notation, tensor shapes, common conventions, and tool choices.

SciFig Team
SciFig Team
Scientific Illustration Experts

A good architecture diagram shows data flow, layer types, and tensor shapes.

Those three things are what a reader needs to reimplement your model, and they are exactly what most architecture scientific figures omit. A diagram of undifferentiated boxes joined by arrows looks like a neural network without communicating one.

A neural network architecture diagram with layer types, tensor shapes, and skip connections labeled (Figure generated with SciFig)
A neural network architecture diagram with layer types, tensor shapes, and skip connections labeled (Figure generated with SciFig)

Decide the Level of Abstraction First

Three levels are common, and mixing them in one figure is the most frequent mistake.

LevelShowsUse when
Neuron levelIndividual units and weightsTeaching how a small network works
Layer levelLayers as blocks with shapesNearly all research papers
Block levelRepeated modules as single unitsDeep architectures with repetition
For a paper, layer level is almost always right. Neuron-level diagrams do not scale past toy examples, and block level alone hides the detail a reader needs unless you expand the block separately.

See AI Scientific Figure Generation in Action

Watch how researchers create publication-ready scientific figures from text descriptions.

Explore the Tool

What to Label

  • Layer type. Conv, pooling, normalisation, attention, fully connected. Distinguish by shape or colour, and be consistent.
  • Tensor shape. Annotate the output shape of each stage. This is the single most useful addition, and it lets a reader verify dimensions without reading your code.
  • Key hyperparameters. Kernel size, stride, number of heads. Only where they are not obvious.
  • Skip and residual connections. Draw them explicitly. They are structural, not decorative.
  • Loss and supervision points. Where gradients enter, especially with auxiliary losses.

Leave out learning rate, optimizer, and batch size — those are training details and belong in the text or a table.

A comparison of an unlabeled box diagram and the same architecture annotated with layer types and tensor shapes (Figure generated with SciFig)
A comparison of an unlabeled box diagram and the same architecture annotated with layer types and tensor shapes (Figure generated with SciFig)
The same model drawn at neuron, layer, and block level to show which abstraction suits a paper (Figure generated with SciFig)
The same model drawn at neuron, layer, and block level to show which abstraction suits a paper (Figure generated with SciFig)

Conventions Readers Expect

  • Flow left to right, or top to bottom. Pick one and hold it across every figure in the paper.
  • Encode by both shape and colour. Colour alone fails for colourblind readers and in grayscale print.
  • Keep repeated blocks visually identical. If two blocks look different, readers assume they are different.
  • Show the input and output explicitly, with shapes. Many diagrams start at the first conv layer and leave the reader guessing what goes in.

Tools

ToolApproachBest for
draw.io / InkscapeManualFull control, publication finishing
Graphviz / D2Text-definedRegeneratable, version-controlled
PlotNeuralNetLaTeXPolished 3D-style CNN figures
NetronAutomatic from model fileVerifying actual architecture
Netron deserves a special mention: it renders a diagram from an ONNX or similar model file, which means it shows what your model actually is rather than what you believe it to be. It is not a publication figure, but it is an excellent check before you draw one.
A workflow showing a model file inspected in Netron, then rebuilt as a clean publication figure (Figure generated with SciFig)
A workflow showing a model file inspected in Netron, then rebuilt as a clean publication figure (Figure generated with SciFig)

Tip

Draw the diagram from your model definition, not from memory or from an earlier draft of the paper. Architecture figures that disagree with the released code are a common and avoidable source of reproducibility complaints.

Create Scientific Figures Now

Describe your scientific figure in natural language — get publication-ready illustrations in minutes.

Try Free
The neural network diagram generator builds a labelled architecture diagram from a written description of your layers, and text to figure covers the other figures a methods section needs.

Related reading: Visualize Research with AI Text-to-Figure, Scientific AI Prompts: S.S.V.D. Framework.

FAQ

Recommended next

Related scientific figure resources

Go deeper with the most relevant SciFig pages for this article.

Call to action background

Ready to start?

Publication-ready scientific figures, in minutes

Start Creating Free

Free to start · No credit card required · Built for researchers