Top-K accuracy in Computer Vision - Model Pipeline Trace

Model Pipeline - Top-K accuracy

This pipeline shows how a computer vision model learns to recognize images and how Top-K accuracy measures if the correct label is among the model's top K guesses.

Data Flow - 6 Stages

1Data in

1000 images x 64 x 64 x 3→Raw images loaded with labels→1000 images x 64 x 64 x 3

Image of a cat with label 'cat'

↓

2Preprocessing

1000 images x 64 x 64 x 3→Normalize pixel values to 0-1 range→1000 images x 64 x 64 x 3

Pixel values scaled from 0-255 to 0-1

↓

3Feature Engineering

1000 images x 64 x 64 x 3→Extract features using convolutional layers→1000 images x 16 x 16 x 32

Feature maps highlighting edges and textures

↓

4Model Trains

1000 images x 16 x 16 x 32→Train CNN classifier to predict 10 classes→1000 predictions x 10 classes

Prediction vector like [0.1, 0.05, 0.6, ..., 0.02]

↓

5Metrics Improve

1000 predictions x 10 classes→Calculate Top-1 and Top-3 accuracy→Accuracy scores

Top-1 accuracy = 0.75, Top-3 accuracy = 0.92

↓

6Prediction

1 image x 64 x 64 x 3→Model outputs probabilities for each class→1 prediction x 10 classes

Output vector [0.05, 0.1, 0.7, ..., 0.01]

Training Trace - Epoch by Epoch

Loss
2.0 |****
1.5 |*** 
1.0 |**  
0.5 |*   
0.0 +----
      1 2 3 4 5 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	1.8	0.35	Model starts learning, accuracy low
2	1.2	0.55	Loss decreases, accuracy improves
3	0.9	0.68	Model getting better at classification
4	0.7	0.75	Accuracy approaching good performance
5	0.6	0.80	Model converging with good accuracy

Prediction Trace - 5 Layers

Layer 1: Input Image

Layer 2: Convolutional Layers

Layer 3: Fully Connected Layer

Layer 4: Softmax Activation

Layer 5: Top-K Accuracy Check

Model Quiz - 3 Questions

What does Top-3 accuracy measure in this model?

A If the correct label is among the top 3 predicted classes

B If the model predicts exactly 3 classes correctly

C If the model's top prediction is correct

D If the model predicts any class with probability above 0.3

Key Insight

Top-K accuracy helps us understand if the model's correct answer is among its best guesses, which is useful when multiple classes look similar. Watching loss decrease and accuracy increase shows the model is learning well.