arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2411.03497cs.CL

(大型)语言模型在临床结局预测中的不确定性量化

Uncertainty Quantification for Clinical Outcome Predictions with (Large) Language Models

Brandeis University(布兰迪斯大学);Google(谷歌);GE Healthcare(GE医疗)

Zizhang Chen, Peizhao Li, Xiaomeng Dong, Pengyu Hong

AI总结:

本研究针对语言模型在临床预测中的不确定性,提出多任务与集成方法,并在白盒和黑盒设置下验证,利用六千余名患者数据,有效降低不确定性,提升AI医疗可靠性。

摘要:

为促进医疗服务的提供,语言模型(LMs)在使用电子健康记录(EHRs)进行临床预测任务方面具有巨大潜力。然而,在这些高风险应用中,由于患者安全和伦理问题受到损害,不可靠的决策可能导致高昂的成本,因此增加了对自动化临床预测进行良好不确定性建模的需求。为解决这一问题,我们考虑了在白盒和黑盒设置下,语言模型在电子健康记录任务中的不确定性量化。我们首先量化白盒模型中的不确定性,在这种情况下,我们可以访问模型参数和输出逻辑值。我们表明,通过使用所提出的多任务和集成方法,可以在电子健康记录中有效降低模型不确定性。延续这一思路,我们将方法扩展到黑盒设置,包括流行的专有语言模型,如GPT-4。我们使用来自超过6,000名患者的纵向临床数据,在十个临床预测任务中验证了我们的框架。结果表明,集成方法和多任务预测提示在不同场景下减少了不确定性。这些发现提高了模型在白盒和黑盒设置中的透明度,从而推进了可靠的人工智能医疗保健。

英文摘要:

To facilitate healthcare delivery, language models (LMs) have significant potential for clinical prediction tasks using electronic health records (EHRs). However, in these high-stakes applications, unreliable decisions can result in high costs due to compromised patient safety and ethical concerns, thus increasing the need for good uncertainty modeling of automated clinical predictions. To address this, we consider the uncertainty quantification of LMs for EHR tasks in white- and black-box settings. We first quantify uncertainty in white-box models, where we can access model parameters and output logits. We show that an effective reduction of model uncertainty can be achieved by using the proposed multi-tasking and ensemble methods in EHRs. Continuing with this idea, we extend our approach to black-box settings, including popular proprietary LMs such as GPT-4. We validate our framework using longitudinal clinical data from more than 6,000 patients in ten clinical prediction tasks. Results show that ensembling methods and multi-task prediction prompts reduce uncertainty across different scenarios. These findings increase the transparency of the model in white-box and black-box settings, thus advancing reliable AI healthcare.

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

↑