Pre-trained embeddings (GloVe) in TensorFlow - Model Pipeline Trace

Model Pipeline - Pre-trained embeddings (GloVe)

This pipeline uses pre-trained GloVe word embeddings to convert words into numbers that capture meaning. Then, a simple neural network learns to classify text based on these embeddings.

Data Flow - 4 Stages

1Raw Text Input

1000 samples (sentences)→Input sentences as raw text→1000 samples (sentences)

"I love machine learning"

↓

2Tokenization

1000 samples (sentences)→Split sentences into words and convert to integer indices→1000 samples x 10 words (max sequence length)

[12, 45, 678, 0, 0, 0, 0, 0, 0, 0]

↓

3Embedding Layer with GloVe

1000 samples x 10 words→Map each word index to a 100-dimensional GloVe vector→1000 samples x 10 words x 100 features

[[[0.12, -0.34, ..., 0.56], ..., [0,0,...,0]]]

↓

4Neural Network Layers

1000 samples x 10 x 100→Apply simple layers (e.g., GlobalAveragePooling, Dense)→1000 samples x 2 (class probabilities)

[[0.8, 0.2], [0.1, 0.9], ...]

Training Trace - Epoch by Epoch

Loss
0.7 |****
0.6 |*** 
0.5 |**  
0.4 |*   
0.3 |*   
     1 2 3 4 5 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	0.65	0.60	Model starts learning with moderate accuracy
2	0.48	0.75	Loss decreases, accuracy improves
3	0.38	0.82	Model continues to improve
4	0.32	0.87	Good convergence, accuracy nearing 90%
5	0.28	0.90	Training stabilizes with high accuracy

Prediction Trace - 4 Layers

Layer 1: Tokenization

Layer 2: Embedding Layer (GloVe)

Layer 3: GlobalAveragePooling

Layer 4: Dense Layer with Softmax

Model Quiz - 3 Questions

What does the embedding layer do in this pipeline?

A Splits sentences into words

B Converts words into meaningful number vectors

C Calculates loss during training

D Outputs final class predictions

Key Insight

Using pre-trained GloVe embeddings helps the model understand word meanings from the start, making training faster and more accurate even with less data.

Practice

1. What is the main purpose of using pre-trained GloVe embeddings in TensorFlow models?

easy

A. To convert words into meaningful number vectors using prior knowledge

B. To generate random numbers for model initialization

C. To increase the size of the training dataset

D. To replace the output layer of the model

Solution

1. Step 1: Understand what embeddings do

   Embeddings convert words into vectors that capture meaning from large text data.

2. Step 2: Identify the role of pre-trained GloVe embeddings

   GloVe embeddings provide these vectors based on prior training on big text corpora, helping models understand word meanings without training from scratch.

3. Final Answer:

   To convert words into meaningful number vectors using prior knowledge -> Option A

4. Quick Check:

   Pre-trained embeddings = meaningful word vectors [OK]

Hint: Pre-trained embeddings map words to vectors using learned knowledge [OK]

Common Mistakes:

• Thinking embeddings generate random numbers
• Confusing embeddings with dataset size
• Assuming embeddings replace output layers

2. Which TensorFlow code snippet correctly loads GloVe embeddings into an Embedding layer with fixed weights?

easy

A. embedding_layer = tf.keras.layers.Embedding(input_dim=vocab_size, output_dim=embedding_dim, embeddings_initializer=tf.keras.initializers.Constant(glove_matrix), trainable=False)

B. embedding_layer = tf.keras.layers.Embedding(input_dim=vocab_size, output_dim=embedding_dim, embeddings_initializer='random_uniform', trainable=True)

C. embedding_layer = tf.keras.layers.Dense(units=embedding_dim, activation='relu')

D. embedding_layer = tf.keras.layers.Embedding(input_dim=vocab_size, output_dim=embedding_dim, embeddings_initializer=tf.keras.initializers.RandomNormal(), trainable=False)

Solution

1. Step 1: Identify correct Embedding layer usage for GloVe

   We use Embedding layer with embeddings_initializer set to Constant(glove_matrix) to load pre-trained weights.

2. Step 2: Check trainable flag for fixed embeddings

   Setting trainable=False ensures embeddings are not updated during training, which matches the question.

3. Final Answer:

   embedding_layer = tf.keras.layers.Embedding(input_dim=vocab_size, output_dim=embedding_dim, embeddings_initializer=tf.keras.initializers.Constant(glove_matrix), trainable=False) -> Option A

4. Quick Check:

   Constant initializer + trainable=False = fixed GloVe embeddings [OK]

Hint: Use Constant initializer and trainable=False for fixed GloVe embeddings [OK]

Common Mistakes:

• Using random initializers instead of Constant
• Setting trainable=True when embeddings should be fixed
• Using Dense layer instead of Embedding

3. Given the code below, what will be the shape of embedding_output?

embedding_layer = tf.keras.layers.Embedding(input_dim=10000, output_dim=50, embeddings_initializer=tf.keras.initializers.Constant(glove_matrix), trainable=False)
input_seq = tf.constant([[1, 5, 10], [2, 3, 4]])
embedding_output = embedding_layer(input_seq)

medium

A. (2, 50)

B. (3, 2, 50)

C. (2, 3, 50)

D. (3, 50)

Solution

1. Step 1: Analyze input sequence shape

   Input sequence has shape (2, 3) because it has 2 sequences each with 3 tokens.

2. Step 2: Understand Embedding output shape

   Embedding layer converts each token to a 50-dimensional vector, so output shape is (2, 3, 50).

3. Final Answer:

   (2, 3, 50) -> Option C

4. Quick Check:

   Input shape (2,3) + embedding_dim 50 = (2,3,50) [OK]

Hint: Embedding output shape = input shape + embedding dimension [OK]

Common Mistakes:

• Swapping batch and sequence dimensions
• Forgetting embedding dimension in output shape
• Assuming output shape is 2D instead of 3D

4. Identify the error in this code snippet that loads GloVe embeddings:

embedding_layer = tf.keras.layers.Embedding(input_dim=vocab_size, output_dim=embedding_dim, embeddings_initializer=tf.keras.initializers.Constant(glove_matrix))
embedding_layer.trainable = False
model = tf.keras.Sequential([embedding_layer, tf.keras.layers.Dense(10)])

medium

A. Dense layer cannot follow an Embedding layer

B. Using Constant initializer is incorrect for GloVe embeddings

C. Embedding layer must be the last layer in the model

D. Setting trainable attribute after layer creation does not freeze embeddings

Solution

1. Step 1: Check how to freeze embeddings

   Setting trainable=False must be done during layer creation, not by assigning attribute later.

2. Step 2: Understand effect of late trainable assignment

   Assigning embedding_layer.trainable = False after creation does not properly freeze weights in Keras models.

3. Final Answer:

   Setting trainable attribute after layer creation does not freeze embeddings -> Option D

4. Quick Check:

   Freeze embeddings by trainable=False in constructor [OK]

Hint: Set trainable=False when creating Embedding layer to freeze weights [OK]

Common Mistakes:

• Setting trainable attribute after layer creation
• Thinking Constant initializer is wrong for GloVe
• Believing Embedding must be last layer

5. You want to fine-tune GloVe embeddings on a small dataset. Which approach is best to avoid overfitting while still improving embeddings?

hard

A. Randomly initialize embeddings and keep trainable=False

B. Initialize Embedding layer with GloVe weights, set trainable=True, and use dropout after embeddings

C. Use GloVe embeddings with trainable=False and no dropout

D. Freeze embeddings and add many Dense layers without regularization

Solution

1. Step 1: Understand fine-tuning embeddings on small data

   Fine-tuning means allowing embeddings to update (trainable=True) but small data risks overfitting.

2. Step 2: Use dropout to reduce overfitting

   Applying dropout after embeddings helps regularize and prevent overfitting during fine-tuning.

3. Final Answer:

   Initialize Embedding layer with GloVe weights, set trainable=True, and use dropout after embeddings -> Option B

4. Quick Check:

   Fine-tune + dropout = better small data training [OK]

Hint: Fine-tune with trainable=True plus dropout to avoid overfitting [OK]

Common Mistakes:

• Freezing embeddings when fine-tuning is needed
• Skipping dropout causing overfitting
• Random initialization losing pre-trained knowledge