Considerations for health care institutions training large language models on electronic health records

Weipeng Zhou,Danielle Bitterman,Majid Afshar,Timothy A. Miller
2023-08-24
Abstract:Large language models (LLMs) like ChatGPT have excited scientists across fields; in medicine, one source of excitement is the potential applications of LLMs trained on electronic health record (EHR) data. But there are tough questions we must first answer if health care institutions are interested in having LLMs trained on their own data; should they train an LLM from scratch or fine-tune it from an open-source model? For healthcare institutions with a predefined budget, what are the biggest LLMs they can afford? In this study, we take steps towards answering these questions with an analysis on dataset sizes, model sizes, and costs for LLM training using EHR data. This analysis provides a framework for thinking about these questions in terms of data scale, compute scale, and training budgets.
Computers and Society,Artificial Intelligence,Computation and Language
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how healthcare institutions can train large - language models (LLMs) on electronic health record (EHR) data. Specifically: 1. **Training from scratch or fine - tuning existing models**: The paper explores whether healthcare institutions should choose to train an LLM from scratch or fine - tune based on existing open - source models. 2. **Maximum model scale under budget constraints**: For healthcare institutions with limited budgets, the research analyzes the maximum LLM scale that can be afforded under a given budget. 3. **Relationship between data volume and model scale**: By analyzing the relationships among data set size, model size, and training cost, the paper provides a framework to help understand these key issues. 4. **Computational resource requirements**: The research also discusses the computational resources required for training models of different scales, including the number of GPUs and time costs. Through these analyses, the paper aims to provide a scientific basis for healthcare institutions in the decision - making process, helping them better understand the opportunities and challenges in the development and deployment of large - language models.