Fine Tuning Large Language Models for Medicine: The Role and Importance of Direct Preference Optimization

Thomas Savage,Stephen Ma,Abdessalem Boukil,Vishwesh Patel,Ekanath Rangan,Ivan Rodriguez,Jonathan H Chen
2024-09-21
Abstract:Large Language Model (LLM) fine tuning is underutilized in the field of medicine. Two of the most common methods of fine tuning are Supervised Fine Tuning (SFT) and Direct Preference Optimization (DPO), but there is little guidance informing users when to use either technique. In this investigation, we compare the performance of SFT and DPO for five common natural language tasks in medicine: Classification with text data, Classification with numeric data, Clinical Reasoning, Summarization, and Clinical Triage. We find that SFT alone is sufficient for Classification with text data, whereas DPO improves performance for the more complex tasks of Clinical Reasoning, Summarization and Clinical Triage. Our results establish the role and importance of DPO fine tuning within medicine, and consequently call attention to current software gaps that prevent widespread deployment of this technique.
Computation and Language,Artificial Intelligence
What problem does this paper attempt to address?
The paper attempts to address the issue of how to effectively fine-tune large language models (LLMs) in the medical field to improve their performance on specific tasks. Specifically, the authors compare two common fine-tuning methods—Supervised Fine-Tuning (SFT) and Direct Preference Optimization (DPO)—across five common natural language processing tasks: 1. **Text-based classification**: Identifying cases describing urinary tract infections (UTI). 2. **Numerical-based classification**: Interpreting urine electrolyte results for patients with hyponatremia. 3. **Clinical reasoning**: Diagnosing and selecting treatment plans. 4. **Summary generation**: Compressing discharge summary notes into 2-3 sentences. 5. **Clinical triage**: Classifying patient messages by urgency and the type of responder needed. Through these tasks, the authors aim to determine in which tasks SFT and DPO are respectively more effective and explore the prospects of these fine-tuning methods in the medical field. Additionally, the paper highlights the current shortcomings of software tools in supporting DPO fine-tuning, emphasizing the importance of improving these tools to facilitate the broader application of DPO technology.