arXiv:2603.10180cs.LG2026-03被引 1

通过疾病轨迹建模,提升电子病历的可解释性与预测性能

DT-BEHRT: Disease Trajectory-aware Transformer for Interpretable Patient Representation Learning

  • 基于器官系统构建疾病轨迹感知的图增强序列模型
  • 在多个数据集上实现优异预测效果,优于主流方法
  • 适合临床辅助决策、可解释医疗AI研究者使用

电子健康记录(EHR)系统的广泛应用为临床决策支持的预测建模提供了前所未有的机遇。结构化EHR包含患者多次就诊的纵向数据,每次就诊由一组医疗代码表示。尽管已有基于序列、图结构及图增强序列的方法用于捕捉代码间的时序或同次就诊内交互,但往往忽视了因不同临床特征和上下文导致的医疗代码角色异质性。为此,本文提出疾病轨迹感知变压器(DT-BEHRT),一种图增强序列架构,通过显式建模器官系统内的诊断中心交互,以及捕捉异步进展模式,实现疾病轨迹解耦。为增强表示鲁棒性,设计了定制预训练策略,结合轨迹级代码掩码与本体引导的祖先预测,促进多模块间语义对齐。在多个基准数据集上的大量实验表明,DT-BEHRT在预测性能上表现强劲,并生成与临床医生以疾病为中心的推理一致的可解释患者表征。源代码已公开于 https://github.com/GatorAIM/DT-BEHRT.git。

原文摘要 · Abstract (English)

The growing adoption of electronic health record (EHR) systems has provided unprecedented opportunities for predictive modeling to guide clinical decision making. Structured EHRs contain longitudinal observations of patients across hospital visits, where each visit is represented by a set of medical codes. While sequence-based, graph-based, and graph-enhanced sequence approaches have been developed to capture rich code interactions over time or within the same visits, they often overlook the inherent heterogeneous roles of medical codes arising from distinct clinical characteristics and contexts. To this end, in this study we propose the Disease Trajectory-aware Transformer for EHR (DT-BEHRT), a graph-enhanced sequential architecture that disentangles disease trajectories by explicitly modeling diagnosis-centric interactions within organ systems and capturing asynchronous progression patterns. To further enhance the representation robustness, we design a tailored pre-training methodology that combines trajectory-level code masking with ontology-informed ancestor prediction, promoting semantic alignment across multiple modeling modules. Extensive experiments on multiple benchmark datasets demonstrate that DT-BEHRT achieves strong predictive performance and provides interpretable patient representations that align with clinicians' disease-centered reasoning. The source code is publicly accessible at https://github.com/GatorAIM/DT-BEHRT.git.

医疗AI图神经网络可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。