TrackList: Tracing Back Query Linguistic Diversity for Head and Tail Medical Knowledge in Open Large Language Models
Summary
TrackList is a fine-grained linguistic and statistical analysis pipeline designed to investigate how pre-training data influences Large Language Models' (LLMs) ability to answer medical questions requiring diverse formats. Humans easily produce varied answers like definitions or examples, but LLMs struggle with this diversity in medical contexts. To facilitate this analysis, researchers introduced RefoMed-EN, a new medical dataset comprising 6,170 human-annotated medical terms, each with corresponding definitions, denominations, exemplifications, explanations, or paraphrases. The study explored whether a concept's frequency (head or tail knowledge) affects LLM performance. Evaluation used syntactic and semantic similarity metrics, statistical correlations, and embeddings. Results indicated that LLM answer quality is highest for definition-type questions and lowest for exemplification-type. Furthermore, for definition-type medical questions ("What is multiple sclerosis?"), LLMs tend to paraphrase popular medical concepts more than specialized knowledge.
Key takeaway
For NLP Engineers developing medical LLMs, you should prioritize fine-tuning efforts on generating diverse answer formats, especially for exemplification-type questions where current models perform poorly. Recognize that LLMs are more reliable for definition-based queries, but their tendency to paraphrase popular concepts lessens with specialized medical knowledge. Consider augmenting pre-training data with more linguistically diverse examples to improve performance across all answer types.
Key insights
LLMs struggle with diverse medical question formats, performing best on definitions and worst on exemplifications.
Principles
- LLM performance varies significantly by answer format.
- Head knowledge prompts more paraphrasing in LLMs.
- Pre-training data impacts linguistic diversity in LLM outputs.
Method
TrackList employs a fine-grained linguistic and statistical analysis pipeline to investigate pre-training data's impact on LLM answers to diverse linguistic queries, using syntactic/semantic similarity, correlations, and embeddings.
In practice
- Prioritize definition-type questions for LLMs.
- Expect lower quality for exemplification tasks.
- Analyze pre-training data for linguistic diversity.
Topics
- Large Language Models
- Medical NLP
- Pre-training Data Analysis
- Linguistic Diversity
- RefoMed-EN Dataset
- Head and Tail Knowledge
Best for: AI Scientist, NLP Engineer, Research Scientist
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Paper Index on ACL Anthology.