中文

基于锚词且由生存分析监督的主题建模:胰腺炎结局刻画

机器学习 2017-12-08 v2

摘要

我们提出了一种由生存分析监督的主题建模新方法。具体而言,我们基于近期关于使用所谓锚词的无监督主题建模工作,通过弹性网正则化的 Cox 比例风险模型提供监督。简言之,锚词在文档中出现强烈表明该文档部分涉及某一特定主题。例如,通过在文档中看到“胆结石”,我们相当确定该文档部分关于医学。我们提出的方法在学习的主题模型与生存模型之间交替,以寻找块凸优化问题的局部极小值。我们将所提方法应用于预测因胰腺炎入住重症监护病房(ICU)的患者在 ICU 的停留时长。我们的方法与各类基线中最佳者同样准确,同时比任何基线都更具可解释性。

关键词

引用

@article{arxiv.1712.00535,
  title  = {Survival-Supervised Topic Modeling with Anchor Words: Characterizing Pancreatitis Outcomes},
  author = {George H. Chen and Jeremy C. Weiss},
  journal= {arXiv preprint arXiv:1712.00535},
  year   = {2017}
}

备注

NIPS Workshop on Machine Learning for Health 2017, fixed some equation typos, some minor wording edits