Detect person names in text: Part 1 (Results)

Jan ProcházkaDeep Learning, Personal Data, Whitepaper

F1 scores for different software doing personal name detection.

Detecting people’s names is part and parcel of PII discovery. Traditional techniques like regexps and keywords don’t work, because the set of all names is too varied. How do open source Named Entity Recognition (NER) engines compare, and can we do better?

This Part 1 has NER results and benchmarks. There’s also Part 2 with technical neural network details.

Developing a state-of-the-art named entity recognizer

A good named entity recognizer (NER) is essential for detecting names, home addresses, passport scans etc, for purposes of compliance or breach incident management.

One could use a prepared list of names and surnames for this task, but such gazetteer obviously cannot be exhaustive and will fail on sentences like this:

Calvin Klein founded Calvin Klein Company in 1968.”

Humans recognize easily what is a person’s name and what isn’t. For machines, the task is more difficult because they have trouble understanding context. This leads to the famous two types of errors:

  • False positives: words detected as personal names that are not, typically because they’re capitalized ("Skin Games and Jesus Jones were one the first western bands."); as well as
  • False negatives: names missed by the detection, typically because they’re lowercased, foreign or uncommon. Example: "bill carpenter was playing football."

So in order to recognize person names in text it is necessary to know not only what names look like, but also in what context they’re used and have a general domain knowledge.

Why not just use open source?

NER is a well-studied task in academia. Naturally, we turned to open source NER solutions first. We evaluated the most popular ready-made software: Stanford NER and Stanza from the Stanford University, FLAIR from Zalando Research, spaCy from Explosion AI.

To cut the story short, none of these open source tools were precise and fast enough for our purposes. While they work great on well-behaved data such as news articles or Wikipedia, their academic pedigree implodes when applied to the wild, messy documents of the real world.

For this reason we developed our own NER, with special focus on names as they appear in real company documents. The resulting NER is proprietary (part of PII Tools) but we decided to share some technical design tips and results here, in the hope they help others.

In this article, we’ll compare our creation against popular open source options. Additionally, we’ll benchmark a simple gazetteer-based NER that uses a predefined list of names, to serve as a baseline.

Description of tested NERs
 versiontypelanguagesource
listlist of namesmulti-languageinhouse
stanford4.1CRF classifiersinglehttps://nlp.stanford.edu/software/CRF-NER.html
stanza1.1.1neural networksinglehttps://stanfordnlp.github.io/stanza/
flair0.6.1neural networkmulti-languagehttps://github.com/flairNLP/flair
spacy2.3.2neural networkmulti-languagehttps://spacy.io/
pii-tools3.8.0neural networkmulti-languagehttps://pii-tools.com/

Our requirements for the new NER were:

  • multi-language, with special focus on English, Spanish, German and Portuguese/Brazilian;
  • to accept arbitrary document contexts: text coming from PDFs, Word documents, Excel, email, database field, OCR…;
  • efficient, to process large amounts of text quickly on a CPU (no need for specialized GPU hardware) and with a low memory footprint,
  • flexible to be able to evolve the model behaviour: retraining, adding new languages, correcting detection errors;
  • and of course to be accurate, to minimize false positives and negatives.

To compare our NER we evaluated performance on standard benchmark datasets, CoNLL-2002 (for Spanish), CoNLL-2003 (English, German) and LeNER-Br (Portuguese).

We also included a manually annotated openweb dataset, to make sure we test on data that no NER system (including ours!) has seen during training. Text for this dataset was randomly sampled from the English OpenWebTextCorpus. We value results on openweb the most, because OpenWeb reflects the real (messy) data found in real documents the closest, among these public datasets.

Accuracy

We measured F1 scores for person names detected by each software. F1 scores range from 0 to 100, the higher the better. A “hit” (true positive) means the entire name was matched exactly, beginning to end. This is the strictest metric; we specifically don’t calculate success over the number of correctly predicted tokens, or even individual characters.

Performance benchmark [F1 score – higher is better]
 EnglishSpanishGermanPortugueseopenweb
list59.162.354.945.274.5
stanford94.991.193.692.6
stanza82.298.487.686.7
flair95.896.991.362.990.6
spacy86.187.074.142.081.3
pii-tools92.390.689.385.995.7
F1 scores for NER software that detect personal names.

F1 scores of NER software that detect person names.

In all tests our NER is among the top contenders – but keep in mind that accuracy was just one of our 5 design goals. For example, we “lost” a few F1 points on purpose by switching from Tensorflow to TF lite, trading accuracy for a much smaller & faster model.

For the Portuguese language (Brazil, LGPD) and openweb dataset, PII Tools is the clear winner. As mentioned above openweb reflects “real data” the closest, so this is great.

Let’s look at some concrete examples:

Text samples and NER detections (actual expected names are in bold)
 stanfordstanzaflairspacypii-tools
Calvin Klein founded Calvin Klein Company in 1968.okokokokok
Nawab Zulfikar Ali Magsi did not see Shama Parveen Magsi coming.okokokfailok
bill carpenter was playing football.failfailfailokok
Llanfihangel Talyllyn is beautiful.failfailfailfailok
Skin Games and Jesus Jones were one the first western bands.failfailfailfailfail

These samples are rather edge cases, but serve to get an idea of what detectors have to deal with, and how successful they are.

Performance

Now we will look into other aspects of our requirements. Detection performance was measured on CPU, our pii-tools NER was artificially restricted to a single process with a single thread, others were left in default settings.

Speed and memory comparison
 speed short [kB/s]speed long [kB/s]startup time [s]RAM usage [MB]
list350.01100.01.11420
stanford30.030.01.5248
stanza0.41.26.11073
flair0.10.15.25341
spacy70.0200.00.5123
pii-tools35.0230.01.3387

Overall FLAIR and Stanza are definitely out, due to their super slow speed and high RAM usage. A worthy competitor from the performance perspective is spaCy, whose authors put a great deal of effort into optimization. Unfortunately spaCy’s tokenization quirks and opinionated architecture proved too inflexible for our needs.

Likewise, Stanford NER is the most accurate among the open source alternatives, but is quite rigid – it’s really hard to update its models or add a new language. Plus its GNU GPL license won’t be to everyone’s liking.

Flexibility

Our focus on industry use calls for frequent model modifications: adding new languages, new kinds of documents (document contexts in which names may appear), fixing detection errors.

In order to adapt the PII Tools NER model quickly, we built a pipeline that utilizes several “weaker” NERs and automatic translation tools to build a huge training corpus from varied sources. This focus on real-world data, along with a robust automated Tensorflow training pipeline allows adding new languages and controlling NER outputs more easily than the open source solutions.

While developing our pii-tools NER we implemented most components from scratch, including:

  • a large scale annotated dataset (proprietary data)
  • a tokenizer (critical; none of the open source variants do this part well)
  • token features (input to the neural network)
  • convolutional neural network (NN architecture)
  • data augmentation and training pipeline (for grounded model updates)
  • various tricks to push performance or to squeeze a lot of information into a smaller parameter space

If you are interested in these technical aspects, check the follow up article with technical NER details.


Questions? Want to see a live demo? Contact us.