HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Biomedical Named Entity Recognition at Scale

Veysel Kocaman; David Talby

Biomedical Named Entity Recognition at Scale

Abstract

Named entity recognition (NER) is a widely applicable natural language processing task and building block of question answering, topic modeling, information retrieval, etc. In the medical domain, NER plays a crucial role by extracting meaningful chunks from clinical notes and reports, which are then fed to downstream tasks like assertion status detection, entity resolution, relation extraction, and de-identification. Reimplementing a Bi-LSTM-CNN-Char deep learning architecture on top of Apache Spark, we present a single trainable NER model that obtains new state-of-the-art results on seven public biomedical benchmarks without using heavy contextual embeddings like BERT. This includes improving BC4CHEMD to 93.72% (4.1% gain), Species800 to 80.91% (4.6% gain), and JNLPBA to 81.29% (5.2% gain). In addition, this model is freely available within a production-grade code base as part of the open-source Spark NLP library; can scale up for training and inference in any Spark cluster; has GPU support and libraries for popular programming languages such as Python, R, Scala and Java; and can be extended to support other human languages with no code changes.

Benchmarks

BenchmarkMethodologyMetrics
named-entity-recognition-ner-on-bc5cdrBLSTM-CNN-Char (SparkNLP)
F1: 89.73
named-entity-recognition-ner-on-bc5cdrSpark NLP
F1: 89.73
named-entity-recognition-ner-on-jnlpbaSpark NLP
F1: 81.29
named-entity-recognition-ner-on-jnlpbaBLSTM-CNN-Char (SparkNLP)
F1: 81.29
named-entity-recognition-ner-on-ncbi-diseaseBLSTM-CNN-Char (SparkNLP)
F1: 89.13
named-entity-recognition-ner-on-ncbi-diseaseSpark NLP
F1: 89.13
named-entity-recognition-on-anatemBLSTM-CNN-Char (SparkNLP)
F1: 89.13
named-entity-recognition-on-bc2gmSpark NLP
F1: 88.75
named-entity-recognition-on-bc4chemdBLSTM-CNN-Char (SparkNLP)
F1: 93.72
named-entity-recognition-on-bc5cdr-chemicalSpark NLP
F1: 94.88
named-entity-recognition-on-bionlp13-cgBLSTM-CNN-Char (SparkNLP)
F1: 85.58
named-entity-recognition-on-linnaeusBLSTM-CNN-Char (SparkNLP)
F1: 86.26
named-entity-recognition-on-linnaeusSpark NLP
F1: 86.26
named-entity-recognition-on-species-800Spark NLP
F1: 80.91
named-entity-recognition-on-species800BLSTM-CNN-Char (SparkNLP)
F1: 80.91

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Biomedical Named Entity Recognition at Scale | Papers | HyperAI