Sentence transformers in Prompt Engineering / GenAI - Model Pipeline Trace

Model Pipeline - Sentence transformers

This pipeline converts sentences into numbers that computers understand well. It helps find how similar sentences are or groups sentences by meaning.

Data Flow - 4 Stages

1Input sentences

1000 sentences→Raw text input→1000 sentences

"I love apples.", "The sky is blue."

↓

2Tokenization

1000 sentences→Split sentences into words or pieces→1000 lists of tokens

[['I', 'love', 'apples', '.'], ['The', 'sky', 'is', 'blue', '.']]

↓

3Embedding generation

1000 lists of tokens→Convert tokens into 768-dimensional vectors using transformer model→1000 lists of token vectors x 768 dimensions

[[0.12, -0.05, ..., 0.33], [0.07, 0.01, ..., -0.22]]

↓

4Pooling

1000 lists of token vectors x 768 dimensions→Combine token vectors into one sentence vector→1000 vectors x 768 dimensions

[[0.10, -0.03, ..., 0.30], [0.05, 0.00, ..., -0.20]]

Training Trace - Epoch by Epoch

Loss
0.9 |*       
0.8 | **     
0.7 |  **    
0.6 |   **   
0.5 |    **  
0.4 |     ** 
0.3 |      **
    +--------
     1 2 3 4 5 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	0.85	0.50	Model starts learning sentence meanings
2	0.65	0.65	Loss decreases, accuracy improves
3	0.50	0.75	Model better understands sentence similarity
4	0.40	0.82	Training converges with good accuracy
5	0.35	0.85	Final epoch with best performance

Prediction Trace - 4 Layers

Layer 1: Input sentence

Layer 2: Tokenization

Layer 3: Transformer embedding

Layer 4: Pooling

Model Quiz - 3 Questions

What does the pooling step do in the sentence transformer pipeline?

A Splits sentences into words

B Combines token vectors into one sentence vector

C Converts sentences to raw text

D Calculates loss during training

Key Insight

Sentence transformers turn sentences into fixed-size number lists that capture meaning. This helps computers compare sentences easily and find similar ones.

Practice

1. What is the main purpose of sentence transformers in AI?

easy

A. To count the number of words in a sentence

B. To translate sentences from one language to another

C. To convert sentences into numbers that computers can understand

D. To generate new sentences from scratch

Solution

1. Step 1: Understand the role of sentence transformers

   Sentence transformers convert sentences into numerical vectors so computers can process them.

2. Step 2: Compare options with this understanding

   Only To convert sentences into numbers that computers can understand describes this conversion; others describe different tasks.

3. Final Answer:

   To convert sentences into numbers that computers can understand -> Option C

4. Quick Check:

   Sentence transformers = convert sentences to numbers [OK]

Hint: Remember: transformers turn text into numbers [OK]

Common Mistakes:

• Confusing sentence transformers with translation models
• Thinking they generate new sentences
• Assuming they only count words

2. Which of the following is the correct way to import a sentence transformer model in Python?

easy

A. from sentence_transformers import sentence_transformer

B. import SentenceTransformer from sentence_transformers

C. import sentence_transformers.SentenceTransformer

D. from sentence_transformers import SentenceTransformer

Solution

1. Step 1: Recall the correct Python import syntax for sentence transformers

   The correct syntax is 'from sentence_transformers import SentenceTransformer' with exact capitalization.

2. Step 2: Check each option for syntax correctness

   from sentence_transformers import SentenceTransformer matches the correct syntax; others have wrong order, case, or module names.

3. Final Answer:

   from sentence_transformers import SentenceTransformer -> Option D

4. Quick Check:

   Correct import syntax = from sentence_transformers import SentenceTransformer [OK]

Hint: Use 'from module import Class' format for imports [OK]

Common Mistakes:

• Swapping import order
• Using wrong capitalization
• Confusing module and class names

3. What will be the output type of the following code snippet?

from sentence_transformers import SentenceTransformer
model = SentenceTransformer('all-MiniLM-L6-v2')
sentence = 'Hello world'
embedding = model.encode(sentence)
print(type(embedding))

medium

A. <class 'list'>

B. <class 'numpy.ndarray'>

C. <class 'str'>

D. <class 'int'>

Solution

1. Step 1: Understand the output of model.encode()

   The encode method returns a numerical vector as a numpy array representing the sentence embedding.

2. Step 2: Identify the type printed

   Printing type(embedding) shows <class 'numpy.ndarray'> because embeddings are numpy arrays.

3. Final Answer:

   <class 'numpy.ndarray'> -> Option B

4. Quick Check:

   model.encode() output type = numpy.ndarray [OK]

Hint: model.encode returns numpy arrays for embeddings [OK]

Common Mistakes:

• Assuming output is a list
• Thinking output is a string
• Expecting an integer type

4. Identify the error in this code snippet using sentence transformers:

from sentence_transformers import SentenceTransformer
model = SentenceTransformer('all-MiniLM-L6-v2')
sentences = ['Hello world', 'Hi there']
embeddings = model.encode(sentences)
print(embeddings.shape)

medium

A. There is no error; the code runs correctly

B. model.encode() cannot take a list of sentences

C. embeddings does not have a shape attribute

D. The model name 'all-MiniLM-L6-v2' is incorrect

Solution

1. Step 1: Check model name validity

   'all-MiniLM-L6-v2' is a valid pre-trained model name for sentence transformers.

2. Step 2: Verify model.encode() input and output

   model.encode() accepts a list of sentences and returns a numpy array with shape attribute.

3. Step 3: Confirm no errors in code

   All syntax and usage are correct; printing embeddings.shape works as expected.

4. Final Answer:

   There is no error; the code runs correctly -> Option A

5. Quick Check:

   Valid model and input = code runs fine [OK]

Hint: model.encode accepts lists and returns arrays with shape [OK]

Common Mistakes:

• Thinking model.encode only accepts single sentences
• Assuming embeddings lack shape attribute
• Believing model name is invalid

5. You want to find the most similar sentence to 'I love machine learning' from a list using sentence transformers. Which approach is best?

hard

A. Encode all sentences, then use cosine similarity to find the closest embedding

B. Compare sentences by counting common words directly

C. Use a translation model to translate sentences before comparison

D. Manually check each sentence for similarity without encoding

Solution

1. Step 1: Understand the goal of similarity search

   Finding the most similar sentence requires comparing sentence meanings numerically.

2. Step 2: Identify the best method for semantic similarity

   Encoding sentences into embeddings and using cosine similarity is the standard and effective approach.

3. Step 3: Evaluate other options

   Counting words or manual checks ignore meaning; translation is unrelated here.

4. Final Answer:

   Encode all sentences, then use cosine similarity to find the closest embedding -> Option A

5. Quick Check:

   Semantic similarity = encode + cosine similarity [OK]

Hint: Use embeddings + cosine similarity for best sentence matching [OK]

Common Mistakes:

• Relying on word count instead of meaning
• Using translation unnecessarily
• Skipping encoding step