LLM2D
不再黑箱操作:时间特征交叉注意力机制揭开临床预测建模的神秘面纱
No Black Box Anymore: Demystifying Clinical Predictive Modeling with Temporal-Feature Cross Attention Mechanism
作者: Yubo Li, Xinyu Yao, Rema Padman
发布日期: 3/28/2025
arXiv ID: oai:arXiv.org:2503.19285v2

摘要

arXiv:2503.19285v2 通告类型: 更新 摘要:尽管深度学习模型在临床预测任务中表现出卓越的性能,但解释性仍是一个显著的挑战。受到变压器架构的启发,我们引入了时变特征交叉注意机制(TFCAM),这是一种用于捕捉临床特征随时间动态交互的新颖深度学习框架,从而增强了预测准确性和解释性。在一项基于1422名慢性肾病患者的实验中,预测进展至终末期肾病,TFCAM 在LSTM和RETAIN基准之上表现出色,AUROC为0.95,F1分数为0.69。除了性能提升之外,TFCAM 还通过识别关键的时间周期、排名特征重要性以及量化特征在时间上的相互影响如何影响预测,提供了多级解释性。我们的方法解决了深度学习在医疗保健中的“黑盒”问题,为临床医生提供了关于疾病进展机制的透明见解,同时保持了最先进的预测性能。