npj Digital Medicine (Jan 2024)

Large language models to identify social determinants of health in electronic health records

  • Marco Guevara,
  • Shan Chen,
  • Spencer Thomas,
  • Tafadzwa L. Chaunzwa,
  • Idalid Franco,
  • Benjamin H. Kann,
  • Shalini Moningi,
  • Jack M. Qian,
  • Madeleine Goldstein,
  • Susan Harper,
  • Hugo J. W. L. Aerts,
  • Paul J. Catalano,
  • Guergana K. Savova,
  • Raymond H. Mak,
  • Danielle S. Bitterman

DOI
https://doi.org/10.1038/s41746-023-00970-0
Journal volume & issue
Vol. 7, no. 1
pp. 1 – 14

Abstract

Read online

Abstract Social determinants of health (SDoH) play a critical role in patient outcomes, yet their documentation is often missing or incomplete in the structured data of electronic health records (EHRs). Large language models (LLMs) could enable high-throughput extraction of SDoH from the EHR to support research and clinical care. However, class imbalance and data limitations present challenges for this sparsely documented yet critical information. Here, we investigated the optimal methods for using LLMs to extract six SDoH categories from narrative text in the EHR: employment, housing, transportation, parental status, relationship, and social support. The best-performing models were fine-tuned Flan-T5 XL for any SDoH mentions (macro-F1 0.71), and Flan-T5 XXL for adverse SDoH mentions (macro-F1 0.70). Adding LLM-generated synthetic data to training varied across models and architecture, but improved the performance of smaller Flan-T5 models (delta F1 + 0.12 to +0.23). Our best-fine-tuned models outperformed zero- and few-shot performance of ChatGPT-family models in the zero- and few-shot setting, except GPT4 with 10-shot prompting for adverse SDoH. Fine-tuned models were less likely than ChatGPT to change their prediction when race/ethnicity and gender descriptors were added to the text, suggesting less algorithmic bias (p < 0.05). Our models identified 93.8% of patients with adverse SDoH, while ICD-10 codes captured 2.0%. These results demonstrate the potential of LLMs in improving real-world evidence on SDoH and assisting in identifying patients who could benefit from resource support.