arXiv:2603.08459cs.LG2026-03

用多模态临床数据构建先验,让医疗预测模型更懂自己有多少把握。

Data-Driven Priors for Uncertainty-Aware Deterioration Risk Prediction with Multimodal Data

  • 基于自监督表征与数据扰动设计数据驱动先验
  • 在MIMIC-IV/CXR上显著提升预测准确率与不确定性量化能力
  • 适合需要可信预测的高风险医疗AI场景

安全的预测是将预测模型集成到临床决策支持系统中的关键要求。一种确保可信性的方法是使模型能够表达对单个预测的不确定性。然而,当前机器学习模型通常缺乏可靠的不确定性估计,限制了实际部署。这一问题在多模态场景中尤为突出,其目标是实现有效信息融合。本文提出$ exttt{MedCertAIn}$,一个利用多模态临床数据进行院内风险预测的不确定性感知框架。通过结合自监督潜在表示中的跨模态相似性与模态特定的数据扰动,设计神经网络参数的数据驱动先验。使用公开数据集MIMIC-IV和MIMIC-CXR中的临床时序数据与胸部X光图像训练和评估模型。结果表明,$ exttt{MedCertAIn}$相比最先进的确定性基线和替代贝叶斯方法,在预测性能和不确定性量化方面均有显著提升。这些发现凸显了数据驱动先验在推动高可靠性、不确定性感知的AI工具用于高风险临床应用方面的潜力。

原文摘要 · Abstract (English)

Safe predictions are a crucial requirement for integrating predictive models into clinical decision support systems. One approach for ensuring trustworthiness is to enable models' ability to express their uncertainty about individual predictions. However, current machine learning models frequently lack reliable uncertainty estimation, hindering real-world deployment. This is further observed in multimodal settings, where the goal is to enable effective information fusion. In this work, we propose $\texttt{MedCertAIn}$, a predictive uncertainty framework that leverages multimodal clinical data for in-hospital risk prediction to improve model performance and reliability. We design data-driven priors over neural network parameters using a hybrid strategy that considers cross-modal similarity in self-supervised latent representations and modality-specific data corruptions. We train and evaluate the models with such priors using clinical time-series and chest X-ray images from the publicly-available datasets MIMIC-IV and MIMIC-CXR. Our results show that $\texttt{MedCertAIn}$ significantly improves predictive performance and uncertainty quantification compared to state-of-the-art deterministic baselines and alternative Bayesian methods. These findings highlight the promise of data-driven priors in advancing robust, uncertainty-aware AI tools for high-stakes clinical applications.

医疗AI不确定性估计多模态临床决策

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。