Evaluation of LLMs in Medical Text Summarization: The Role of Vocabulary Adaptation in High OOV Settings

TL;DR

This study evaluates LLMs in medical summarization, emphasizing vocabulary adaptation's role in high OOV scenarios.

cs.CL 🔴 Advanced 2025-05-27 48 views
Gunjan Balde Soumyadeep Roy Mainack Mondal Niloy Ganguly
NLP medical AI vocabulary adaptation model evaluation high OOV

Key Findings

Methodology

The research employs multi-strategy vocabulary expansion combined with continual pretraining, assessing models like Llama-2 and Llama-3.1 on high OOV and novelty scenarios. Candidate vocabularies are generated from PubMed abstracts and target datasets, then filtered using fragment scores and frequency. Embeddings are trained via LoRA-based continual pretraining on domain-specific texts. Evaluation uses Rouge-L and UMLS concept overlap across datasets (BioASQ, EBM, PubMedQA). Human expert assessments validate the relevance and faithfulness of summaries.

Key Results

  • Vocabulary adaptation strategies outperform baseline models in 7/8 high OOV and novelty scenarios, with an average Rouge-L improvement of 14-19%. ScafFix notably enhances performance in high OOV contexts, with gains of 8.74%-14.64%. Continual pretraining alone (CPT-Only) improves some cases, confirming domain adaptation benefits. Human evaluations show increased relevance and fidelity, indicating better clinical applicability.
  • Across datasets, models with vocabulary adaptation achieve significant gains over unadapted models, especially in high OOV and high novelty settings. The combined strategies reduce vocabulary fragmentation, improve semantic understanding of medical terms, and produce more accurate summaries. Results demonstrate the importance of tailored vocabulary expansion for specialized domains.
  • Expert assessments confirm that summaries generated with vocabulary adaptation are more aligned with medical knowledge, enhancing trustworthiness. The study establishes that vocabulary adaptation is a crucial step for deploying LLMs in clinical environments, addressing core limitations of high OOV rates and specialized terminology.

Significance

This work provides a comprehensive framework for domain-specific vocabulary adaptation, crucial for deploying LLMs in high-stakes medical contexts. It addresses longstanding issues of vocabulary mismatch and fragmentation, enabling models to generate more accurate, relevant summaries. The findings support the development of reliable AI tools for clinical decision support, knowledge extraction, and automated literature review, bridging the gap between general models and specialized medical applications. The integration of human validation underscores the practical relevance and trustworthiness of the proposed methods.

Technical Contribution

The paper introduces a multi-layered vocabulary adaptation framework combining candidate vocabulary generation, filtering, and embedding training via LoRA. It innovates with methods like MEDVOC and ScafFix, which directly select high-frequency medical terms, reducing fragmentation. The approach leverages efficient continual pretraining, enabling domain adaptation with minimal computational overhead. Extensive experiments demonstrate superior performance over traditional fine-tuning, establishing new benchmarks for high OOV and novelty scenarios in medical summarization.

Novelty

This is the first systematic comparison of multiple vocabulary adaptation strategies specifically tailored for medical text summarization under high OOV and novelty conditions. The introduction of ScafFix, which bypasses tokenization fragmentation by directly selecting high-frequency medical terms, represents a novel approach. The integration of domain-specific vocabulary expansion with parameter-efficient continual pretraining offers a new paradigm for domain adaptation in large language models, surpassing prior methods that mainly focused on general or multilingual settings.

Limitations

  • Despite improvements, the models still face challenges with extremely rare or newly emerging medical terms, especially when the vocabulary expansion is limited by the quality of candidate lists.
  • The approach relies heavily on high-quality, domain-specific corpora for vocabulary generation; in low-resource languages or emerging fields, effectiveness may diminish.
  • Experiments are primarily conducted on English texts; cross-lingual and multi-modal applications require further validation.

Future Work

Future research will explore multi-modal domain adaptation, integrating imaging and structured data with text. Developing more intelligent, adaptive vocabulary selection algorithms, possibly leveraging active learning, can further improve efficiency. Extending methods to multilingual medical texts and real-time clinical settings will enhance practical deployment. Additionally, refining models to better handle extremely rare or novel terms remains a key challenge.

AI Executive Summary

This study tackles a critical challenge in deploying large language models (LLMs) for medical text summarization: the high out-of-vocabulary (OOV) rate and the presence of novel terminology. As LLMs grow in size, their vocabulary limitations become apparent, especially in specialized domains like medicine, where new terms and abbreviations frequently emerge. The researchers systematically evaluate multiple vocabulary adaptation strategies, including candidate vocabulary generation from domain-specific corpora, filtering based on fragment scores and frequency, and embedding training via parameter-efficient continual pretraining with LoRA. They introduce two innovative methods—MEDVOC and ScafFix—that directly incorporate high-frequency medical terms, significantly reducing vocabulary fragmentation and enhancing semantic understanding.

Extensive experiments across three benchmark datasets—BioASQ, EBM, and PubMedQA—demonstrate that models with integrated vocabulary adaptation outperform baseline models in high OOV and high novelty scenarios, with average improvements exceeding 14% in Rouge-L scores. Notably, ScafFix excels in high OOV contexts, boosting performance by up to 14.64%. Human expert evaluations confirm that summaries generated with these strategies are more relevant and faithful, aligning better with clinical knowledge.

The significance of this work lies in its comprehensive approach to domain-specific model adaptation, addressing a core bottleneck in medical NLP applications. By reducing vocabulary fragmentation and improving understanding of specialized terms, the methods pave the way for more reliable AI-assisted clinical decision-making, literature summarization, and knowledge extraction. The study’s framework offers a scalable blueprint for adapting large models to other specialized fields, with future directions including multi-modal integration, multilingual extension, and active vocabulary refinement to handle emerging terminology. Overall, this research marks a substantial step toward deploying trustworthy, domain-aware LLMs in real-world healthcare settings.

Deep Analysis

Background

The evolution of NLP has led to the widespread adoption of pre-trained language models (PLMs) like BERT, GPT, and their derivatives, which have revolutionized tasks such as summarization, question answering, and translation. In the medical domain, these models promise to automate literature review, clinical documentation, and decision support. However, the specialized vocabulary, frequent neologisms, and abbreviations in medical texts pose significant challenges. Existing models trained on general corpora often encounter high OOV rates when applied directly to medical data, leading to fragmented tokenization and reduced semantic fidelity. Prior efforts, including domain-specific fine-tuning and vocabulary expansion, have shown promise but lack comprehensive evaluation under high OOV and novelty scenarios. This gap limits the deployment of reliable AI tools in clinical settings, where accuracy and trustworthiness are paramount. The current work aims to systematically analyze and improve vocabulary adaptation strategies, bridging this critical gap.

Core Problem

The core issue addressed is the high OOV rate in medical texts, which causes token fragmentation and hampers the model’s understanding and generation capabilities. Existing models, even large ones like Llama-3.1 with 128K tokens, struggle with over-fragmentation of medical terms, leading to degraded summarization quality. The problem is exacerbated in high novelty scenarios where new terms appear in summaries but not in source documents, further challenging the models’ ability to generate faithful outputs. Overcoming these limitations requires effective vocabulary expansion, better tokenization strategies, and domain-specific pretraining, all while maintaining computational efficiency.

Innovation

The paper introduces a multi-faceted vocabulary adaptation framework that combines candidate vocabulary generation, importance filtering, and embedding training via LoRA. Notably, it proposes ScafFix, which directly selects high-frequency medical terms without relying solely on tokenization-based methods, thereby reducing fragmentation. The integration of these strategies with parameter-efficient continual pretraining enables models to better understand and generate medical summaries, especially in high OOV and novelty scenarios. This approach differs from prior work by directly addressing token fragmentation and leveraging domain-specific corpora for targeted vocabulary expansion, resulting in significant performance gains.

Methodology

  • �� Generate candidate vocabularies from PubMed abstracts and target datasets, focusing on high OOV medical terms.
  • �� Filter candidates using fragment scores and frequency to select important tokens.
  • �� Expand the model vocabulary by adding selected tokens, employing methods like MEDVOC and ScafFix.
  • �� Initialize embeddings of new tokens as averages of existing subword embeddings.
  • �� Perform continual pretraining using LoRA, either end-to-end or in two stages, on domain-specific texts.
  • �� Fine-tune models on medical summarization datasets, evaluating with Rouge-L and Concept-Score.
  • �� Conduct human evaluations to assess relevance and faithfulness of summaries.

Experiments

The experimental setup involves three datasets—BioASQ, EBM, and PubMedQA—covering query-focused medical summarization tasks. Baselines include models without vocabulary adaptation, with continual pretraining, and with proposed adaptation strategies. Hyperparameters include a batch size of 32, learning rate of 1e-4, with pretraining on 20K PubMed abstracts using LoRA. Evaluation metrics encompass Rouge-L for informativeness and UMLS Concept-Score for faithfulness. Multiple ablation studies compare different vocabulary filtering and training strategies, analyzing performance across high OOV and high novelty thresholds. Results demonstrate consistent improvements with vocabulary adaptation, especially ScafFix, across datasets and scenarios.

Results

The results show that vocabulary adaptation strategies improve model performance in 7/8 high OOV and high novelty scenarios, with average Rouge-L gains of 14-19%. ScafFix consistently outperforms other methods in high OOV contexts, with improvements up to 14.64%. Continual pretraining alone offers benefits but is less effective than combined strategies. Human evaluations confirm that summaries are more relevant and faithful when vocabulary adaptation is employed, indicating better alignment with medical knowledge. These findings validate the importance of targeted vocabulary expansion for domain-specific NLP tasks.

Applications

The methods can be directly applied to clinical documentation summarization, medical literature review, and automated knowledge extraction. They enable models to better understand domain-specific terminology, improving accuracy and trustworthiness. Long-term, these strategies could facilitate real-time clinical decision support, personalized medicine, and AI-assisted diagnostics, especially as models adapt to rapidly evolving medical vocabularies and multilingual contexts.

Limitations & Outlook

Despite significant improvements, the approach depends on high-quality, domain-specific corpora for vocabulary generation, which may not be available in low-resource languages or emerging fields. The methods still face challenges with extremely rare or newly emerging terms. Computational costs, although reduced via LoRA, remain non-trivial for large-scale deployment. Cross-lingual and multi-modal validation is needed to generalize findings beyond English medical texts. Future work should focus on adaptive, real-time vocabulary refinement and broader domain adaptation.

Plain Language Accessible to non-experts

Imagine you’re in a big factory that makes all kinds of gadgets. Most workers know the usual parts, but sometimes new special parts come in that no one has seen before. When a machine breaks, workers need to find the right parts quickly to fix it. If they don’t know the new part names, they have to guess or break down the problem into smaller, less helpful pieces. This slows everything down. Now, think of a language model like this factory. It has a list of words it understands, but many new medical words are missing or split into tiny pieces, making it hard for the model to understand and summarize medical texts. The researchers found ways to add these new words directly into the model’s vocabulary, like giving the workers a new parts list. This helps the model understand complex medical terms better, so it can produce clearer, more accurate summaries, just like the factory can fix machines faster with the right parts list. This work makes AI smarter at understanding specialized language, helping doctors and researchers get better information faster.

ELI14 Explained like you're 14

Imagine you’re playing a video game where your character learns new words and tricks as you go along. Sometimes, the game introduces special words or items that you’ve never seen before, and if your character doesn’t know what they are, it’s hard to use them properly. That’s what happens with language models when they encounter medical terms they haven’t learned yet. They tend to break these words into smaller pieces, making it hard to understand the full meaning. The researchers in this study found a clever way to teach the model these new words directly, kind of like giving your game character a new vocabulary book. Once the character knows more words, it can understand and explain medical texts much better, like a student who finally learns the tricky vocabulary for a test. This makes the AI much more helpful for doctors and scientists, who need accurate summaries of complicated medical papers. It’s like upgrading your game character so it can handle tougher challenges!

Abstract

Large Language Models (LLMs) recently achieved great success in medical text summarization by simply using in-context learning. However, these recent efforts do not perform fine-grained evaluations under difficult settings where LLMs might fail. They typically report performance scores over the entire dataset. Through our benchmarking study, we show that LLMs show a significant performance drop for data points with high concentration of out-of-vocabulary (OOV) words or with high novelty. Vocabulary adaptation is an intuitive solution to this vocabulary mismatch issue where the LLM vocabulary gets updated with certain expert domain (here, medical) words or subwords. An interesting finding from our study is that Llama-3.1, even with a vocabulary size of around 128K tokens, still faces over-fragmentation issue with medical words. To that end, we show vocabulary adaptation helps improve the LLM summarization performance even in difficult settings. Through extensive experimentation of multiple vocabulary adaptation strategies, two continual pretraining strategies, and three benchmark medical summarization datasets, we gain valuable insights into the role of vocabulary adaptation strategies for customizing LLMs to the medical domain. We also performed a human evaluation study with medical experts where they found that vocabulary adaptation results in more relevant and faithful summaries. Our codebase is made publicly available at https://github.com/gb-kgp/LLM-MedicalSummarization-Benchmark.

cs.CL