中文

基于多源数据的深度序列模型预测自杀意念

机器学习 2019-11-12 v1 机器学习

摘要

本文提出一种利用深度序列模型从电子健康记录(EHR)与生态瞬时评估(EMA)数据预测自杀意念的新方法。EHR 纵向数据与 EMA 问卷表单均由异步、变长、随机采样的数据序列定义。在我们的方法中,我们分别用循环神经网络(RNN)对二者建模,并通过使用时间标记拼接各自隐藏态来对齐这两个序列。此外,我们引入注意力机制以提升长序列性能,并采用时间无关的预训练方案以应对极短序列。使用包含 1023 名患者的数据库,实验结果表明加入 EMA 记录将预测自杀意念诊断的系统召回率从仅基于 EHR 的先进方法的 48.13% 提升至 67.78%。此外,我们的方法通过隐空间的 t-SNE 表示提供可解释性。进而,最具相关性的输入特征被识别并从医学角度加以解释。

关键词

引用

@article{arxiv.1911.03522,
  title  = {Deep Sequential Models for Suicidal Ideation from Multiple Source Data},
  author = {Ignacio Peis and Pablo M. Olmos and Constanza Vera-Varela and María Luisa Barrigón and Philippe Courtet and Enrique Baca-García and Antonio Artés-Rodríguez},
  journal= {arXiv preprint arXiv:1911.03522},
  year   = {2019}
}

备注

Accepted for publication in IEEE Journal of Biomedical and Health Informatics (JBHI)