用专家标注辅助学习多模态病历中的治疗策略,提升决策精准度。
Annotation-Assisted Learning of Treatment Policies From Multimodal Electronic Health Records
- 利用专家标注调整混杂因素,结合多模态数据学习治疗效果
- 在真实和模拟数据上优于基于风险或表示的基线方法
- 适合临床决策支持系统,推动因果机器学习落地
本文研究如何从包含表格数据与临床文本的多模态电子健康记录(EHR)中学习治疗策略,以辅助医生制定更优治疗方案并高效配置医疗资源。现有因果政策学习方法依赖于表格协变量的因果假设,在多模态场景下难以成立。直接将因果估计器应用于多模态表示可能导致偏差,因表示未保留关键混杂信息。实践中常使用基线风险预测模型指导治疗,但无法识别真正受益的患者。为此,本文提出AACE(Annotation-Assisted Coarsened Effects)方法:训练阶段利用专家提供的标注进行混杂调整,推理时仅基于多模态表示预测治疗收益。实验表明,该方法在合成、半合成及真实世界EHR数据集上均表现优异,显著优于基于风险和表示的因果基线,并为临床实践中的因果机器学习应用提供了实用洞见。
原文摘要 · Abstract (English)
We study how to learn treatment policies from multimodal electronic health records (EHRs) that consist of tabular data and clinical text. These policies can help physicians make better treatment decisions and allocate healthcare resources more efficiently. Causal policy learning methods prioritize patients with the largest expected treatment benefit. Yet, existing estimators are designed for tabular covariates under causal assumptions that may be hard to justify in the multimodal setting. A pragmatic alternative is to apply causal estimators directly to multimodal representations, but this can produce biased treatment effect estimates when the representations do not preserve the relevant confounding information. As a result, predictive models of baseline risk are commonly used in practice to guide treatment decisions, although they are not designed to identify which patients benefit most from treatment. We propose AACE (Annotation-Assisted Coarsened Effects), an annotation-assisted approach to causal policy learning for multimodal EHRs. The method uses expert-provided annotations during training to support confounding adjustment, and then predicts treatment benefit from only multimodal representations at inference. We show that the proposed method achieves strong empirical performance across synthetic, semi-synthetic, and real-world EHR datasets, outperforming risk-based and representation-based causal baselines, and offering practical insights for applying causal machine learning in clinical practice.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。