English

Social Determinants of Health Prediction for ICD-9 Code with Reasoning Models

Machine Learning 2026-01-16 v1 Computation and Language Computers and Society

Abstract

Social Determinants of Health correlate with patient outcomes but are rarely captured in structured data. Recent attention has been given to automatically extracting these markers from clinical text to supplement diagnostic systems with knowledge of patients' social circumstances. Large language models demonstrate strong performance in identifying Social Determinants of Health labels from sentences. However, prediction in large admissions or longitudinal notes is challenging given long distance dependencies. In this paper, we explore hospital admission multi-label Social Determinants of Health ICD-9 code classification on the MIMIC-III dataset using reasoning models and traditional large language models. We exploit existing ICD-9 codes for prediction on admissions, which achieved an 89% F1. Our contributions include our findings, missing SDoH codes in 139 admissions, and code to reproduce the results.

Keywords

Cite

@article{arxiv.2601.09709,
  title  = {Social Determinants of Health Prediction for ICD-9 Code with Reasoning Models},
  author = {Sharim Khan and Paul Landes and Adam Cross and Jimeng Sun},
  journal= {arXiv preprint arXiv:2601.09709},
  year   = {2026}
}

Comments

Published as part of Machine Learning for Health (ML4H) 2025 Findings Track