Survival-Supervised Topic Modeling with Anchor Words: Characterizing Pancreatitis Outcomes
We introduce a new approach for topic modeling that is supervised by survival analysis. Specifically, we build on recent work on unsupervised topic modeling with so-called anchor words by providing supervision through an elastic-net regularized Cox proportional hazards model. In short, an anchor word being present in a document provides strong indication that the document is partially about a specific topic. For example, by seeing "gallstones" in a document, we are fairly certain that the document is partially about medicine. Our proposed method alternates between learning a topic model and learning a survival model to find a local minimum of a block convex optimization problem. We apply our proposed approach to predicting how long patients with pancreatitis admitted to an intensive care unit (ICU) will stay in the ICU. Our approach is as accurate as the best of a variety of baselines while being more interpretable than any of the baselines.
Code (0)
등록된 구현이 없습니다.
Tasks
Survival AnalysisSimilar Papers 제목 키워드 기반
Topic Models with Survival Supervision: Archetypal Analysis and Neural Approaches
We introduce two approaches to topic modeling supervised by survival analysis. Both approaches predict time-to-event outcomes while simultaneously learning topics over features that help prediction. The high-level idea i…
Survival AnalysisTime-to-Event PredictionTopic ModelsInteractive Topic Modeling with Anchor Words
The formalism of anchor words has enabled the development of fast topic modeling algorithms with provable guarantees. In this paper, we introduce a protocol that allows users to interact with anchor words to build custom…
Topic ModelsAnchored Correlation Explanation: Topic Modeling with Minimal Domain Knowledge
While generative models such as Latent Dirichlet Allocation (LDA) have proven fruitful in topic modeling, they often require detailed assumptions and careful specification of hyperparameters. Such model complexity issues…
Labeled Anchors and a Scalable, Transparent, and Interactive Classifier
We propose Labeled Anchors, an interactive and supervised topic model based on the anchor words algorithm (Arora et al., 2013). Labeled Anchors is similar to Supervised Anchors (Nguyen et al., 2014) in that it extends th…
Document ClassificationGeneral ClassificationText ClassificationTopic ModelsA New Anchor Word Selection Method for the Separable Topic Discovery
Separable Non-negative Matrix Factorization (SNMF) is an important method for topic modeling, where "separable" assumes every topic contains at least one anchor word, defined as a word that has non-zero probability only …
Word Similarity