Lightweight transformers for clinical natural language processing

Omid Rohanian,Mohammadmahdi Nouriborji,Hannah Jauncey,Samaneh Kouchaki,Farhad Nooralahzadeh,Lei Clifton,Laura Merson,David A. Clifton,ISARIC Clinical Characterisation Group
DOI: https://doi.org/10.1017/s1351324923000542
IF: 1.841
2024-01-14
Natural Language Engineering
Abstract:Specialised pre-trained language models are becoming more frequent in Natural language Processing (NLP) since they can potentially outperform models trained on generic texts. BioBERT (Sanh et al., Distilbert, a distilled version of bert: smaller, faster, cheaper and lighter. arXiv preprint arXiv: 1910.01108, 2019) and BioClinicalBERT (Alsentzer et al., Publicly available clinical bert embeddings. In Proceedings of the 2nd Clinical Natural Language Processing Workshop, pp. 72–78, 2019) are two examples of such models that have shown promise in medical NLP tasks. Many of these models are overparametrised and resource-intensive, but thanks to techniques like knowledge distillation, it is possible to create smaller versions that perform almost as well as their larger counterparts. In this work, we specifically focus on development of compact language models for processing clinical texts (i.e. progress notes, discharge summaries, etc). We developed a number of efficient lightweight clinical transformers using knowledge distillation and continual learning, with the number of parameters ranging from million to million. These models performed comparably to larger models such as BioBERT and ClinicalBioBERT and significantly outperformed other compact models trained on general or biomedical data. Our extensive evaluation was done across several standard datasets and covered a wide range of clinical text-mining tasks, including natural language inference, relation extraction, named entity recognition and sequence classification. To our knowledge, this is the first comprehensive study specifically focused on creating efficient and compact transformers for clinical NLP tasks. The models and code used in this study can be found on our Huggingface profile at https://huggingface.co/nlpie and Github page at https://github.com/nlpie-research/Lightweight-Clinical-Transformers, respectively, promoting reproducibility of our results.
computer science, artificial intelligence,linguistics
What problem does this paper attempt to address?
The paper aims to address the issue of large language models being too massive and having too many parameters in Clinical Natural Language Processing (cNLP). Specifically, the goal of the paper is to develop a series of lightweight Transformer models that can maintain high performance while processing clinical texts and reducing resource consumption. The main problems the paper attempts to solve are as follows: 1. **Developing Compact Clinical Language Models**: Many specialized pre-trained language models (such as BioBERT and BioClinicalBERT) perform well in medical NLP tasks, but their large number of parameters and computational requirements limit their applicability. Therefore, the paper develops a series of efficient and lightweight clinical Transformer models with parameter counts ranging from 15 million to 65 million through techniques like Knowledge Distillation and Continual Learning. 2. **Improving Model Usability and Accessibility**: To ensure these models can run on devices with limited memory and processing power, and to enable researchers in low-income countries to utilize these tools for research, the paper focuses on creating efficient and compact models. 3. **Comprehensive Evaluation of Model Performance**: The paper extensively evaluates these lightweight clinical models across multiple standard datasets on various clinical text mining tasks, including Natural Language Inference (NLI), Relation Extraction (RE), Named Entity Recognition (NER), and Sequence Classification (CLS). In summary, the core objective of the paper is to develop efficient and compact clinical language models and validate their effectiveness in various clinical NLP tasks, with the aim of advancing the field further.