Exploring the Value of Pre-trained Language Models for Clinical Named Entity Recognition

Research output: Contribution to conferencePaperpeer-review

Abstract

The practice of fine-tuning Pre-trained Language Models (PLMs) from general or domain-specific data to a specific task with limited resources, has gained popularity within the field of natural language processing (NLP). In this work, we re-visit this assumption and carry out an investigation in clinical NLP, specifically Named Entity Recognition (NER) on drugs and their related attributes. We compare Transformer models that are trained from scratch to fine-tuned BERT-based Large Language Models (LLMs) namely BERT, BioBERT, and ClinicalBERT. Furthermore, we examine the impact of an additional Conditional Random Field (CRF) layer on such models to encourage contextual learning. We use n2c2-2018 shared task data for model development and evaluations. The experimental outcomes show that 1) CRF layers improved all language models; 2) referring to BIO-strict span level evaluation using macro-average F1 score, although the fine-tuned LLMs achieved 0.83+ scores, the TransformerCRF model trained from scratch achieved 0.78+, demonstrating comparable performances with much lower cost, e.g. with 39.80% less training parameters; 3) referring to BIO-strict span-level evaluation using weighted-average F1 score, ClinicalBERT-CRF, BERT-CRF, and TransformerCRF exhibited lower score differences, with 97.59%/97.44%/96.84% respectively. 4) applying efficient training by down-sampling for better data distribution further reduced the training cost and need for data, while maintaining similar scores -i.e. around 0.02 points lower compared to using the full dataset. This This TRANSFORMERCRF project is hosted at https://github.com/HECTA-UoM/TransformerCRF

Original languageEnglish
Pages3660-3669
Number of pages10
DOIs
Publication statusPublished - 15 Dec 2023
Event2023 IEEE International Conference on Big Data (BigData) - Sorrento, Italy
Duration: 15 Dec 202318 Dec 2023

Conference

Conference2023 IEEE International Conference on Big Data (BigData)
Period15/12/2318/12/23

Fingerprint

Dive into the research topics of 'Exploring the Value of Pre-trained Language Models for Clinical Named Entity Recognition'. Together they form a unique fingerprint.

Cite this