Why ResNet and skip connections in Computer Vision? - Purpose & Use Cases

The Big Idea

What if adding more layers made your AI smarter instead of confused? ResNet shows how!

The Scenario

Imagine trying to teach a very deep neural network to recognize images by stacking many layers one after another, hoping it learns better features at each step.

But as you add more layers, the network starts to perform worse, not better.

The Problem

Simply adding more layers makes training slow and unstable.

The network forgets earlier learned features and struggles to improve, causing errors to pile up.

This is like a long chain where a small mistake early on ruins the whole result.

The Solution

ResNet introduces skip connections that let information jump over layers.

This helps the network remember important features from earlier layers and makes training deep networks easier and more reliable.

Before vs After

✗ Before

output = layer3(layer2(layer1(input)))

✓ After

output = layer3(layer2(layer1(input))) + input

What It Enables

With skip connections, we can build very deep networks that learn complex patterns without losing important information.

Real Life Example

ResNet helps self-driving cars recognize objects on the road accurately by using very deep networks that don't forget earlier details.

Key Takeaways

Deep networks can struggle to learn as they get deeper.

Skip connections let information flow smoothly across layers.

ResNet uses this idea to train very deep, powerful models effectively.

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