Dropout layers in TensorFlow - ML Experiment: Train & Evaluate

Experiment - Dropout layers

Problem:We are training a neural network to classify handwritten digits from the MNIST dataset. The current model achieves 99% accuracy on training data but only 75% on validation data.

Current Metrics:Training accuracy: 99%, Validation accuracy: 75%, Training loss: 0.02, Validation loss: 0.85

Issue:The model is overfitting. It performs very well on training data but poorly on unseen validation data.

Your Task

Reduce overfitting by improving validation accuracy to at least 85% while keeping training accuracy below 92%.

You can only add dropout layers to the existing model.

Do not change the dataset or the number of epochs (keep epochs=10).

Keep the batch size the same.

Hint 1

Hint 2

Hint 3

Solution

TensorFlow

import tensorflow as tf
from tensorflow.keras import layers, models

# Load MNIST data
(X_train, y_train), (X_val, y_val) = tf.keras.datasets.mnist.load_data()

# Normalize data
X_train = X_train.astype('float32') / 255.0
X_val = X_val.astype('float32') / 255.0

# Flatten images
X_train = X_train.reshape(-1, 28*28)
X_val = X_val.reshape(-1, 28*28)

# Build model with dropout
model = models.Sequential([
    layers.Dense(512, activation='relu', input_shape=(28*28,)),
    layers.Dropout(0.3),
    layers.Dense(256, activation='relu'),
    layers.Dropout(0.3),
    layers.Dense(10, activation='softmax')
])

model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])

history = model.fit(X_train, y_train, epochs=10, batch_size=128, validation_data=(X_val, y_val))

Added Dropout layers with rate 0.3 after each dense layer except the output layer.

This randomly disables 30% of neurons during training to reduce overfitting.

Results Interpretation

Before adding dropout: Training accuracy was 99%, validation accuracy was 75%. The model memorized training data but failed to generalize.

After adding dropout: Training accuracy dropped to 90%, validation accuracy improved to 87%. The model learned more general patterns and overfitting reduced.

Dropout layers help prevent overfitting by randomly turning off neurons during training. This forces the model to learn more robust features that generalize better to new data.

Bonus Experiment

Try different dropout rates (0.2, 0.4, 0.5) and observe how validation accuracy changes.

💡 Hint

Higher dropout rates increase regularization but too high can underfit. Find a balance for best validation accuracy.

Practice

1. What is the main purpose of a Dropout layer in a TensorFlow model?

easy

A. To randomly ignore some neurons during training to prevent overfitting

B. To increase the number of neurons in the model

C. To speed up the training by skipping layers

D. To permanently remove neurons from the model

Solution

1. Step 1: Understand what dropout does

   Dropout randomly disables some neurons during training to reduce overfitting by preventing the model from relying too much on any one neuron.

2. Step 2: Compare options with dropout purpose

   Only To randomly ignore some neurons during training to prevent overfitting correctly describes this behavior. Options A, B, and D describe incorrect or unrelated effects.

3. Final Answer:

   To randomly ignore some neurons during training to prevent overfitting -> Option A

4. Quick Check:

   Dropout prevents overfitting = C [OK]

Hint: Dropout drops neurons randomly during training to avoid overfitting [OK]

Common Mistakes:

• Thinking dropout speeds up training
• Believing dropout removes neurons permanently
• Confusing dropout with adding neurons

2. Which of the following is the correct way to add a dropout layer with a dropout rate of 0.3 in TensorFlow?

easy

A. tf.keras.layers.Dropout(0.3)

B. tf.keras.DropoutLayer(rate=0.3)

C. tf.layers.dropout(0.3)

D. tf.keras.layers.DropoutLayer(0.3)

Solution

1. Step 1: Recall TensorFlow dropout syntax

   The correct syntax to add dropout is tf.keras.layers.Dropout(rate), where rate is a float between 0 and 1.

2. Step 2: Match options with correct syntax

   Only tf.keras.layers.Dropout(0.3) matches the correct class and parameter style. Options A, B, and C use incorrect class names or module paths.

3. Final Answer:

   tf.keras.layers.Dropout(0.3) -> Option A

4. Quick Check:

   Correct dropout syntax = D [OK]

Hint: Use tf.keras.layers.Dropout(rate) to add dropout [OK]

Common Mistakes:

• Using wrong class names like DropoutLayer
• Calling dropout from tf.layers instead of tf.keras.layers
• Passing rate as a keyword argument incorrectly

3. Consider the following code snippet:

model = tf.keras.Sequential([
  tf.keras.layers.Dense(10, activation='relu'),
  tf.keras.layers.Dropout(0.5),
  tf.keras.layers.Dense(1)
])
model.compile(optimizer='adam', loss='mse')
output = model(tf.ones((1, 5)), training=True).numpy()

What is the shape of output?

medium

A. (1, 10)

B. (1, 1)

C. (5, 1)

D. (10, 1)

Solution

1. Step 1: Analyze model layers and input shape

   The input shape is (1, 5). The first Dense layer outputs (1, 10). Dropout does not change shape. The final Dense layer outputs (1, 1).

2. Step 2: Determine output shape after final layer

   The final Dense layer has 1 unit, so output shape is (1, 1).

3. Final Answer:

   (1, 1) -> Option B

4. Quick Check:

   Final Dense units = output shape = A [OK]

Hint: Output shape matches last Dense layer units, dropout doesn't change shape [OK]

Common Mistakes:

• Assuming dropout changes output shape
• Confusing input batch size with output shape
• Mixing up number of units in layers

4. You wrote this code but get an error:

model = tf.keras.Sequential([
  tf.keras.layers.Dense(64, activation='relu'),
  tf.keras.layers.Dropout(1.5),
  tf.keras.layers.Dense(10)
])

What is the problem and how to fix it?

medium

A. Dropout layer must be last; move it after Dense(10)

B. Dropout requires an integer rate; change 1.5 to 1

C. Dropout rate must be between 0 and 1; change 1.5 to 0.5

D. Dense layers cannot be followed by Dropout

Solution

1. Step 1: Identify dropout rate constraints

   Dropout rate must be a float between 0 and 1 representing the fraction of neurons to drop.

2. Step 2: Fix the invalid dropout rate

   1.5 is invalid; changing it to 0.5 fixes the error.

3. Final Answer:

   Dropout rate must be between 0 and 1; change 1.5 to 0.5 -> Option C

4. Quick Check:

   Dropout rate ∈ [0,1] = B [OK]

Hint: Dropout rate must be between 0 and 1, not more [OK]

Common Mistakes:

• Using dropout rate > 1
• Placing dropout layer incorrectly
• Thinking dropout rate must be integer

5. You want to improve your model's robustness by adding dropout only after the first Dense layer and before the second Dense layer. Which of the following code snippets correctly implements this with a dropout rate of 0.4?

hard

A. tf.keras.Sequential([
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dense(10),
tf.keras.layers.Dropout(0.4)
])

B. tf.keras.Sequential([
tf.keras.layers.Dropout(0.4),
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dense(10)
])

C. tf.keras.Sequential([
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dropout(0.4),
tf.keras.layers.Dense(10)
])

D. tf.keras.Sequential([
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dropout(0.4),
tf.keras.layers.Dense(64, activation='relu'),
tf.keras.layers.Dense(10)
])

Solution

1. Step 1: Identify where dropout should be placed

   Dropout should be after the first Dense layer and before the second Dense layer.

2. Step 2: Check each option's layer order

   tf.keras.Sequential([ tf.keras.layers.Dense(128, activation='relu'), tf.keras.layers.Dropout(0.4), tf.keras.layers.Dense(64, activation='relu'), tf.keras.layers.Dense(10) ]) places dropout correctly between first and second Dense layers. Options B, C, and D place dropout incorrectly before first Dense, after second Dense, or after output layer.

3. Final Answer:

   Dropout placed between first and second Dense layers with rate 0.4 -> Option D

4. Quick Check:

   Dropout after first Dense = A [OK]

Hint: Place dropout exactly between layers you want to regularize [OK]

Common Mistakes:

• Placing dropout before first Dense layer
• Placing dropout after output layer
• Using dropout after last Dense layer instead of between