Information Extraction from Clinical Notes: Are We Ready to Switch to Large Language Models?

Yan Hu,Xu Zuo,Yujia Zhou,Xueqing Peng,Jimin Huang,Vipina K. Keloth,Vincent J. Zhang,Ruey-Ling Weng,Qingyu Chen,Xiaoqian Jiang,Kirk E. Roberts,Hua Xu
2024-11-18
Abstract:Backgrounds: Information extraction (IE) is critical in clinical natural language processing (NLP). While large language models (LLMs) excel on generative tasks, their performance on extractive tasks remains debated. Methods: We investigated Named Entity Recognition (NER) and Relation Extraction (RE) using 1,588 clinical notes from four sources (UT Physicians, MTSamples, MIMIC-III, and i2b2). We developed an annotated corpus covering 4 clinical entities and 16 modifiers, and compared instruction-tuned LLaMA-2 and LLaMA-3 against BiomedBERT in terms of performance, generalizability, computational resources, and throughput to BiomedBERT. Results: LLaMA models outperformed BiomedBERT across datasets. With sufficient training data, LLaMA showed modest improvements (1% on NER, 1.5-3.7% on RE); improvements were larger with limited training data. On unseen i2b2 data, LLaMA-3-70B outperformed BiomedBERT by 7% (F1) on NER and 4% on RE. However, LLaMA models required more computing resources and ran up to 28 times slower. We implemented "Kiwi," a clinical IE package featuring both models, available at <a class="link-external link-https" href="https://kiwi.clinicalnlp.org/" rel="external noopener nofollow">this https URL</a>. Conclusion: This study is among the first to develop and evaluate a comprehensive clinical IE system using open-source LLMs. Results indicate that LLaMA models outperform BiomedBERT for clinical NER and RE but with higher computational costs and lower throughputs. These findings highlight that choosing between LLMs and traditional deep learning methods for clinical IE applications should remain task-specific, taking into account both performance metrics and practical considerations such as available computing resources and the intended use case scenarios.
Computation and Language
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is to evaluate the performance and generality of large - language models (LLMs) in clinical information extraction (IE) tasks and compare them with existing BERT - based models. Specifically, the research focuses on the following two aspects: 1. **Performance evaluation**: By performing instruction - tuning on LLMs (such as LLaMA - 2 and LLaMA - 3) on multiple datasets and comparing them with the BiomedBERT model, evaluate the performance of LLMs in named - entity recognition (NER) and relation extraction (RE) tasks. The research pays special attention to the performance on limited training data and cross - institutional datasets to explore the potential of LLMs in actual clinical applications. 2. **Resource consumption and processing efficiency**: In addition to performance evaluation, the paper also analyzes in detail the computational resource requirements (such as GPU usage time, memory usage, energy consumption, and carbon emissions) and processing speed (throughput) of different models during training and inference, in order to help healthcare institutions select appropriate models according to their computational resources. Through these evaluations, the paper aims to provide guidance for researchers and practitioners in the healthcare field on whether and how to adopt large - language models for clinical information extraction, while emphasizing the importance of comprehensively considering performance, processing efficiency, and available computational resources when selecting models.