提出双向注意力机制,提升电子病历分类对缺失数据的鲁棒性
Bi-Axial Transformers: Addressing the Increasing Complexity of EHR Classification
- 同时关注临床变量和时间轴的双向注意力机制
- 在脓毒症预测上达当前最优,对缺失数据更鲁棒
- 可迁移的传感器嵌入适合下游任务,代码开源可复现
电子健康记录(EHR)是流行病学与临床研究的重要资源,其规模、时间序列长度及多模态融合趋势日益增长。尽管变压器模型因擅长捕捉长程依赖与并行处理而具备潜力,但受限于数据表示方式,常因无法有效建模信息缺失而性能下降。本文提出双轴变压器(Bi-Axial Transformer, BAT),通过同时关注临床变量与时间点两个轴,学习更丰富的数据关系,缓解数据稀疏问题。BAT在脓毒症预测任务中达到当前最优性能,在死亡率分类上也具有竞争力。相比其他变压器,其对数据缺失更具鲁棒性,并能学习可迁移的传感器嵌入。原有基线模型已使用PyTorch重实现并开源,便于复现与未来基准测试。
原文摘要 · Abstract (English)
Electronic Health Records (EHRs), the digital representation of a patient's medical history, are a valuable resource for epidemiological and clinical research. They are also becoming increasingly complex, with recent trends indicating larger datasets, longer time series, and multi-modal integrations. Transformers, which have rapidly gained popularity due to their success in natural language processing and other domains, are well-suited to address these challenges due to their ability to model long-range dependencies and process data in parallel. But their application to EHR classification remains limited by data representations, which can reduce performance or fail to capture informative missingness. In this paper, we present the Bi-Axial Transformer (BAT), which attends to both the clinical variable and time point axes of EHR data to learn richer data relationships and address the difficulties of data sparsity. BAT achieves state-of-the-art performance on sepsis prediction and is competitive to top methods for mortality classification. In comparison to other transformers, BAT demonstrates increased robustness to data missingness, and learns unique sensor embeddings which can be used in transfer learning. Baseline models, which were previously located across multiple repositories or utilized deprecated libraries, were re-implemented with PyTorch and made available for reproduction and future benchmarking.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。