arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.03482cs.LGstat.ML

临床试验结果预测中的不确定性量化

Uncertainty Quantification on Clinical Trial Outcome Prediction

Rensselaer Polytechnic Institute(伦斯勒理工学院);Stanford University(斯坦福大学);Stony Brook University Hospital(石溪大学医院);Purdue University(普渡大学);University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校)

Tianyi Chen, Yingzhou Lu, Nan Hao, Yuanyuan Zhang, Capucine Van Rechem, Jintai Chen, Tianfan Fu

AI总结:

本文将选择性分类与 HINT 结合用于临床试验结果预测,通过对低置信样本弃权(不执行)来量化不确定性,显著提升各期试验的 PR-AUC 等指标。

摘要:

在机器学习的各个领域中,不确定性量化的重要性正日益得到认可。准确评估模型预测的不确定性,有助于为研究人员和从业者提供更深入的理解与信心。这一点在医学诊断和药物发现领域尤为关键,因为可靠的预测会直接影响研究质量和患者健康。在本文中,我们提出将不确定性量化纳入临床试验结果预测。我们的主要目标是增强模型辨别细微差异的能力,从而显著提升其整体性能。为实现这一目标,我们采用了选择性分类方法,并将其与处于临床试验预测建模前沿的 Hierarchical Interaction Network(HINT,层次交互网络)无缝集成。选择性分类涵盖一系列不确定性量化方法,使模型在面对被标记为模糊或低置信度的样本时能够弃权(不执行)决策,从而提高其选择分类的实例的预测准确率。一系列综合实验表明,将选择性分类纳入临床试验预测可显著增强模型性能,PR-AUC、F1、ROC-AUC 和总体准确率等关键指标均有明显提升。具体而言,在 I 期、II 期和 III 期试验结果预测中,所提方法相对于基础模型(HINT)在 PR-AUC 上分别实现了 32.37%、21.43% 和 13.27% 的相对提升。在预测 III 期时,我们的方法达到了 0.9022 的 PR-AUC 分数。这些发现说明了该策略在临床试验预测领域中的稳健性和潜在效用,并有可能在该领域设立新的基准。

英文摘要:

The importance of uncertainty quantification is increasingly recognized in the diverse field of machine learning. Accurately assessing model prediction uncertainty can help provide deeper understanding and confidence for researchers and practitioners. This is especially critical in medical diagnosis and drug discovery areas, where reliable predictions directly impact research quality and patient health. In this paper, we proposed incorporating uncertainty quantification into clinical trial outcome predictions. Our main goal is to enhance the model's ability to discern nuanced differences, thereby significantly improving its overall performance. We have adopted a selective classification approach to fulfill our objective, integrating it seamlessly with the Hierarchical Interaction Network (HINT), which is at the forefront of clinical trial prediction modeling. Selective classification, encompassing a spectrum of methods for uncertainty quantification, empowers the model to withhold decision-making in the face of samples marked by ambiguity or low confidence, thereby amplifying the accuracy of predictions for the instances it chooses to classify. A series of comprehensive experiments demonstrate that incorporating selective classification into clinical trial predictions markedly enhances the model's performance, as evidenced by significant upticks in pivotal metrics such as PR-AUC, F1, ROC-AUC, and overall accuracy. Specifically, the proposed method achieved 32.37\%, 21.43\%, and 13.27\% relative improvement on PR-AUC over the base model (HINT) in phase I, II, and III trial outcome prediction, respectively. When predicting phase III, our method reaches 0.9022 PR-AUC scores. These findings illustrate the robustness and prospective utility of this strategy within the area of clinical trial predictions, potentially setting a new benchmark in the field.

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

↑