ResNet and skip connections in Computer Vision - Model Pipeline Trace

Model Pipeline - ResNet and skip connections

This pipeline shows how a ResNet model uses skip connections to help the model learn better by allowing information to flow directly across layers. This helps avoid problems like forgetting earlier information.

Data Flow - 6 Stages

1Input Image

224 rows x 224 columns x 3 channels→Raw image pixels→224 rows x 224 columns x 3 channels

A color photo of a cat

↓

2Initial Convolution + Pooling

224 x 224 x 3→Apply convolution filters and reduce size with max pooling→56 x 56 x 64

Feature maps highlighting edges and textures

↓

3Residual Block with Skip Connection

56 x 56 x 64→Two convolution layers + add input (skip connection)→56 x 56 x 64

Feature maps combined with original input to preserve information

↓

4Stacked Residual Blocks

56 x 56 x 64→Repeat residual blocks multiple times→56 x 56 x 256

Deeper features with preserved original signals

↓

5Global Average Pooling

56 x 56 x 256→Average each feature map to single value→1 x 1 x 256

Summary vector representing the image

↓

6Fully Connected Layer

256→Classify features into categories→1000 classes

Probabilities for 1000 object categories

Training Trace - Epoch by Epoch

Loss
2.0 |\
1.8 | \
1.6 |  \
1.4 |   \
1.2 |    \
1.0 |     \
0.8 |      \
0.6 |       \
0.4 |        \
0.2 |         \
0.0 +----------
      1 5 10 15 20 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	1.8	0.35	Model starts learning basic features
5	1.2	0.55	Skip connections help gradients flow better
10	0.8	0.70	Model learns deeper features without forgetting
15	0.5	0.82	Training loss decreases steadily, accuracy improves
20	0.35	0.88	Model converges with good accuracy

Prediction Trace - 6 Layers

Layer 1: Input Image

Layer 2: Initial Convolution + Pooling

Layer 3: Residual Block with Skip Connection

Layer 4: Stacked Residual Blocks

Layer 5: Global Average Pooling

Layer 6: Fully Connected Layer

Model Quiz - 3 Questions

What is the main purpose of the skip connection in a ResNet block?

A To reduce the number of layers

B To let information flow directly and avoid forgetting

C To increase the image size

D To add noise to the data

Key Insight

Skip connections in ResNet help the model learn deeper layers without losing earlier information. This improves training by allowing gradients to flow better, leading to faster convergence and higher accuracy.

Practice

1. What is the main purpose of skip connections in a ResNet model?

easy

A. To replace convolutional layers with fully connected layers

B. To reduce the number of layers in the network

C. To allow information to flow directly across layers, helping training

D. To increase the size of the input images

Solution

1. Step 1: Understand skip connections role

   Skip connections let the input bypass some layers and add directly to the output, helping information flow.

2. Step 2: Connect to training deep networks

   This helps avoid problems like vanishing gradients, making training deep networks easier and more accurate.

3. Final Answer:

   To allow information to flow directly across layers, helping training -> Option C

4. Quick Check:

   Skip connections improve training by direct flow [OK]

Hint: Skip connections let info skip layers to ease training [OK]

Common Mistakes:

• Thinking skip connections reduce layers
• Confusing skip connections with input size changes
• Assuming skip connections replace convolution

2. Which of the following is the correct way to add a skip connection in PyTorch between input tensor x and output tensor out?

easy

A. out = x - out

B. out = x * out

C. out = x / out

D. out = x + out

Solution

1. Step 1: Recall skip connection operation

   Skip connections add the input tensor to the output tensor element-wise.

2. Step 2: Match with correct syntax

   The addition operation out = x + out correctly implements the skip connection.

3. Final Answer:

   out = x + out -> Option D

4. Quick Check:

   Skip connection = addition [OK]

Hint: Skip connections use addition, not multiplication or division [OK]

Common Mistakes:

• Using multiplication instead of addition
• Using subtraction or division which breaks skip connection
• Confusing order of operands

3. Consider this PyTorch code snippet for a ResNet block:

import torch
import torch.nn as nn

class SimpleResBlock(nn.Module):
    def __init__(self):
        super().__init__()
        self.conv = nn.Conv2d(3, 3, kernel_size=3, padding=1)
        self.relu = nn.ReLU()
        self.conv.weight.data.fill_(0.0)
        self.conv.bias.data.fill_(1.0)

    def forward(self, x):
        out = self.conv(x)
        out = self.relu(out)
        out = out + x
        return out

block = SimpleResBlock()
input_tensor = torch.ones(1, 3, 5, 5)
output = block(input_tensor)
print(output[0,0,0,0].item())

What will be printed?

medium

A. 2.0

B. 1.0

C. 0.0

D. An error occurs

Solution

1. Step 1: Analyze convolution output

   The convolution with kernel size 3 and padding 1 keeps the input size. Since input is all ones, convolution output will be some positive values (not zero).

2. Step 2: Add input and apply ReLU

   ReLU keeps positive values. Then adding input tensor (all ones) increases values. So output values > 1.

3. Final Answer:

   2.0 -> Option A

4. Quick Check:

   Output = conv + input > 1 [OK]

Hint: Skip connection adds input, so output > input value [OK]

Common Mistakes:

• Assuming output equals input without addition
• Ignoring padding effect on size
• Expecting zero or error due to shape mismatch

4. You wrote this PyTorch code for a ResNet block but get a runtime error:

def forward(self, x):
    out = self.conv(x)
    out = self.relu(out)
    out = out + x
    return out

The error says: "The size of tensor a (64) must match the size of tensor b (128) at non-singleton dimension 1." What is the likely cause?

medium

A. The convolution changes the number of channels, so shapes don't match for addition

B. ReLU changes tensor shape unexpectedly

C. Input tensor is None

D. The addition operator is used incorrectly

Solution

1. Step 1: Understand error message

   The error says channel sizes differ (64 vs 128), so tensors can't be added element-wise.

2. Step 2: Check convolution output channels

   If convolution changes channels from 64 to 128, input and output shapes differ, causing addition error.

3. Final Answer:

   The convolution changes the number of channels, so shapes don't match for addition -> Option A

4. Quick Check:

   Channel mismatch causes addition error [OK]

Hint: Check channel sizes before adding tensors [OK]

Common Mistakes:

• Blaming ReLU for shape errors
• Ignoring channel dimension mismatch
• Assuming addition works regardless of shape

5. In a ResNet architecture, if the input tensor has shape (batch_size, 64, 32, 32) and the convolution layer in the block changes channels to 128 with stride 2, how can you correctly implement the skip connection?

hard

A. Add input tensor directly without changes

B. Use a 1x1 convolution with stride 2 on the input to match shape before addition

C. Use max pooling on output tensor before addition

D. Skip connection is not needed in this case

Solution

1. Step 1: Identify shape mismatch

   Input has 64 channels and size 32x32; output has 128 channels and size 16x16 due to stride 2.

2. Step 2: Match shapes for addition

   To add tensors, input must be transformed to 128 channels and 16x16 size, done by 1x1 convolution with stride 2.

3. Final Answer:

   Use a 1x1 convolution with stride 2 on the input to match shape before addition -> Option B

4. Quick Check:

   Match shape with 1x1 conv before skip add [OK]

Hint: Use 1x1 conv to match shape for skip connection [OK]

Common Mistakes:

• Adding tensors with different shapes directly
• Using pooling on output instead of input
• Skipping skip connection when channels differ