arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.10054cs.LGcs.AIstat.ML
收藏

StageNet: Stage-Aware Neural Networks for Health Risk Prediction

Peking University(北京大学);IQVIA(艾昆纬);Peking University Third Hospital(北京大学第三医院);Temple University(天普大学);University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);Georgia Institute of Technology(佐治亚理工学院)

Junyi Gao, Cao Xiao, Yasha Wang, Wen Tang, Lucas M. Glass, Jimeng Sun

英文摘要:

Deep learning has demonstrated success in health risk prediction especially for patients with chronic and progressing conditions. Most existing works focus on learning disease Network (StageNet) model to extract disease stage information from patient data and integrate it into risk prediction. StageNet is enabled by (1) a stage-aware long short-term memory (LSTM) module that extracts health stage variations unsupervisedly; (2) a stage-adaptive convolutional module that incorporates stage-related progression patterns into risk prediction. We evaluate StageNet on two real-world datasets and show that StageNet outperforms state-of-the-art models in risk prediction task and patient subtyping task. Compared to the best baseline model, StageNet achieves up to 12% higher AUPRC for risk prediction task on two real-world patient datasets. StageNet also achieves over 58% higher Calinski-Harabasz score (a cluster quality metric) for a patient subtyping task.

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

↑