For clinical data extraction, QLoRA attains accuracy close to LoRA while requiring lower compute resources.
Shakya, P. R.; Khaneja, A.; Wagholikar, K. B.
Show abstract
BackgroundLarge language models (LLMs) can accurately extract structured data from free text, yet fine-tuning for specific clinical tasks is often compute- and memory-intensive. We examine whether Parameter-Efficient Fine-Tuning (PEFT)--updating only a small subset of weights in the LLM-- preserves accuracy on quantized models while further reducing memory and graphical processing unit (GPU) requirements for hardware-limited teams. MethodsWe fine-tuned three Llama-3.1-8B-Instruct variants: (i) a non-quantized low-rank adaptation (LoRA) model and (ii-iii) quantized low-rank adaptation (QLoRA) models initialized from 8-bit and 4-bit quantized bases. We used the ELMTEX corpus of 60,000 clinical summaries extracted from PubMed Central, which included manual annotations for 15 categories. Models were evaluated with naive and advanced prompting to extract data from the corpus for the 15 categories. Advanced prompting involved a detailed task description and three examples selected using similarity scores. Metrics included ROUGE and BERTScore for lexical/semantic alignment, and entity-level precision, recall, and F1 to assess clinical concept extraction. ResultsFine-tuning consistently outperformed prompting alone. LoRA improved metrics by 10-20 points over the base model, while QLoRA improved by 8-14 points--only 2-4 points below LoRA. Quantization lowered the need for resources-- LoRA required 4 GPUs, versus 3 (8-bit) and 2 (4-bit) for QLoRA. Compared with LoRA, 4-bit QLoRA used about two-thirds of the peak GPU RAM. However training of quantized models took 28-32% longer, likely due to dequantization overhead and less-mature library routines. ConclusionPEFT on quantized models preserves most of LoRAs accuracy gains while substantially reducing GPU count and memory footprint, providing a practical path for accurate clinical information extraction in resource-constrained settings. This study was limited to a single architecture (Llama-3.1-8B) and use of clinical summaries that are less complex than routine clinical notes, which constrains generalizability of the results. Future work should test QLoRA across diverse architectures and sizes and on clinical corpora representative of real-world practice.
Matching journals
The top 7 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Evaluating Knowledge Fusion Models on Detecting Adverse Drug Events in Text 94%
- Modular Clinical Decision Support Networks (MoDN)—Updatable, Interpretable, and Portable Predictions for Evolving Clinical Environments 94%
- Uncovering the effects of model initialization on deep model generalization: A study with adult and pediatric chest X-ray images 93%
Similar papers in this journal
- LCD Benchmark: Long Clinical Document Benchmark on Mortality Prediction for Language Models 97%
- Annotation-preserving machine translation of English corpora to validate Dutch clinical concept extraction tools 95%
- Analysis of Eligibility Criteria Clusters Based on Large Language Models for Clinical Trial Design 94%
Similar papers in this journal
- DeepPhe-CR: Natural Language Processing Software Services for Cancer Registrar Case Abstraction 94%
- Actionability of Synthetic Data in a Heterogeneous and Rare Healthcare Demographic; Adolescents and Young Adults (AYAs) with Cancer 92%
- Exploring Integrative Analysis using the BioMedical Evidence Graph 91%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.