arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.19987cs.LGcs.IR
收藏

基于检索感知多模态建模的反事实理解用于生存预测

Counterfactual Understanding via Retrieval-aware Multimodal Modeling for Time-to-Event Survival Prediction

University of Engineering and Technology(工程大学);Hanoi Vietnam National University(河内越南国家大学);Delft University of Technology(代尔夫特理工大学)

Ha-Anh Hoang Nguyen, Tri-Duc Phan Le, Duc-Hoang Pham, Huy-Son Nguyen, Cam-Van Thi Nguyen, Duc-Trong Le, Hoang-Quynh Le

AI总结:

CURE通过多模态建模和潜在子组检索提升生存预测,优于现有基线模型。

摘要:

本文针对时间到事件的反事实生存预测问题,旨在优化异质性和删失数据存在下的个体化生存结果。我们提出了CURE框架,通过全面的多模态嵌入和潜在子组检索来推进反事实生存建模。CURE整合了临床、辅助临床、人口统计学和多组学信息,通过交叉注意力机制对齐并融合这些信息。复杂的多组学信号可以利用专家混合架构进行自适应细化,强调最有信息量的组学成分。在此表示基础上,CURE隐式检索患者特定的潜在子组,这些子组捕捉了基线生存动态和治疗依赖的变化。在METABRIC和TCGA-LUAD数据集上的实验结果表明,所提出的CURE模型在生存分析中一致优于强基线模型,通过时间依赖一致性指数(C^td)和整合布里尔分数(IBS)进行评估。这些发现突显了CURE在增强多模态理解方面的潜力,并为未来治疗推荐模型奠定了基础。所有代码和相关资源均已公开,以促进可重复性 https://github.com/L2R-UET/CURE。

英文摘要:

This paper tackles the problem of time-to-event counterfactual survival prediction, aiming to optimize individualized survival outcomes in the presence of heterogeneity and censored data. We propose CURE, a framework that advances counterfactual survival modeling via comprehensive multimodal embedding and latent subgroup retrieval. CURE integrates clinical, paraclinical, demographic, and multi-omics information, which are aligned and fused through cross-attention mechanisms. Complex multi-omics signals can be adaptively refined using a mixture-of-experts architecture, emphasizing the most informative omics components. Building upon this representation, CURE implicitly retrieves patient-specific latent subgroups that capture both baseline survival dynamics and treatment-dependent variations. Experimental results on METABRIC and TCGA-LUAD datasets demonstrate that proposed CURE model consistently outperforms strong baselines in survival analysis, evaluated using the Time-dependent Concordance Index ($C^{td}$) and Integrated Brier Score (IBS). These findings highlight the potential of CURE to enhance multimodal understanding and serve as a foundation for future treatment recommendation models. All code and related resources are publicly available to facilitate the reproducibility https://github.com/L2R-UET/CURE.

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

↑