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