MEDFuse: Multimodal EHR Data Fusion with Masked Lab-Test Modeling and Large Language Models
Electronic health records (EHRs) are multimodal by nature, consisting of structured tabular features like lab tests and unstructured clinical notes. In real-life clinical practice, doctors use complementary multimodal EHR data sources to get a clearer picture of patients' health and support clinical decision-making. However, most EHR predictive models do not reflect these procedures, as they either focus on a single modality or overlook the inter-modality interactions/redundancy. In this work, we propose MEDFuse, a Multimodal EHR Data Fusion framework that incorporates masked lab-test modeling and large language models (LLMs) to effectively integrate structured and unstructured medical data. MEDFuse leverages multimodal embeddings extracted from two sources: LLMs fine-tuned on free clinical text and masked tabular transformers trained on structured lab test results. We design a disentangled transformer module, optimized by a mutual information loss to 1) decouple modality-specific and modality-shared information and 2) extract useful joint representation from the noise and redundancy present in clinical notes. Through comprehensive validation on the public MIMIC-III dataset and the in-house FEMH dataset, MEDFuse demonstrates great potential in advancing clinical predictions, achieving over 90% F1 score in the 10-disease multi-label classification task.
Authors
- David Restrepo (ORCID: https://orcid.org/0000-0002-3789-1957)
- Wen-Chih Peng (ORCID: https://orcid.org/0000-0002-0172-7311)
- Tai Tan Phan (ORCID: https://orcid.org/0000-0001-6814-9459)
- Fang‐Ming Hung (ORCID: https://orcid.org/0000-0003-3501-5459)
- Jun-En Ding (ORCID: https://orcid.org/0000-0002-1233-138X)
- Jian-Zhe Wang
- Cong-Tinh Dao (ORCID: https://orcid.org/0000-0002-0777-6915)
- Feng Liu (ORCID: https://orcid.org/0000-0002-5225-8199)
- Chenwei Wu (ORCID: https://orcid.org/0009-0003-7178-1377)
- Shun Liu (ORCID: https://orcid.org/0009-0007-5240-9114)
Institutions
- Stevens Institute of Technology (US)
- National Yang Ming Chiao Tung University (TW)
- Shanghai University of Finance and Economics (CN)
- University of Michigan (US)
- Far Eastern Memorial Hospital (TW)
- Massachusetts Institute of Technology (US)
Publication Details
- Published
- 2024-10-20
- DOI
- https://doi.org/10.1145/3627673.3679962
- Citations
- 6
- Primary Topic
- Machine Learning in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 2.08