QA with Hugging Face pipeline in NLP - Model Pipeline Trace

Model Pipeline - QA with Hugging Face pipeline

This pipeline uses a pre-trained Hugging Face model to answer questions based on a given text. It reads the question and context, then predicts the answer span from the context.

Data Flow - 4 Stages

1Input

1 question string, 1 context string→Receive question and context text→1 question string, 1 context string

Question: 'Where is the Eiffel Tower?'; Context: 'The Eiffel Tower is in Paris, France.'

↓

2Tokenization

1 question string, 1 context string→Convert text into tokens (numbers) for model input→1 tokenized input sequence (e.g., 30 tokens)

[CLS] Where is the Eiffel Tower? [SEP] The Eiffel Tower is in Paris, France. [SEP]

↓

3Model Inference

1 tokenized input sequence→Model predicts start and end token positions of answer→Start and end token scores arrays

Start scores: [0.1, 0.2, ..., 0.9]; End scores: [0.05, 0.1, ..., 0.85]

↓

4Answer Extraction

Start and end token scores→Select tokens with highest scores and convert back to text→Answer string

'Paris, France'

Training Trace - Epoch by Epoch

Loss
1.2 |*       
0.9 | *      
0.7 |  *     
0.5 |   *    
0.4 |    *   
    +--------
     1 2 3 4 5 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	1.2	0.45	Model starts learning basic patterns
2	0.9	0.60	Loss decreases, accuracy improves
3	0.7	0.72	Model better at locating answers
4	0.5	0.80	Good convergence, stable learning
5	0.4	0.85	Model fine-tuned for QA task

Prediction Trace - 4 Layers

Layer 1: Input

Layer 2: Tokenization

Layer 3: Model Inference

Layer 4: Answer Extraction

Model Quiz - 3 Questions

What does the tokenization step do in the QA pipeline?

A Converts text into numbers the model can understand

B Predicts the answer span in the context

C Extracts the final answer text

D Receives the question and context strings

Key Insight

This visualization shows how a QA model reads a question and context, then finds the answer span by predicting start and end positions. Training improves the model's ability to locate answers accurately, demonstrated by decreasing loss and increasing accuracy.

Practice

1. What does the Hugging Face QA pipeline do when given a question and a context?

easy

A. It translates the question into another language.

B. It summarizes the context without answering the question.

C. It finds the answer to the question from the given context.

D. It generates a new question based on the context.

Solution

1. Step 1: Understand the QA pipeline purpose

   The QA pipeline is designed to find answers from a given text based on a question.

2. Step 2: Match function to options

   Only It finds the answer to the question from the given context. describes finding an answer from the context, which is the pipeline's main job.

3. Final Answer:

   It finds the answer to the question from the given context. -> Option C

4. Quick Check:

   QA pipeline = find answer from context [OK]

Hint: QA pipeline = question + context -> answer [OK]

Common Mistakes:

• Confusing QA with translation or summarization
• Thinking it generates new questions
• Assuming it works without context

2. Which of the following is the correct way to create a QA pipeline using Hugging Face Transformers in Python?

easy

A. import pipeline from transformers
qa = pipeline('qa')

B. from transformers import QA
qa = QA('pipeline')

C. from transformers import question_answering
qa = question_answering()

D. from transformers import pipeline
qa = pipeline('question-answering')

Solution

1. Step 1: Recall correct import and pipeline creation

   The correct import is from transformers import pipeline, then call pipeline('question-answering').

2. Step 2: Check each option syntax

   Only from transformers import pipeline qa = pipeline('question-answering') matches the correct syntax and function call.

3. Final Answer:

   from transformers import pipeline qa = pipeline('question-answering') -> Option D

4. Quick Check:

   Correct import and pipeline call = from transformers import pipeline qa = pipeline('question-answering') [OK]

Hint: Use pipeline('question-answering') from transformers [OK]

Common Mistakes:

• Wrong import statement
• Incorrect pipeline argument
• Using non-existent classes or functions

3. What will be the output of this code snippet?

from transformers import pipeline
qa = pipeline('question-answering')
result = qa(question='Where is the Eiffel Tower?', context='The Eiffel Tower is in Paris.')
print(result['answer'])

medium

A. In Paris

B. Paris

C. The Eiffel Tower

D. Eiffel Tower

Solution

1. Step 1: Understand the question and context

   The question asks for the location of the Eiffel Tower, and the context states it is in Paris.

2. Step 2: Predict the pipeline answer output

   The pipeline extracts the answer span from the context, which is 'Paris'.

3. Final Answer:

   Paris -> Option B

4. Quick Check:

   Answer extracted = Paris [OK]

Hint: Answer is the location mentioned in context [OK]

Common Mistakes:

• Choosing the full phrase instead of the exact answer
• Confusing question with context text
• Expecting the pipeline to generate new text

4. Identify the error in this code snippet that uses the Hugging Face QA pipeline:

from transformers import pipeline
qa = pipeline('question-answering')
result = qa(question='Who wrote Hamlet?', text='Hamlet was written by Shakespeare.')
print(result['answer'])

medium

A. The argument 'text' should be 'context'.

B. The pipeline name should be 'qa' instead of 'question-answering'.

C. The print statement should use result.answer instead of result['answer'].

D. The import statement is incorrect.

Solution

1. Step 1: Check pipeline argument names

   The QA pipeline expects 'question' and 'context' as arguments, not 'text'.

2. Step 2: Verify other parts of the code

   Pipeline name and import are correct; accessing result['answer'] is valid.

3. Final Answer:

   The argument 'text' should be 'context'. -> Option A

4. Quick Check:

   Use 'context' argument for QA pipeline [OK]

Hint: Use 'context' not 'text' for QA input [OK]

Common Mistakes:

• Using 'text' instead of 'context'
• Changing pipeline name incorrectly
• Wrong result access syntax

5. You want to build a QA system that answers questions from multiple documents. Which approach using Hugging Face pipelines is best?

hard

A. Run the QA pipeline separately on each document and pick the answer with highest score.

B. Concatenate all documents into one string and run the QA pipeline once.

C. Use the QA pipeline only on the first document and ignore others.

D. Train a new model from scratch for multiple documents.

Solution

1. Step 1: Understand pipeline input limits

   QA pipelines work best on one context at a time; long concatenated text may reduce accuracy.

2. Step 2: Evaluate options for multiple documents

   Running QA on each document separately and selecting the best answer is effective and practical.

3. Final Answer:

   Run the QA pipeline separately on each document and pick the answer with highest score. -> Option A

4. Quick Check:

   Separate runs + best score = best multi-doc QA [OK]

Hint: Run QA on each doc, choose best answer [OK]

Common Mistakes:

• Concatenating all documents causing context overflow
• Ignoring documents except first
• Unnecessarily retraining models