Subgraph augmented non-negative tensor factorization (SANTF) for modeling clinical narrative text

Yuan Luo*, Yu Xin, Ephraim Hochberg, Rohit Joshi, Ozlem Uzuner, Peter Szolovits

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

36 Scopus citations


Objective Extracting medical knowledge from electronic medical records requires automated approaches to combat scalability limitations and selection biases. However, existing machine learning approaches are often regarded by clinicians as black boxes. Moreover, training data for these automated approaches at often sparsely annotated at best. The authors target unsupervised learning for modeling clinical narrative text, aiming at improving both accuracy and interpretability. Methods The authors introduce a novel framework named subgraph augmented non-negative tensor factorization (SANTF). In addition to relying on atomic features (e.g., words in clinical narrative text), SANTF automatically mines higher-order features (e.g., relations of lymphoid cells expressing antigens) from clinical narrative text by converting sentences into a graph representation and identifying important subgraphs. The authors compose a tensor using patients, higher-order features, and atomic features as its respective modes. We then apply non-negative tensor factorization to cluster patients, and simultaneously identify latent groups of higher-order features that link to patient clusters, as in clinical guidelines where a panel of immunophenotypic features and laboratory results are used to specify diagnostic criteria. Results and Conclusion SANTF demonstrated over 10% improvement in averaged F-measure on patient clustering compared to widely used nonnegative matrix factorization (NMF) and k-means clustering methods. Multiple baselines were established by modeling patient data using patientby- features matrices with different feature configurations and then performing NMF or k-means to cluster patients. Feature analysis identified latent groups of higher-order features that lead to medical insights. We also found that the latent groups of atomic features help to better correlate the latent groups of higher-order features.

Original languageEnglish (US)
Pages (from-to)1009-1019
Number of pages11
JournalJournal of the American Medical Informatics Association
Issue number5
StatePublished - Sep 2015


  • Natural language processing
  • Non-negative tensor factorization
  • Subgraph mining
  • Unsupervised learning

ASJC Scopus subject areas

  • Health Informatics


Dive into the research topics of 'Subgraph augmented non-negative tensor factorization (SANTF) for modeling clinical narrative text'. Together they form a unique fingerprint.

Cite this