Hierarchical Modeling of ICD Codes in EHR Foundation Models

In three linesStudy on integrating ICD-10-CM hierarchy into EHR foundation models. Authors compare two approaches: augmenting BERT sequences with hierarchical tokens and injecting hierarchy into graph-based code representations. Experiments on MIMIC-IV and eICU show explicit hierarchy encoding improves predictions in-domain and in cross-dataset transfer.

Summary generated by Claude — human-verified