arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2204.04542cs.LG
收藏

生存Seq2Seq:一种基于序列到序列结构的生存模型

Survival Seq2Seq: A Survival Model based on Sequence to Sequence Architecture

Ortho Biomed Inc.(Ortho Biomed公司);University Health Network(大学健康网络);Toronto Rehabilitation Institute, University Health Network(大学健康网络多伦多康复研究所)

Ebrahim Pourjafari, Navid Ziaei, Mohammad R. Rezaei, Amir Sameizadeh, Mohammad Shafiee, Mohammad Alavinia, Mansour Abolghasemian, Nick Sajadi

摘要:

本文介绍了一种新的非参数深度模型,用于估计删失数据和竞争风险下的至事件时间(生存分析)。该模型基于序列到序列(Seq 2Seq)架构设计,因此我们将其命名为Survival Seq 2Seq。我们模型编码器的第一个递归神经网络(RNN)层由GRU-D细胞组成。这些细胞有能力有效地填补非缺失随机值的纵向数据集具有非常高的缺失率,如电子健康记录(EHR)。Survival Seq 2Seq的解码器为每个竞争风险生成概率分布函数(PDF),而不假设风险的任何先验分布。利用RNN单元,解码器能够生成平滑且几乎无尖峰的PDF。这超出了现有非参数深度模型的生存分析能力。在合成和医学数据集上的训练结果证明,Survival Seq 2Seq在预测的准确性和生成PDF的质量方面优于其他现有的深度生存模型。

英文摘要:

This paper introduces a novel non-parametric deep model for estimating time-to-event (survival analysis) in presence of censored data and competing risks. The model is designed based on the sequence-to-sequence (Seq2Seq) architecture, therefore we name it Survival Seq2Seq. The first recurrent neural network (RNN) layer of the encoder of our model is made up of Gated Recurrent Unit with Decay (GRU-D) cells. These cells have the ability to effectively impute not-missing-at-random values of longitudinal datasets with very high missing rates, such as electronic health records (EHRs). The decoder of Survival Seq2Seq generates a probability distribution function (PDF) for each competing risk without assuming any prior distribution for the risks. Taking advantage of RNN cells, the decoder is able to generate smooth and virtually spike-free PDFs. This is beyond the capability of existing non-parametric deep models for survival analysis. Training results on synthetic and medical datasets prove that Survival Seq2Seq surpasses other existing deep survival models in terms of the accuracy of predictions and the quality of generated PDFs.

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

↑