arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.04785cs.CLcs.AI
收藏

基于电子健康记录数据的大语言多模态模型用于五年慢性病队列预测

Large Language Multimodal Models for 5-Year Chronic Disease Cohort Prediction Using EHR Data

School of Systems and Enterprises, Stevens Institute of Technology(史蒂文斯理工学院系统与企业学院);National Yang Ming Chiao Tung University(国立阳明交通大学);Institute of Hospital and Health Care Administration, National Yang Ming Chiao Tung University(国立阳明交通大学医院暨健康保健管理研究所);School of Computing and Information Science, Florida International University(佛罗里达国际大学计算与信息科学学院);University of Michigan(密歇根大学);Far Eastern Memorial Hospital(远东纪念医院)
Jun-En Ding, Phan Nguyen Minh Thao, Wen-Chih Peng, Jian-Zhe Wang, Chun-Cheng Chug, Min-Chen Hsieh, Yun-Chien Tseng, Ling Chen, Dongsheng Luo, Chi-Te Wang, Pei-f… 展开作者

Jun-En Ding, Phan Nguyen Minh Thao, Wen-Chih Peng, Jian-Zhe Wang, Chun-Cheng Chug, Min-Chen Hsieh, Yun-Chien Tseng, Ling Chen, Dongsheng Luo, Chi-Te Wang, Pei-fu Chen, Feng Liu, Fang-Ming Hung

AI总结:

本文提出大语言多模态模型框架,融合台湾医院五年电子健康记录中的临床笔记与实验室数据预测慢性病风险,结合注意力融合与Flan T-5实现最高76% AUROC。

摘要:

糖尿病等慢性病是全球发病率和死亡率的主要原因。已有大量研究尝试使用各种深度学习模型进行诊断。然而,大多数先前研究存在一定局限性,包括使用公开数据集(如MIMIC)以及数据不平衡问题。在本研究中,我们从台湾医院数据库收集了五年电子健康记录(EHRs),包括1,420,596条临床笔记、387,392项实验室检验结果以及超过1,505个实验室检验项目,重点研究预训练大语言模型。我们提出了一个新的大语言多模态模型(LLMMs)框架,融合临床笔记和实验室检验结果中的多模态数据,用于慢性病风险预测。我们的方法结合了文本嵌入编码器和多头注意力层来学习实验室检验值,并利用深度神经网络(DNN)模块将血液特征与慢性病语义融合到潜在空间中。在实验中,我们观察到clinicalBERT和PubMed-BERT在结合注意力融合时,可在多类慢性病和糖尿病预测中达到73%的准确率。通过将实验室检验值转换为文本描述并使用Flan T-5模型,我们实现了76%的ROC曲线下面积(AUROC),证明了利用数值文本数据进行语言模型训练和推理的有效性。该方法显著提高了早期糖尿病预测的准确性。

英文摘要:

Chronic diseases such as diabetes are the leading causes of morbidity and mortality worldwide. Numerous research studies have been attempted with various deep learning models in diagnosis. However, most previous studies had certain limitations, including using publicly available datasets (e.g. MIMIC), and imbalanced data. In this study, we collected five-year electronic health records (EHRs) from the Taiwan hospital database, including 1,420,596 clinical notes, 387,392 laboratory test results, and more than 1,505 laboratory test items, focusing on research pre-training large language models. We proposed a novel Large Language Multimodal Models (LLMMs) framework incorporating multimodal data from clinical notes and laboratory test results for the prediction of chronic disease risk. Our method combined a text embedding encoder and multi-head attention layer to learn laboratory test values, utilizing a deep neural network (DNN) module to merge blood features with chronic disease semantics into a latent space. In our experiments, we observe that clinicalBERT and PubMed-BERT, when combined with attention fusion, can achieve an accuracy of 73% in multiclass chronic diseases and diabetes prediction. By transforming laboratory test values into textual descriptions and employing the Flan T-5 model, we achieved a 76% Area Under the ROC Curve (AUROC), demonstrating the effectiveness of leveraging numerical text data for training and inference in language models. This approach significantly improves the accuracy of early-stage diabetes prediction.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑