arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2103.02768cs.LG
收藏

学习用支持证据预测:临床风险预测的应用

Learning to Predict with Supporting Evidence: Applications to Clinical Risk Prediction

Massachusetts Institute of Technology(麻省理工学院);Harvard Medical School(哈佛医学院);Massachusetts General Hospital(麻省总医院)

Aniruddh Raghu, John Guttag, Katherine Young, Eugene Pomerantsev, Adrian V. Dalca, Collin M. Stultz

摘要:

机器学习模型对医疗保健的影响将取决于医疗保健专业人员对这些模型预测的信任程度。在本文中,我们提出了一种方法,为人们提供临床专业知识与领域相关的证据,为什么预测应该被信任。我们首先设计了一个概率模型,将有意义的潜在概念与预测目标和观测数据联系起来。在这个模型中,潜变量的推断对应于做出预测和为该预测提供支持证据。我们提出了一个两步的过程,以有效地近似推理:(i)估计模型参数,使用变分学习,和(ii)近似最大后验估计的模型中的潜变量使用神经网络,训练的目标来自概率模型。我们展示了预测心血管疾病患者死亡风险的任务的方法。具体来说,使用心电图和表格数据作为输入,我们表明,我们的方法提供了适当的领域相关的支持证据,准确的预测。

英文摘要:

The impact of machine learning models on healthcare will depend on the degree of trust that healthcare professionals place in the predictions made by these models. In this paper, we present a method to provide people with clinical expertise with domain-relevant evidence about why a prediction should be trusted. We first design a probabilistic model that relates meaningful latent concepts to prediction targets and observed data. Inference of latent variables in this model corresponds to both making a prediction and providing supporting evidence for that prediction. We present a two-step process to efficiently approximate inference: (i) estimating model parameters using variational learning, and (ii) approximating maximum a posteriori estimation of latent variables in the model using a neural network, trained with an objective derived from the probabilistic model. We demonstrate the method on the task of predicting mortality risk for patients with cardiovascular disease. Specifically, using electrocardiogram and tabular data as input, we show that our approach provides appropriate domain-relevant supporting evidence for accurate predictions.

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

↑