LLM2D
不再神秘莫测: temporal-feature 跨注意力机制揭秘临床预测建模
No Black Box Anymore: Demystifying Clinical Predictive Modeling with Temporal-Feature Cross Attention Mechanism
作者: Yubo Li, Xinyu Yao, Rema Padman
发布日期: 3/26/2025
arXiv ID: oai:arXiv.org:2503.19285v1

摘要

arXiv:2503.19285v1 Announce Type: cross 摘要:尽管深度学习模型在临床预测任务中表现出色,但可解释性仍然是一个重要的挑战。受变压器架构的启发,我们引入了时间-特征交叉注意机制(TFCAM),这是一种新颖的深度学习框架,旨在捕捉临床特征随时间动态交互,从而提高预测准确性和可解释性。在包含1,422名慢性肾病患者的实验中,预测进展至终末期肾病,TFCAM在与LSTM和RETAINT基线相比时表现出色,实现了AUROC为0.95和F1分数为0.69。除了性能提升外,TFCAM还通过识别关键的时间周期、对特征重要性进行排名以及量化特征如何在影响预测前随时间相互影响,提供了多级解释性。我们的方法解决了深度学习在医疗保健中的“黑箱”局限性,为临床医生提供了透明的疾病进展机制见解,同时维持了最先进的预测性能。