A Similarity-Driven Sequential Model for Predicting Medical Treatment Initiation via Graph Laplacian

Tong Wu,Yue Wang,Yunlong Wang,Cao Xiao,Emily Zhao,Lucas,Glass,Jimeng Sun
2019-01-01
Abstract:A diverse range of computational models have been proposed to extract temporal patterns from clinical time series data for each patient and among patient group for predictive healthcare, including multivariate regression, neural network, and decision tree, etc. These models typically learn an analytical mapping from the numerical or categorical patients’ medical data to the targeted labels. However, the common relations among patients (e.g., share the same doctor, or receive the same medical treatments) were rarely considered. In this paper, we represent the relation between patients and clinicians by bipartite graphs to explicitly address relations such as from which clinician a patient is diagnosed or offered a medical service. We then solve for the top eigenvectors of the Laplacian matrix of the bipartite graph, and include the eigenvectors as latent representations of the similarity between patient-clinician pairs into a time-sensitive prediction model. We conducted experiments using real-world data to predict the initiation of first-line treatment for Chronic Lymphocytic Leukemia (CLL) patients. Results show that the relational similarity uncovered from the patient-clinician bipartite graph can enhance prediction accuracies over multiple baselines, for example, a 5% increment over the long short-term memory sequential model in terms of area under precision-recall curve.
What problem does this paper attempt to address?