arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1806.02873cs.CLcs.AI
收藏

具有时间感知注意的医学概念嵌入

Medical Concept Embedding with Time-Aware Attention

Nankai University(南开大学);National University of Singapore(新加坡国立大学);National University Health System(新加坡国立大学卫生系统)

Xiangrui Cai, Jinyang Gao, Kee Yuan Ngiam, Beng Chin Ooi, Ying Zhang, Xiaojie Yuan

摘要:

在电子病历(EMR)中嵌入医疗概念(如药物、程序和诊断代码)是医疗保健分析的核心。以往的医学概念嵌入工作将医学概念和电子病历分别作为词和文档。然而,这样的模型错过了EMR数据的时间性质。一方面,两个连续的医学概念并不表明它们在时间上是接近的,但它们之间的相关性可以通过时间间隔来揭示。另一方面,医学概念的时间范围通常变化很大(例如,\textit{common cold}和\textit{diabetes})。在本文中,我们建议将时间信息嵌入医疗代码。基于连续词袋模型,我们采用注意力机制为每个医学概念学习一个“软”时间感知的上下文窗口。通过聚类和最近邻搜索任务在公共和专有数据集上进行的实验证明了我们模型的有效性,表明它优于五个最先进的基线。

英文摘要:

Embeddings of medical concepts such as medication, procedure and diagnosis codes in Electronic Medical Records (EMRs) are central to healthcare analytics. Previous work on medical concept embedding takes medical concepts and EMRs as words and documents respectively. Nevertheless, such models miss out the temporal nature of EMR data. On the one hand, two consecutive medical concepts do not indicate they are temporally close, but the correlations between them can be revealed by the time gap. On the other hand, the temporal scopes of medical concepts often vary greatly (e.g., \textit{common cold} and \textit{diabetes}). In this paper, we propose to incorporate the temporal information to embed medical codes. Based on the Continuous Bag-of-Words model, we employ the attention mechanism to learn a "soft" time-aware context window for each medical concept. Experiments on public and proprietary datasets through clustering and nearest neighbour search tasks demonstrate the effectiveness of our model, showing that it outperforms five state-of-the-art baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑