Named entity recognition in NLP - Model Metrics & Evaluation

Metrics & Evaluation - Named entity recognition

Which metric matters for Named Entity Recognition and WHY

Named Entity Recognition (NER) finds names like people, places, or dates in text. We want to know how well the model finds these names and how correct those found names are. So, Precision tells us how many found names are actually correct. Recall tells us how many real names the model found out of all names in the text. F1 score balances both precision and recall to give a single score. These metrics matter because NER needs to find as many correct names as possible without too many mistakes.

Confusion Matrix for Named Entity Recognition

For NER, we look at each entity found as either correct or wrong. Here is a simple confusion matrix example for one entity type (e.g., Person):

      | Predicted Entity | Predicted Not Entity |
      |------------------|---------------------|
      | True Positive (TP) = 80  | False Negative (FN) = 20 |
      | False Positive (FP) = 10 | True Negative (TN) = 890 |
    

Total samples = TP + FP + TN + FN = 80 + 10 + 890 + 20 = 1000 tokens.

From this, Precision = 80 / (80 + 10) = 0.89, Recall = 80 / (80 + 20) = 0.80, F1 = 2 * (0.89 * 0.80) / (0.89 + 0.80) ≈ 0.84.

Precision vs Recall Tradeoff with Examples

In NER, if you want to avoid missing any important names (high recall), you might accept some wrong names (lower precision). For example, in medical records, missing a disease name is bad, so recall is key.

On the other hand, if you want to avoid wrong names (high precision), you might miss some names (lower recall). For example, in legal documents, wrongly tagging a name can cause confusion, so precision is important.

Balancing precision and recall with the F1 score helps find a good middle ground.

What Good vs Bad Metric Values Look Like for NER

Good: Precision and recall both above 0.85 means the model finds most names correctly and misses few. F1 score above 0.85 shows balanced performance.

Bad: Precision below 0.5 means many wrong names are found. Recall below 0.5 means many real names are missed. F1 below 0.5 means poor overall performance.

Example: Precision=0.9, Recall=0.9, F1=0.9 is good. Precision=0.3, Recall=0.7, F1=0.42 is bad.

Common Metrics Pitfalls in NER

• Accuracy paradox: Most tokens are not entities, so accuracy can be high even if the model never finds entities.
• Data leakage: If test data is too similar to training data, metrics look better but model may fail on new text.
• Overfitting: Very high training metrics but low test metrics means the model memorizes training names but cannot generalize.
• Ignoring entity boundaries: Partial matches count as wrong, so exact match metrics are stricter but more meaningful.

Self Check: Your model has 98% accuracy but 12% recall on entities. Is it good?

No, this is not good for NER. The high accuracy is misleading because most text is not entities. The very low recall (12%) means the model misses almost all real names. It finds very few entities, so it is not useful for finding names.

Key Result

For Named Entity Recognition, balanced precision and recall (measured by F1 score) best show model quality because they capture correct and missed entity detections.

Practice

1. What is the main goal of Named Entity Recognition (NER) in natural language processing?

easy

A. To find and label names of people, places, and dates in text

B. To translate text from one language to another

C. To summarize long documents into short paragraphs

D. To generate new text based on input

Solution

1. Step 1: Understand NER purpose

   NER is designed to identify and label specific types of information like names, places, and dates in text.

2. Step 2: Compare with other NLP tasks

   Translation, summarization, and text generation are different tasks unrelated to labeling entities.

3. Final Answer:

   To find and label names of people, places, and dates in text -> Option A

4. Quick Check:

   NER = Labeling names in text [OK]

Hint: NER finds names and dates in text, not translations or summaries [OK]

Common Mistakes:

• Confusing NER with translation or summarization
• Thinking NER generates new text
• Believing NER only finds keywords, not entities

2. Which of the following is the correct way to import a Named Entity Recognition pipeline using Hugging Face Transformers in Python?

easy

A. import pipeline from transformers; ner = pipeline('named_entity')

B. from transformers import pipeline; ner = pipeline('ner')

C. from transformers import ner_pipeline; ner = ner_pipeline()

D. import ner from transformers; ner = pipeline('ner')

Solution

1. Step 1: Recall correct import syntax

   The Hugging Face library uses 'from transformers import pipeline' to import the pipeline function.

2. Step 2: Check pipeline usage for NER

   Calling pipeline('ner') creates a named entity recognition pipeline correctly.

3. Final Answer:

   from transformers import pipeline; ner = pipeline('ner') -> Option B

4. Quick Check:

   Correct import and pipeline call = from transformers import pipeline; ner = pipeline('ner') [OK]

Hint: Use 'from transformers import pipeline' and call pipeline('ner') [OK]

Common Mistakes:

• Using incorrect import syntax
• Calling pipeline with wrong task name
• Trying to import non-existent functions

3. Given the following Python code using Hugging Face Transformers NER pipeline:

from transformers import pipeline
ner = pipeline('ner')
text = "Barack Obama was born in Hawaii on August 4, 1961."
results = ner(text)
print(results)

What will be the output type of results?

medium

A. A single string with all entities concatenated

B. A dictionary with entity counts

C. A list of dictionaries with entity details

D. An integer representing number of entities

Solution

1. Step 1: Understand pipeline output format

   The NER pipeline returns a list where each item is a dictionary describing an entity found in the text.

2. Step 2: Check example output structure

   Each dictionary contains keys like 'entity', 'score', 'index', and 'word' describing the entity.

3. Final Answer:

   A list of dictionaries with entity details -> Option C

4. Quick Check:

   NER output = list of entity dictionaries [OK]

Hint: NER pipeline returns list of dicts, not strings or counts [OK]

Common Mistakes:

• Expecting a single string output
• Thinking output is a dictionary summary
• Assuming output is just a count number

4. Consider this code snippet for NER using Hugging Face Transformers:

from transformers import pipeline
ner = pipeline('ner')
text = "Apple is looking at buying U.K. startup for $1 billion"
results = ner(text, grouped_entities=True)
print(results)

What is the likely error or issue here?

medium

A. The text input must be a list, not a string

B. The pipeline call is missing the model parameter

C. There is no error; code runs correctly

D. The argument 'grouped_entities' is invalid and causes a TypeError

Solution

1. Step 1: Check pipeline argument validity

   The 'grouped_entities' argument is not supported in the current pipeline call and will raise a TypeError.

2. Step 2: Confirm correct usage

   To group entities, the argument should be 'aggregation_strategy' with values like 'simple', not 'grouped_entities'.

3. Final Answer:

   The argument 'grouped_entities' is invalid and causes a TypeError -> Option D

4. Quick Check:

   Invalid argument causes error = The argument 'grouped_entities' is invalid and causes a TypeError [OK]

Hint: Check pipeline argument names carefully; 'grouped_entities' is wrong [OK]

Common Mistakes:

• Using unsupported argument names
• Assuming text input must be a list
• Thinking missing model parameter causes error

5. You want to extract all person names and locations from a news article using NER. Which approach best ensures you only get these two entity types from the pipeline output?

hard

A. Filter the NER results by checking if the entity label is 'PER' or 'LOC'

B. Use the pipeline with task='ner' and no filtering

C. Manually search the text for capitalized words

D. Train a new model only on person and location data

Solution

1. Step 1: Understand NER output labels

   NER results include entity labels like 'PER' for person and 'LOC' for location.

2. Step 2: Filter results for desired entities

   Filtering the output by these labels extracts only person and location entities effectively.

3. Final Answer:

   Filter the NER results by checking if the entity label is 'PER' or 'LOC' -> Option A

4. Quick Check:

   Filter by labels 'PER' and 'LOC' to get persons and locations [OK]

Hint: Filter NER output by entity labels to get specific types [OK]

Common Mistakes:

• Not filtering and getting all entity types
• Trying manual text search instead of using labels
• Assuming retraining is needed for filtering