Improved Interpretability of Machine Learning Model Using Unsupervised Clustering: Predicting Time to First Treatment in Chronic Lymphocytic Leukemia.

Journal: JCO clinical cancer informatics
PMID:

Abstract

PURPOSE: Time to event is an important aspect of clinical decision making. This is particularly true when diseases have highly heterogeneous presentations and prognoses, as in chronic lymphocytic lymphoma (CLL). Although machine learning methods can readily learn complex nonlinear relationships, many methods are criticized as inadequate because of limited interpretability. We propose using unsupervised clustering of the continuous output of machine learning models to provide discrete risk stratification for predicting time to first treatment in a cohort of patients with CLL.

Authors

  • David Chen
    Temerty Faculty of Medicine, University of Toronto, Toronto, ON, Canada.
  • Gaurav Goyal
    Mayo Clinic Rochester, Rochester, MN.
  • Ronald S Go
    Mayo Clinic Rochester, Rochester, MN.
  • Sameer A Parikh
    Mayo Clinic Rochester, Rochester, MN.
  • Che G Ngufor
    Mayo Clinic Rochester, Rochester, MN.